a
    é´«j<C  ã                   @  s  d dl mZ d dlmZmZ d dlZd dlmZmZm	Z	 d dl
Zd dlmZ d dlZd dlmZ er€d dlmZ d dlmZmZ d	d	d
œdd„Zd*d	d	dddœdd„Zdd	dd	ddœdd„Zdd	ddœdd„Zd+dd	ddœdd „Zd,dd"d#d$d$d%d	dd&d'œ	d(d)„ZdS )-é    )Úannotations)ÚabcÚdefaultdictN)ÚTYPE_CHECKINGÚAnyÚDefaultDict©Úconvert_json_to_lines)Ú	DataFrame)ÚIterable)ÚIgnoreRaiseÚScalarÚstr)ÚsÚreturnc                 C  s0   | d dks| d dkr| S | dd… } t | ƒS )zJ
    Helper function that converts JSON lists to line delimited JSON.
    r   ú[éÿÿÿÿú]é   r   )r   © r   úX/var/www/html/laksyashrimp/venv/lib/python3.9/site-packages/pandas/io/json/_normalize.pyÚconvert_to_line_delimits    s    r   Ú Ú.Úintz
int | None)ÚprefixÚsepÚlevelÚ	max_levelc              
   C  sâ   d}t | tƒr| g} d}g }| D ]°}t |¡}| ¡ D ]Ž\}	}
t |	tƒsPt|	ƒ}	|dkr^|	}n|| |	 }t |
tƒr„|dur ||kr |dkr6| |	¡}
|
||< q6| |	¡}
| t|
|||d |ƒ¡ q6| 	|¡ q |rÞ|d S |S )a�  
    A simplified json_normalize

    Converts a nested dict into a flat dict ("record"), unlike json_normalize,
    it does not attempt to extract a subset of the data.

    Parameters
    ----------
    ds : dict or list of dicts
    prefix: the prefix, optional, default: ""
    sep : str, default '.'
        Nested records will generate names separated by sep,
        e.g., for sep='.', { 'foo' : { 'bar' : 0 } } -> foo.bar
    level: int, optional, default: 0
        The number of levels in the json string.

    max_level: int, optional, default: None
        The max depth to normalize.

    Returns
    -------
    d - dict or list of dicts, matching `ds`

    Examples
    --------
    >>> nested_to_record(
    ...     dict(flat1=1, dict1=dict(c=1, d=2), nested=dict(e=dict(c=1, d=2), d=2))
    ... )
    {'flat1': 1, 'dict1.c': 1, 'dict1.d': 2, 'nested.e.c': 1, 'nested.e.d': 2, 'nested.d': 2}
    FTr   Nr   )
Ú
isinstanceÚdictÚcopyÚdeepcopyÚitemsr   ÚpopÚupdateÚnested_to_recordÚappend)Údsr   r   r   r   Z	singletonZnew_dsÚdZnew_dÚkÚvZnewkeyr   r   r   r&   -   s8    ,



ÿÿ

r&   r   údict[str, Any])ÚdataÚ
key_stringÚnormalized_dictÚ	separatorr   c                 C  sX   t | tƒrL|  ¡ D ]6\}}|› |› |› �}|s8| |¡}t||||d� qn| ||< |S )a3  
    Main recursive function
    Designed for the most basic use case of pd.json_normalize(data)
    intended as a performance improvement, see #15621

    Parameters
    ----------
    data : Any
        Type dependent on types contained within nested Json
    key_string : str
        New key (with separator(s) in) for data
    normalized_dict : dict
        The new normalized/flattened Json dict
    separator : str, default '.'
        Nested records will generate names separated by sep,
        e.g., for sep='.', { 'foo' : { 'bar' : 0 } } -> foo.bar
    ©r-   r.   r/   r0   )r   r    r#   ÚremoveprefixÚ_normalise_json)r-   r.   r/   r0   ÚkeyÚvalueZnew_keyr   r   r   r3   ~   s    

ü
r3   )r-   r0   r   c                 C  s<   dd„ |   ¡ D ƒ}tdd„ |   ¡ D ƒdi |d�}i |¥|¥S )aw  
    Order the top level keys and then recursively go to depth

    Parameters
    ----------
    data : dict or list of dicts
    separator : str, default '.'
        Nested records will generate names separated by sep,
        e.g., for sep='.', { 'foo' : { 'bar' : 0 } } -> foo.bar

    Returns
    -------
    dict or list of dicts, matching `normalised_json_object`
    c                 S  s    i | ]\}}t |tƒs||“qS r   ©r   r    ©Ú.0r*   r+   r   r   r   Ú
<dictcomp>¶   ó    z+_normalise_json_ordered.<locals>.<dictcomp>c                 S  s    i | ]\}}t |tƒr||“qS r   r6   r7   r   r   r   r9   ¸   r:   r   r1   )r#   r3   )r-   r0   Z	top_dict_Znested_dict_r   r   r   Ú_normalise_json_ordered§   s    ür;   zdict | list[dict]zdict | list[dict] | Any)r(   r   r   c                   s@   i }t | tƒrt| ˆ d�}n t | tƒr<‡ fdd„| D ƒ}|S |S )a˜  
    A optimized basic json_normalize

    Converts a nested dict into a flat dict ("record"), unlike
    json_normalize and nested_to_record it doesn't do anything clever.
    But for the most basic use cases it enhances performance.
    E.g. pd.json_normalize(data)

    Parameters
    ----------
    ds : dict or list of dicts
    sep : str, default '.'
        Nested records will generate names separated by sep,
        e.g., for sep='.', { 'foo' : { 'bar' : 0 } } -> foo.bar

    Returns
    -------
    frame : DataFrame
    d - dict or list of dicts, matching `normalised_json_object`

    Examples
    --------
    >>> _simple_json_normalize(
    ...     {
    ...         "flat1": 1,
    ...         "dict1": {"c": 1, "d": 2},
    ...         "nested": {"e": {"c": 1, "d": 2}, "d": 2},
    ...     }
    ... )
    {'flat1': 1, 'dict1.c': 1, 'dict1.d': 2, 'nested.e.c': 1, 'nested.e.d': 2, 'nested.d': 2}

    )r-   r0   c                   s   g | ]}t |ˆ d �‘qS )©r   )Ú_simple_json_normalize)r8   Úrowr<   r   r   Ú
<listcomp>ð   r:   z*_simple_json_normalize.<locals>.<listcomp>)r   r    r;   Úlist)r(   r   Znormalised_json_objectZnormalised_json_listr   r<   r   r=   À   s    +

r=   Úraisezstr | list | Nonez"str | list[str | list[str]] | Nonez
str | Noner   r
   )	r-   Úrecord_pathÚmetaÚmeta_prefixÚrecord_prefixÚerrorsr   r   r   c                   s>  d#dddddœ‡fdd„‰ddd	d
œ‡fdd„‰t | tƒrD| sDtƒ S t | tƒrV| g} n$t | tjƒrvt | tƒsvt| ƒ} nt‚|du r²|du r²|du r²ˆ	du r²ˆdu r²tt| ˆd�ƒS |du rât	dd„ | D ƒƒrÚt
| ˆˆd�} t| ƒS t |tƒsò|g}|du �rg }nt |tƒ�s|g}dd„ |D ƒ‰ g ‰
g ‰ttƒ‰‡fdd„ˆ D ƒ‰d$dddœ‡ ‡‡‡‡‡‡‡‡
‡f
dd„‰ˆ| |i dd� tˆ
ƒ}ˆ	du�r¢|j‡	fdd„d�}ˆ ¡ D ]Ž\}	}
|du�rÄ||	 }	|	|v �rÞtd|	› d �ƒ‚tj|
td!�}|jd"k�r(tjt|
ƒftd!�}t|
ƒD ]\}}
|
||< �q| ˆ¡||	< �qª|S )%a´  
    Normalize semi-structured JSON data into a flat table.

    Parameters
    ----------
    data : dict or list of dicts
        Unserialized JSON objects.
    record_path : str or list of str, default None
        Path in each object to list of records. If not passed, data will be
        assumed to be an array of records.
    meta : list of paths (str or list of str), default None
        Fields to use as metadata for each record in resulting table.
    meta_prefix : str, default None
        If True, prefix records with dotted (?) path, e.g. foo.bar.field if
        meta is ['foo', 'bar'].
    record_prefix : str, default None
        If True, prefix records with dotted (?) path, e.g. foo.bar.field if
        path to records is ['foo', 'bar'].
    errors : {'raise', 'ignore'}, default 'raise'
        Configures error handling.

        * 'ignore' : will ignore KeyError if keys listed in meta are not
          always present.
        * 'raise' : will raise KeyError if keys listed in meta are not
          always present.
    sep : str, default '.'
        Nested records will generate names separated by sep.
        e.g., for sep='.', {'foo': {'bar': 0}} -> foo.bar.
    max_level : int, default None
        Max number of levels(depth of dict) to normalize.
        if None, normalizes all levels.

    Returns
    -------
    frame : DataFrame
    Normalize semi-structured JSON data into a flat table.

    Examples
    --------
    >>> data = [
    ...     {"id": 1, "name": {"first": "Coleen", "last": "Volk"}},
    ...     {"name": {"given": "Mark", "family": "Regner"}},
    ...     {"id": 2, "name": "Faye Raker"},
    ... ]
    >>> pd.json_normalize(data)
        id name.first name.last name.given name.family        name
    0  1.0     Coleen      Volk        NaN         NaN         NaN
    1  NaN        NaN       NaN       Mark      Regner         NaN
    2  2.0        NaN       NaN        NaN         NaN  Faye Raker

    >>> data = [
    ...     {
    ...         "id": 1,
    ...         "name": "Cole Volk",
    ...         "fitness": {"height": 130, "weight": 60},
    ...     },
    ...     {"name": "Mark Reg", "fitness": {"height": 130, "weight": 60}},
    ...     {
    ...         "id": 2,
    ...         "name": "Faye Raker",
    ...         "fitness": {"height": 130, "weight": 60},
    ...     },
    ... ]
    >>> pd.json_normalize(data, max_level=0)
        id        name                        fitness
    0  1.0   Cole Volk  {'height': 130, 'weight': 60}
    1  NaN    Mark Reg  {'height': 130, 'weight': 60}
    2  2.0  Faye Raker  {'height': 130, 'weight': 60}

    Normalizes nested data up to level 1.

    >>> data = [
    ...     {
    ...         "id": 1,
    ...         "name": "Cole Volk",
    ...         "fitness": {"height": 130, "weight": 60},
    ...     },
    ...     {"name": "Mark Reg", "fitness": {"height": 130, "weight": 60}},
    ...     {
    ...         "id": 2,
    ...         "name": "Faye Raker",
    ...         "fitness": {"height": 130, "weight": 60},
    ...     },
    ... ]
    >>> pd.json_normalize(data, max_level=1)
        id        name  fitness.height  fitness.weight
    0  1.0   Cole Volk             130              60
    1  NaN    Mark Reg             130              60
    2  2.0  Faye Raker             130              60

    >>> data = [
    ...     {
    ...         "state": "Florida",
    ...         "shortname": "FL",
    ...         "info": {"governor": "Rick Scott"},
    ...         "counties": [
    ...             {"name": "Dade", "population": 12345},
    ...             {"name": "Broward", "population": 40000},
    ...             {"name": "Palm Beach", "population": 60000},
    ...         ],
    ...     },
    ...     {
    ...         "state": "Ohio",
    ...         "shortname": "OH",
    ...         "info": {"governor": "John Kasich"},
    ...         "counties": [
    ...             {"name": "Summit", "population": 1234},
    ...             {"name": "Cuyahoga", "population": 1337},
    ...         ],
    ...     },
    ... ]
    >>> result = pd.json_normalize(
    ...     data, "counties", ["state", "shortname", ["info", "governor"]]
    ... )
    >>> result
             name  population    state shortname info.governor
    0        Dade       12345   Florida    FL    Rick Scott
    1     Broward       40000   Florida    FL    Rick Scott
    2  Palm Beach       60000   Florida    FL    Rick Scott
    3      Summit        1234   Ohio       OH    John Kasich
    4    Cuyahoga        1337   Ohio       OH    John Kasich

    >>> data = {"A": [1, 2]}
    >>> pd.json_normalize(data, "A", record_prefix="Prefix.")
        Prefix.0
    0          1
    1          2

    Returns normalized data with columns prefixed with the given string.
    Fr,   z
list | strÚboolzScalar | Iterable)ÚjsÚspecÚextract_recordr   c              
     s°   | }z:t |tƒr4|D ]}|du r(t|ƒ‚|| }qn|| }W nl tyª } zT|rdtd|› d�ƒ|‚ˆ dkr~tjW  Y d}~S td|› d|› d�ƒ|‚W Y d}~n
d}~0 0 |S )zInternal function to pull fieldNzKey zS not found. If specifying a record_path, all elements of data should have the path.Úignorez) not found. To replace missing values of z% with np.nan, pass in errors='ignore')r   r@   ÚKeyErrorÚnpÚnan)rH   rI   rJ   ÚresultÚfieldÚe)rF   r   r   Ú_pull_field‚  s.    

ÿýÿýz#json_normalize.<locals>._pull_fieldr@   )rH   rI   r   c                   sF   ˆ | |dd�}t |tƒsBt |¡r(g }nt| › d|› d|› d�ƒ‚|S )z¶
        Internal function to pull field for records, and similar to
        _pull_field, but require to return list. And will raise error
        if has non iterable value.
        T)rJ   z has non list value z
 for path z. Must be list or null.)r   r@   ÚpdZisnullÚ	TypeError)rH   rI   rO   )rR   r   r   Ú_pull_recordsŸ  s    

ÿz%json_normalize.<locals>._pull_recordsNr<   c                 s  s    | ]}d d„ |  ¡ D ƒV  qdS )c                 S  s   g | ]}t |tƒ‘qS r   r6   )r8   Úxr   r   r   r?   Ì  r:   z,json_normalize.<locals>.<genexpr>.<listcomp>N)Úvalues)r8   Úyr   r   r   Ú	<genexpr>Ì  r:   z!json_normalize.<locals>.<genexpr>©r   r   c                 S  s    g | ]}t |tƒr|n|g‘qS r   )r   r@   )r8   Úmr   r   r   r?   Þ  r:   z"json_normalize.<locals>.<listcomp>c                   s   g | ]}ˆ   |¡‘qS r   )Újoin)r8   Úvalr<   r   r   r?   å  r:   r   r   ÚNone)r   r   c           	        s  t | tƒr| g} t|ƒdkr‚| D ]^}tˆ ˆƒD ]*\}}|d t|ƒkr.ˆ||d ƒ||< q.ˆ||d  |dd … ||d d� q n’| D ]Œ}ˆ||d ƒ}‡‡	fdd„|D ƒ}ˆ t|ƒ¡ tˆ ˆƒD ]B\}}|d t|ƒkræ|| }nˆ|||d … ƒ}ˆ|  |¡ qÄˆ |¡ q†d S )Nr   r   r   ©r   c                   s(   g | ] }t |tƒr t|ˆˆ d �n|‘qS )rZ   )r   r    r&   )r8   Úr)r   r   r   r   r?   ô  s   þÿz>json_normalize.<locals>._recursive_extract.<locals>.<listcomp>)r   r    ÚlenÚzipr'   Úextend)	r-   ÚpathZ	seen_metar   Úobjr]   r4   ZrecsZmeta_val)
Ú_metarR   rU   Ú_recursive_extractÚlengthsr   Ú	meta_keysÚ	meta_valsÚrecordsr   r   r   rg   ç  s(    
(ü
z*json_normalize.<locals>._recursive_extractr_   c                   s   ˆ › | › �S )Nr   )rV   )rE   r   r   Ú<lambda>
  r:   z json_normalize.<locals>.<lambda>)ÚcolumnszConflicting metadata name z, need distinguishing prefix )Zdtyper   )F)r   )r   r@   r
   r    r   r   r   ÚNotImplementedErrorr=   Úanyr&   r   Úrenamer#   Ú
ValueErrorrM   ÚarrayÚobjectÚndimÚemptyra   Ú	enumerateÚrepeat)r-   rB   rC   rD   rE   rF   r   r   rO   r*   r+   rW   Úir   )rf   rR   rU   rg   rF   rh   r   ri   rj   rE   rk   r   r   Újson_normalizeõ   sp      ÿ

ÿþýüû

(



ÿry   )r   r   r   N)r   )NNNNrA   r   N)Ú
__future__r   Úcollectionsr   r   r!   Útypingr   r   r   ÚnumpyrM   Zpandas._libs.writersr	   ZpandasrS   r
   Úcollections.abcr   Zpandas._typingr   r   r   r&   r3   r;   r=   ry   r   r   r   r   Ú<module>   s8       ûQ) þ7       ø