FZjM\ddlmZddlmZGddejZGddeZy))UserList)apic"eZdZdZfdZxZS)APIzAn experimental wrapper around the base MyGeotab API class that adds some helper methods to results when retrieving results with `get()`. c :tt||fi||S)aGets entities using the API. Shortcut for using call() with the 'Get' method. This returns an EntityList with added convience methods. :param type_name: The type of entity. :type type_name: str :param parameters: Additional parameters to send. :raise MyGeotabException: Raises when an exception occurs on the MyGeotab server. :raise TimeoutException: Raises when the request does not respond after some time. :return: The results from the server. :rtype: EntityList  type_name) EntityListsuperget)selfr parameters __class__s YC:\Users\Rodrigo\frota-trucks-cliente\python\Lib\site-packages\mygeotab/ext/entitylist.pyr zAPI.get s %'+i>:>)TT)__name__ __module__ __qualname____doc__r __classcell__rs@rrrs U UrrceZdZdZfdZdZdZdZdZdZ dZ e Z d Z dd Z ed Zed Zed ZddZxZS)r zThe customized result listc:tt| |||_y)zGets entities using the API. Shortcut for using call() with the 'Get' method. :param data: The list of result data. :type data: list :param type_name: The type of entity. :type type_name: str N)r r __init__r )r datar rs rrzEntityList.__init__s j$(."rc|r+|jdj|jy|jddj|jd5t |j D]9\}}|r!|jd|j |j|; dddy#1swYyxYw)zThe pretty printer for IPythonz{}(...)z{}([z]),N)textformatr group enumerater breakablepretty)r pcycleidxitems r _repr_pretty_zEntityList._repr_pretty_)s  FF9##DNN3 4FMM$..94@!*499!5ICs  HHTN "6A@@s AB66B?ct|tr)|j|j||jS|j|SN) isinstanceslicerrr )r is r __getitem__zEntityList.__getitem__5s7 a >>$))A,? ?99Q< rct|d}t|d}|j|j|||jS)Nr)maxrrr )r r.js r __getslice__zEntityList.__getslice__;s7 1I 1I~~dii!ndnn==rc|t|tr3|j|j|jz|jSt|t |jr)|j|j|z|jS|j|jt |z|jSr+r,rrrr typelistr others r__add__zEntityList.__add__@s eX &>>$))ejj"8$..I I tDII />>$))e"3T^^D D~~dii$u+5t~~FFrc|t|tr3|j|j|jz|jSt|t |jr)|j||jz|jS|jt ||jz|jSr+r5r8s r__radd__zEntityList.__radd__Gs eX &>>%**tyy"8$..I I tDII />>%$))"3T^^D D~~d5kDII5t~~FFrcT|j|j|z|jSr+)rrr )r ns r__mul__zEntityList.__mul__Ns~~dii!mT^^<.sort_by_keyds'#;D$$zz|#Kr)rLreverser)rsortedrr )r rLrNrMs ` rsort_byzEntityList.sort_byZs1  ~~fTYYKQ]a]k]k~llrc<|jr|jdSdS)zOGets the first entity in the list, if it exists. :rtype: dict rNrr s rfirstzEntityList.firstls $yytyy|2d2rc<|jr|jdSdS)zNGets the last entity in the list, if it exists. :rtype: dict NrRrSs rlastzEntityList.lastts !% tyy}3t3rczt|j}|dk(sJdj||jS)zqLike `first`, but first asserts that there is only one entity in the results list. :rtype: dict z3Expecting one entity, but {} entities were returned)lenrr rT)r data_lengths rrJzEntityList.entity|s; $))n aj!V!]!]^i!jjzzrc ddl}|r |j|jS|j j|jS#t$r}td|d}~wwxYw#t$rddlm}||jcYSwxYw)zTransforms the data into a pandas DataFrame :param normalize: Whether or not to normalize any nested objects in the results into distinct columns. :type normalize: bool :rtype: pandas.DataFrame rNz*The 'pandas' package could not be imported)json_normalize)pandas ImportErrorr]rAttributeErrorpandas.io.json DataFrame from_dict)r normalizer^excr]s r to_dataframezEntityList.to_dataframes U   1,,TYY77 ))$))44 UJKQT T U " 19%dii00 1s(A A& A# AA#&!B  B )F)rrrrrr)r/r3r:r<r?__rmul__rErPpropertyrTrWrJrfrrs@rr r s~$ # # > GG=Hm$33445rr N) collectionsrmygeotabrrr rrrls-!U#''U(~5~5r