U
SW* @ s@ d dl Z d dlZd dlmZ d dlZd dlZzd dlmZ W n ek
r\ d dlmZ Y nX d dl Z d dl
T d dl m Z d dl
mZ d dl
mZ d dlmZ e jd d kreZd d
lmZ G d d d eZG d
d deZG dd deZG dd deZG dd deZ G dd deZ G dd de Z G dd de Z dS ) N)Decimal)urlparse)*connect)migrate)SchemaMigrator) Introspector )reducec @ s e Zd Zdd Zdd Zdd Zdd Zd d
Zed d Z d
d Z
dd Z dd Z d'ddZ
dd Zdd Zd(dd Zd d Zd d Zd)d"d#Zd*d%d&ZdS )+DataSetc s | _ t|}|jdd _t| _ j t j _ jj dd _
t j _ G fdddt
}| _ _ _d S )N T) skip_invalidc s e Zd ZG fdddZdS )z#DataSet.__init__.<locals>.BaseModelc s e Zd Z jZdS )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
t tdS N)csvjson) CSVExporter JSONExporterr r r r r 0 s zDataSet.get_export_formatsc C s
t tdS r& ) CSVImporter JSONImporterr r r r r" 5 s zDataSet.get_import_formatsc C s t | || j|S N)Tabler getr tabler r r __getitem__: s zDataSet.__getitem__c C s
| j S r- )r Z
get_tablesr r r r tables= s zDataSet.tablesc C s
|| j kS r- )r3 r0 r r r __contains__A s zDataSet.__contains__c C s | j d S r- )r r r r r r r D s zDataSet.connectc C s | j d S r- )r closer r r r r5 G s z
DataSet.closeNc C sL |r | j | }|jjdd}nd }| jjddd |D d}| j | d S )NT)Zbackrefsc S s g | ] }|j jqS r )_metadb_table).0Zrelatedr r r
<listcomp>R s z(DataSet.update_cache.<locals>.<listcomp>)r Z table_names)r r6 Zrelated_modelsr r update)r r1 model_classZ dependenciesupdatedr r r update_cacheJ s
zDataSet.update_cachec C s | | S r- r r r r r __enter__U s zDataSet.__enter__c C s | j s| d S r- )r Z is_closedr5 )r exc_typeZexc_valZexc_tbr r r __exit__Y s
zDataSet.__exit__Tc C s | j |||S r- )r Z execute_sql)r ZsqlZparamsZcommitr r r query] s z
DataSet.queryc C s& | j dkr| j S | j S d S )Nr )r Ztransaction_depth transactionZ savepointr r r r rB ` s
zDataSet.transactionc C sN |r|rt d|s |s t d||krJdt| }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 sL | |||| j |r t|d}| j| |}|j|f| |rH| d S )Nw)rJ r! openexportr5 )r rA rI rG rH kwargsZexporterr r r freezer s
zDataSet.freezeFc K sR | |||| j |r t|d}| j| | | |}|j|f|}|rN| |S )Nr)rJ r! rL r# loadr5 ) r r1 rI rG rH strictrN Zimportercountr r r thaw~ s
z DataSet.thaw)N)NT)r' NN)r' NNF)r r r r$ r% r r" r2 propertyr3 r4 r r5 r= r> r@ rA rB rJ rO rT r r r r r s(
r c @ s e Zd Zdd Zedd Zdd Zdd Zd d
Zd d Z d+ddZ
dd Z edd Z dd Z
dd Zd,ddZd-dd Zd d Zd d Zd!d" Zd#d$ Zd%d& Zd'd( Zd)d* ZdS ).r. c C s4 || _ || _|d kr0| }| || j j|< d S r- )datasetname
_create_modelZ create_tabler )r rV rW r; r r r r$ s zTable.__init__c C s | j j| j S r- )rV r rW r r r r r; s zTable.model_classc C s
d| j S )Nz <Table: %s>rW r r r r r% s zTable.__repr__c C s | S r- )findrS r r r r __len__ s z
Table.__len__c C s t | S r- )iterrZ iteratorr r r r __iter__ s zTable.__iter__c s. G fddd}t t j jjfd|iS )Nc s e Zd Z jZdS )z!Table._create_model.<locals>.MetaN)r r r rW r7 r r r r r s r )typestrrW rV r )r r r r r rX s zTable._create_modelFc C s | j jj| j||d d S )N)unique)rV r create_indexr; )r columnsra r r r rb s
zTable.create_indexc C sf t |trtS t |tjtjfr$tS |dks4|dkr8tS t |trFtS t |t rTt
S t |t rbt S tS )NTF)
isinstance
basestringZ TextFielddatetimedateZ
DateTimeFieldZ BooleanFieldintZ IntegerFieldfloatZ
FloatFieldr Z DecimalField)r valuer r r _guess_field_type s
zTable._guess_field_typec C s dd | j jjD S )Nc S s g | ]
}|j qS r rY )r8 fr r r r9 s z!Table.columns.<locals>.<listcomp>)r; r6
sorted_fieldsr r r r rc s z
Table.columnsc C s t |t | jjj }|r~g }|D ]D}| || }|dd}|| jj| j || |
| j| q"t | | j | j d S )NT)Znull)
setr; r6 fieldsrk appendrV r Z
add_columnrW Z add_to_classr r= )r dataZnew_keysZ
operationskeyZ field_classfieldr r r _migrate_new_columns s
zTable._migrate_new_columnsc K s | | | jjf | S r- )rt r; insertexecute)r rq r r r ru s
z Table.insertNc s8 |pt j}|r4 fdd| D }|t||}|S )Nc s" g | ]\}} j jj| |kqS r )r; r6 ro )r8 columnrj r r r r9 s z&Table._apply_where.<locals>.<listcomp>)operatorand_itemswherer )r rA filters conjunctionZ expressionsr r r _apply_where s
zTable._apply_wherec K sF | | i }|r*|D ]}||||< q| | jjf ||| S r- )rt popr~ r; r: rv )r rc r} rq r| rw r r r r: s
z Table.updatec K s | | j |S r- )r~ r; Zselectr rA r r r _query s z Table._queryc K s | j f | S r- )r Zdictsr r r r rZ s z
Table.findc K s2 z| j f | W S | jjk
r, Y d S X d S r- )rZ r/ r; Z DoesNotExistr r r r find_one s zTable.find_onec C s | S r- )rZ r r r r all s z Table.allc K s | | j | S r- )r~ r; deleterv r r r r r s z Table.deletec O s | j j| f||S r- )rV rO r r argsrN r r r rO s z Table.freezec O s | j j| jf||S r- )rV rT rW r r r r rT s z
Table.thaw)F)N)NN)r r r r$ rU r; r% r[ r^ rX rb rk rc rt ru r~ r: r rZ r r r rO rT r r r r r. s,
r. c @ s e Zd Zdd Zdd ZdS )Exporterc C s
|| _ d S r- )rA r r r r r$ s zExporter.__init__c C s t d S r- NotImplementedErrorr rH r r r rM s zExporter.exportN)r r r r$ rM r r r r r s r c @ s e Zd Zedd Zdd ZdS )r* c C s@ t | tjtjtjfr | S t | tr0t| S td| d S )Nz Unable to serialize %r as JSON.)rd rf rg timeZ isoformatr r` TypeError)or r r default s
zJSONExporter.defaultc K s$ t jt| j|fdtji| d S )Nr )r( dumplistrA r* r )r rH rN r r r rM s zJSONExporter.exportN)r r r staticmethodr rM r r r r r*
s
r* c @ s e Zd ZdddZdS )r) Tc K sT t j|f|}|r6t| jdr6|dd | jjD | j D ]}|| q@d S )N_selectc S s g | ]
}|j qS r rY )r8 rs r r r r9 " s z&CSVExporter.export.<locals>.<listcomp>)r' writerhasattrrA Zwriterowr Ztuples)r rH headerrN r rowr r r rM s
zCSVExporter.exportN)T)r r r rM r r r r r) s r) c @ s e Zd ZdddZdd ZdS )ImporterFc C s2 || _ || _| j j}|jj| _| j|jj d S r- )r1 rR r; r6 rc r: ro )r r1 rR modelr r r r$ ( s
zImporter.__init__c C s t d S r- r r r r r rQ 0 s z
Importer.loadN)F)r r r r$ rQ r r r r r ' s
r c @ s e Zd Zdd ZdS )r, c K s| t j|f|}d}|D ]`}| jrXi }|D ],}| j|}|d k r(||| ||j< q(n|}|r| jjf | |d7 }q|S Nr r
) r( rQ rR rc r/ python_valuerW r1 ru ) r rH rN rq rS r objrr rs r r r rQ 5 s
zJSONImporter.loadNr r r rQ r r r r r, 4 s r, c @ s e Zd ZdddZdS )r+ Tc
K s d}t j|f|}|rz t|}W n tk
r: | Y S X | jrxg }t|D ]&\}} | | jkrN||| j| f qNqtt|}ntt| j j
j }|s|S |D ]T}
i } |D ]0\}} | jr| |
| | | j
< q|
| | | < q| jjf | |d7 }q|S r )r' readernext
StopIterationrR enumeraterc rp r r r6 rm r rW r1 ru )
r rH r rN rS r Z header_keysZ
header_fieldsidxrr r r rs r r r rQ K s2
zCSVImporter.loadN)Tr r r r r r+ J s r+ ) r' rf Zdecimalr r( rx r ImportErrorZ urllib.parsesysZpeeweeZplayhouse.db_urlr Zplayhouse.migrater r Zplayhouse.reflectionr version_infor` re functoolsr objectr r. r r* r) r r, r+ r r r r <module> s2 vx