R6W* @ s d d l Z d d l Z d d l m Z d d l Z d d l Z y d d l m Z Wn" e k
rx d d l m Z Yn Xd d l Z d d l
Td d l m Z d d l
m Z d d l
m Z d d l m Z e j d d k r e Z d d
l m Z Gd d d e Z Gd
d d e Z Gd d d e Z Gd d d e Z Gd d d e Z Gd d d e Z Gd d d e Z Gd d d e Z d S) N)Decimal)urlparse)*)connect)migrate)SchemaMigrator) Introspector )reducec @ s e Z d Z d d Z d d Z d d Z d d Z d d
Z e d d Z d
d Z
d d Z d d Z d d d Z
d d Z d d Z d d d d Z d d Z d d Z d! d d d" d# Z d! d d d$ d% d& Z d S)'DataSetc s | _ t | } | j d d _ t | _ j j t j j _ j j d d _
t j j _ G f d d d t
} | _ j _ j _ d S)N skip_invalidTc s) e Z d Z G f d d d Z d S)z#DataSet.__init__.<locals>.BaseModelc s e Z d Z j Z d S)z(DataSet.__init__.<locals>.BaseModel.MetaN)__name__
__module__ __qualname__ _databaseZdatabase )selfr /dataset.pyMeta' s r N)r r r r r )r r r BaseModel& s r )Z_urlr path_database_pathr r r Z
from_database
_introspectorgenerate_models_modelsr _migratorZModel _base_modelget_export_formats_export_formatsget_import_formats_import_formats)r ZurlZ parse_resultr r )r r __init__ s
zDataSet.__init__c C s d | j S)Nz
<DataSet: %s>)r )r r r r __repr__- s zDataSet.__repr__c C s d t d t i S)Ncsvjson) CSVExporter JSONExporter)r r r r r 0 s zDataSet.get_export_formatsc C s d t d t i S)Nr$ r% ) CSVImporter JSONImporter)r r r r r 5 s zDataSet.get_import_formatsc C s t | | | j j | S)N)Tabler get)r tabler r r __getitem__: s zDataSet.__getitem__c C s
| j j S)N)r Z
get_tables)r r r r tables= s zDataSet.tablesc C s
| | j k S)N)r. )r r, r r r __contains__A s zDataSet.__contains__c C s | j j d S)N)r r )r r r r r D s zDataSet.connectc C s | j j d S)N)r close)r r r r r0 G s z
DataSet.closeNc C sm | r+ | j | } | j j d d } n d } | j j d d d d d | D } | j j | d S)NZbackrefsTr
Z table_namesc S s g | ] } | j j q Sr )_metadb_table).0Zrelatedr r r
<listcomp>R s z(DataSet.update_cache.<locals>.<listcomp>)r r1 Zrelated_modelsr r update)r r, model_classZ dependenciesupdatedr r r update_cacheJ s
zDataSet.update_cachec C s | j | S)N)r )r r r r __enter__U s
zDataSet.__enter__c C s | j j s | j d S)N)r Z is_closedr0 )r exc_typeexc_valexc_tbr r r __exit__Y s zDataSet.__exit__Tc C s | j j | | | S)N)r Z execute_sql)r ZsqlZparamsZcommitr r r query] s z
DataSet.queryc C s3 | j j d k r" | j j S| j j Sd S)Nr )r Ztransaction_depth transactionZ savepoint)r r r r r? ` s
zDataSet.transactionc C ss | r | r t d | r2 | r2 t d | | k ro d j t | j } t d | | f d S)NzMfile is over-specified. Please use either filename or file_obj, but not both.z1A filename or file-like object must be specified.z, z'Unsupported format "%s". Use one of %s.)
ValueErrorjoinsortedkeys)r filenamefile_objformatZ format_dictZ
valid_formatsr r r _check_argumentsf s zDataSet._check_argumentsr$ c K se | j | | | | j | r. t | d } | j | | } | j | | | ra | j d S)Nw)rG r openexportr0 )r r> rF rD rE kwargsZexporterr r r freezer s zDataSet.freezeFc K sn | j | | | | j | r. t | d } | j | | | | } | j | | } | rj | j | S)Nr)rG r rI r! loadr0 ) r r, rF rD rE strictrK Zimportercountr r r thaw~ s
z DataSet.thaw)r r r r" r# r r r- propertyr. r/ r r0 r8 r9 r= r> r? rG rL rQ r r r r r s$ r c @ s e Z d Z d d Z e d d Z d d Z d d Z d d
Z d d Z d
d d Z
d d Z e d d Z d d Z
d d Z d d d Z d d d d Z d d Z d d Z d! d" Z d# d$ Z d% d& Z d' d( Z d) d* Z d S)+r* c C sH | | _ | | _ | d k rD | j } | j | | j j | <d S)N)datasetname
_create_modelZ create_tabler )r rS rT r6 r r r r" s
zTable.__init__c C s | j j | j S)N)rS r rT )r r r r r6 s zTable.model_classc C s d | j S)Nz <Table: %s>)rT )r r r r r# s zTable.__repr__c C s | j j S)N)findrP )r r r r __len__ s z
Table.__len__c C s t | j j S)N)iterrV iterator)r r r r __iter__ s zTable.__iter__c sA G f d d d } t t j j j f d | i S)Nc s e Z d Z j Z d S)z!Table._create_model.<locals>.MetaN)r r r rT r2 r )r r r r s r )typestrrT rS r )r r r )r r rU s
zTable._create_modelFc C s# | j j j | j | d | d S)Nunique)rS r create_indexr6 )r columnsr] r r r r^ s zTable.create_indexc C s t | t r t St | t j t j f r2 t S| d k sJ | d k rN t St | t ra t St | t rt t
St | t r t St S)NTF)
isinstance
basestringZ TextFielddatetimedateZ
DateTimeFieldZ BooleanFieldintZ IntegerFieldfloatZ
FloatFieldr Z DecimalField)r valuer r r _guess_field_type s zTable._guess_field_typec C s d d | j j j D S)Nc S s g | ] } | j q Sr )rT )r3 fr r r r4 s z!Table.columns.<locals>.<listcomp>)r6 r1
sorted_fields)r r r r r_ s z
Table.columnsc C s t | t | j j j } | r g } xh | D]` } | j | | } | d d } | j | j j j | j | | | j
| j | q2 Wt | | j j | j d S)NZnullT)
setr6 r1 fieldsrg appendrS r Z
add_columnrT Z add_to_classr r8 )r dataZnew_keysZ
operationskeyZ field_classfieldr r r _migrate_new_columns s
zTable._migrate_new_columnsc K s# | j | | j j | j S)N)rp r6 insertexecute)r rm r r r rq s
z Table.insertNc sP | p t j } | rL f d d | j D } | j t | | } | S)Nc s/ g | ]% \ } } j j j | | k q Sr )r6 r1 rk )r3 columnrf )r r r r4 s z&Table._apply_where.<locals>.<listcomp>)operatorand_itemsZwherer
)r r> filters conjunctionZ expressionsr )r r _apply_where s zTable._apply_wherec K sb | j | i } | r= x! | D] } | j | | | <q W| j | j j | | | j S)N)rp popry r6 r5 rr )r r_ rx rm rw rs r r r r5 s
z Table.updatec K s | j | j j | S)N)ry r6 select)r r> r r r _query s z Table._queryc K s | j | j S)N)r| Zdicts)r r> r r r rV s z
Table.findc K s: y | j | j SWn | j j k
r5 d SYn Xd S)N)rV r+ r6 Z DoesNotExist)r r> r r r find_one s zTable.find_onec C s
| j S)N)rV )r r r r all s z Table.allc K s | j | j j | j S)N)ry r6 deleterr )r r> r r r r s z Table.deletec O s | j j | j | | S)N)rS rL r~ )r argsrK r r r rL s z Table.freezec O s | j j | j | | S)N)rS rQ rT )r r rK r r r rQ s z
Table.thaw)r r r r" rR r6 r# rW rZ rU r^ rg r_ rp rq ry r5 r| rV r} r~ r rL rQ r r r r r* s( r* c @ s( e Z d Z d d Z d d Z d S)Exporterc C s
| | _ d S)N)r> )r r> r r r r" s zExporter.__init__c C s
t d S)N)NotImplementedError)r rE r r r rJ s zExporter.exportN)r r r r" rJ r r r r r s r c @ s. e Z d Z e d d Z d d Z d S)r' c C sX t | t j t j t j f r+ | j St | t rD t | St d | d S)Nz Unable to serialize %r as JSON.)r` rb rc timeZ isoformatr r\ TypeError)or r r default s
!
zJSONExporter.defaultc K s) t j t | j | d t j | d S)Nr )r% dumplistr> r' r )r rE rK r r r rJ s
zJSONExporter.exportN)r r r staticmethodr rJ r r r r r'
s r' c @ s e Z d Z d d d Z d S)r& Tc K su t j | | } | rJ t | j d rJ | j d d | j j D x$ | j j D] } | j | qZ Wd S)N_selectc S s g | ] } | j q Sr )rT )r3 ro r r r r4 " s z&CSVExporter.export.<locals>.<listcomp>)r$ writerhasattrr> Zwriterowr Ztuples)r rE headerrK r rowr r r rJ s
zCSVExporter.exportN)r r r rJ r r r r r& s r& c @ s+ e Z d Z d d d Z d d Z d S)ImporterFc C sG | | _ | | _ | j j } | j j | _ | j j | j j d S)N)r, rO r6 r1 r_ r5 rk )r r, rO modelr r r r" ( s
zImporter.__init__c C s
t d S)N)r )r rE r r r rN 0 s z
Importer.loadN)r r r r" rN r r r r r ' s r c @ s e Z d Z d d Z d S)r) c K s t j | | } d } x | D] } | j r i } xO | D]> } | j j | } | d k r; | j | | | | j <q; Wn | } | r | j j | | d 7} q W| S)Nr r ) r% rN rO r_ r+ python_valuerT r, rq ) r rE rK rm rP r objrn ro r r r rN 5 s
!zJSONImporter.loadN)r r r rN r r r r r) 4 s r) c @ s e Z d Z d d d Z d S)r( Tc
K sY d } t j | | } | r y t | } Wn t k
rF | SYn X| j r g } xX t | D]5 \ } } | | j k rc | j | | j | f qc Wq t t | } n t t | j j
j } | s | Sxy | D]q }
i } xH | D]@ \ } } | j r%| j |
| | | j
<q |
| | | <q W| j j | | d 7} q W| S)Nr r )r$ readernext
StopIterationrO enumerater_ rl r r r1 ri r rT r, rq )
r rE r rK rP r Z header_keysZ
header_fieldsidxrn r r ro r r r rN K s2
!
zCSVImporter.loadN)r r r rN r r r r r( J s r( ) r$ rb Zdecimalr r% rt r ImportErrorZ urllib.parsesysZpeeweeZplayhouse.db_urlr Zplayhouse.migrater r Zplayhouse.reflectionr version_infor\ ra functoolsr
objectr r* r r' r& r r) r( r r r r <module> s2
vx