ó
|
zÙPc @ s d d l Z d d l m Z d d l Z d d l Z d d l m Z d d l m Z d d l m
|
Z
|
d Z d e f d YZ d e f d
|
YZ d e f d YZ d Z d d d YZ d Z d Z d Z d Z e d k re n d S( iÿÿÿÿN( t SONManipulator( t wraps( t sleep( t
|
CommFuncExc C sÛ y´ d } | d | } | j d i | d 6d i i | d 6d 6d t } | r] | d } nV | j d i | d 6d i i | d 6d 6d t d
|
t } | r© | d } n
|
t d f SWn t k
|
rÐ } t d f SXt | f S( Ni s %s_seqt queryt _idt updatet seqs $inct news $sett upsert( t find_and_modifyt Truet Falset Nonet Exception( t dbt collectionNamet fieldNamet feedt incrementt resultt
|
collectiont resultObjt e( ( s6 .\server\db\PyMongoDataServer\DBCommon\DBController.pyR s 3 0 t ObjectIdRemoverc B s e Z d Z RS( c C s d | k r | d =n | S( NR ( ( t selft sonR ( ( s6 .\server\db\PyMongoDataServer\DBCommon\DBController.pyt transform_outgoing% s
|
( t __name__t
|
__module__R ( ( ( s6 .\server\db\PyMongoDataServer\DBCommon\DBController.pyR $ s t EncodeStringManipulatorc B s# e Z d Z d Z d Z RS( c C s | | _ d S( N( t encoding( R R ( ( s6 .\server\db\PyMongoDataServer\DBCommon\DBController.pyt __init__+ s c si f d f d f d } t | t rL | St | t re | | S| S( Nc sw t | t r | St | t rE g | D] } | ^ q/ St | t rs t j j | \ } } | S| S( N( t
|
isinstancet dictt listt
|
basestringR t EncodingToUnicodeR ( t valuet vR ( R t transform_valuet transform_dict( s6 .\server\db\PyMongoDataServer\DBCommon\DBController.pyR( / s
|
c s1 x* | j D] \ } } | | | <q W| S( N( t items( t objectt keyR& ( R( ( s6 .\server\db\PyMongoDataServer\DBCommon\DBController.pyR) 9 s c s x | D] } | q W| S( N( ( t containert item( R) ( s6 .\server\db\PyMongoDataServer\DBCommon\DBController.pyt transform_list> s ( R! R" R# ( R R R R/ ( ( R R( R) s6 .\server\db\PyMongoDataServer\DBCommon\DBController.pyt transform_incoming- s
|
|
|
c si f d f d f d } t | t rL | St | t re | | S| S( Nc sw t | t r | St | t rE g | D] } | ^ q/ St | t rs t j j | \ } } | S| S( N( R! R" R# R$ R t UnicodeToEncodingR ( R& R' R ( R R( R) ( s6 .\server\db\PyMongoDataServer\DBCommon\DBController.pyR( K s
|
c s1 x* | j D] \ } } | | | <q W| S( N( R* ( R+ R, R& ( R( ( s6 .\server\db\PyMongoDataServer\DBCommon\DBController.pyR) U s c s x | D] } | q W| S( N( ( R- R. ( R) ( s6 .\server\db\PyMongoDataServer\DBCommon\DBController.pyR/ Z s ( R! R" R# ( R R R R/ ( ( R R( R) s6 .\server\db\PyMongoDataServer\DBCommon\DBController.pyR I s
|
|
|
( R R R R0 R ( ( ( s6 .\server\db\PyMongoDataServer\DBCommon\DBController.pyR * s t Base64StringManipulatorc B s e Z d Z d Z RS( c sf f d f d f d } t | t rI | St | t rb | | S| S( Nc se t | t r | St | t rE g | D] } | ^ q/ St | t ra t j | S| S( N( R! R" R# R$ t base64t b64encode( R& R' ( R( R) ( s6 .\server\db\PyMongoDataServer\DBCommon\DBController.pyR( i s
|
c s1 x* | j D] \ } } | | | <q W| S( N( R* ( R+ R, R& ( R( ( s6 .\server\db\PyMongoDataServer\DBCommon\DBController.pyR) r s c s x | D] } | q W| S( N( ( R- R. ( R) ( s6 .\server\db\PyMongoDataServer\DBCommon\DBController.pyR/ w s ( R! R" R# ( R R R R/ ( ( R( R) s6 .\server\db\PyMongoDataServer\DBCommon\DBController.pyR0 g s
|
|
c sf f d f d f d } t | t rI | St | t rb | | S| S( Nc se t | t r | St | t rE g | D] } | ^ q/ St | t ra t j | S| S( N( R! R" R# R$ R3 t b64decode( R& R' ( R( R) ( s6 .\server\db\PyMongoDataServer\DBCommon\DBController.pyR( s
|
c s1 x* | j D] \ } } | | | <q W| S( N( R* ( R+ R, R& ( R( ( s6 .\server\db\PyMongoDataServer\DBCommon\DBController.pyR) s c s x | D] } | q W| S( N( ( R- R. ( R) ( s6 .\server\db\PyMongoDataServer\DBCommon\DBController.pyR/ s ( R! R" R# ( R R R R/ ( ( R( R) s6 .\server\db\PyMongoDataServer\DBCommon\DBController.pyR s
|
|
( R R R0 R ( ( ( s6 .\server\db\PyMongoDataServer\DBCommon\DBController.pyR2 e s c s t f d } | S( Nc sy d } d } d } x` t rt y | | SWq t j j k
|
rp } | d 7} t | | | k rq | qq q Xq Wd S( Ni
|
g¹?i i ( R t pymongot errorst AutoReconnectR ( t argst kwdst MAX_RECONNECTt RECONNECT_INTERVALt failCntR ( t func( s6 .\server\db\PyMongoDataServer\DBCommon\DBController.pyt wrapper s
|
|
( R ( R> R? ( ( R> s6 .\server\db\PyMongoDataServer\DBCommon\DBController.pyt reconnect_decorator s t DBControllerc B s e Z d Z d Z d Z d Z d d Z e d d d d d Z e e
|
d Z e e e
|
e d Z e e
|
d Z e d e
|
d
|
Z e d Z e d Z RS( c C s | | _ | | _ | | _ | | _ | | _ t | _ d | _ d | _ d | _
|
| d k rl t | _ n t | | _ | j d S( NR3 ( t hostt portt dbNamet usert pwdR t connectedR t conR t lastErrorR2 t
|
translatorR t
|
initialize( R RB RC RD RE RF R ( ( s6 .\server\db\PyMongoDataServer\DBCommon\DBController.pyR ² s c C sj | j sf | j | j | j s% t S| j | j | j | j } | j rb | j j
|
t n | St S( N( RG t doConnectRB RC R t doAuthenticationRD RE RF R t add_son_manipulatorR R ( R t
|
authResult( ( s6 .\server\db\PyMongoDataServer\DBCommon\DBController.pyRK Ã s c C s y t j | | | _ Wnd t k
|
r1 } nO t j j k
|
rS } | | _ t St k
|
ro } | | _ t Sd | _ t SXt | _
|
t S( Ns Unknown exception occur!( R6 t
|
ConnectionRH t TypeErrorR7 t ConnectionFailureRI R R R RG ( R t ipRC t typeErrort failureR ( ( s6 .\server\db\PyMongoDataServer\DBCommon\DBController.pyRL Í s c C s¢ | j s | j r! d | _ t S| j | | _ | j d } y | j | | SWnI t k
|
rq } | | _ t St k
|
r } | | _ t Sd | _ t SXd S( Ns Not connected yet!t admins Unknown exception occur!( RG RH RI R R t authenticateRQ R ( R RD RE RF t authDBRT R ( ( s6 .\server\db\PyMongoDataServer\DBCommon\DBController.pyRM Þ s c C sP | j | | | d \ } } | s. t d f Sx | D] } t | f SWt d f S( Ni ( t findR R R ( R t colNamet spect filterR t recListt rec( ( s6 .\server\db\PyMongoDataServer\DBCommon\DBController.pyt find_oneñ s
|
i c C sE | j s" | j s" t g f Sn t } g } | j | } | j r\ | j j | d } n y_ | j | | d | d | } | j r¤ | j j t | d } n t | } t
|
| f SWn t k
|
rà }
|
|
|
| _ | | f St j j k
|
r} | | _ | | f St k
|
r*} | | _ | | f Sd | _ | | f SXd S( Nt limitt sorts Unknown exception occur!( RG RK R R RJ R0 R RY R R# R RQ RI R6 R7 t OperationFailureR ( R RZ R[ R\ t maxCntt sortByR t resultDictListt colt resultCollectionRT t errR ( ( s6 .\server\db\PyMongoDataServer\DBCommon\DBController.pyRY ù s4
|
|
|
c C s· | j s | j s t Sn | j | } | j rJ | j j | d } n y | j | d | WnO t j j
|
k
|
r } | | _ t St k
|
r¢ } | | _ t Sd | _ t SXt S( Nt safes Unknown exception occur!( RG RK R R RJ R0 R t insertR6 R7 Rb RI R R ( R RZ t doc_or_docst isSafeRf Rh R ( ( s6 .\server\db\PyMongoDataServer\DBCommon\DBController.pyRj s$ c C s÷ | j s | j s t Sn | j | } | j r_ | j j | d } | j j | d } n y& | j | | d | d | d | Wnk t k
|
r¤ } | | _ t St
|
j j k
|
rÆ } | | _ t St k
|
râ }
|
|
|
| _ t Sd | _ t SXt S( NR Ri t multis Unknown exception occur!( RG RK R R RJ R0 R R RQ RI R6 R7 Rb R R ( R RZ R[ t doct isUpsertRl t isMultiRf RT Rh R ( ( s6 .\server\db\PyMongoDataServer\DBCommon\DBController.pyR . s, & c C sÓ | j s | j s t Sn | j | } | j rJ | j j | d } n y | j | d | Wnk t k
|
r } | | _ t St
|
j j k
|
r¢ } | | _ t St k
|
r¾ } | | _ t Sd | _ t SXt S( NRi s Unknown exception occur!( RG RK R R RJ R0 R t saveRQ RI R6 R7 Rb R R ( R RZ Rn Rl Rf RT Rh R ( ( s6 .\server\db\PyMongoDataServer\DBCommon\DBController.pyRq I s* c C s· | j s | j s t Sn | j | } | j rJ | j j | d } n y | j | d | WnO t j j
|
k
|
r } | | _ t St k
|
r¢ } | | _ t Sd | _ t SXt S( NRi s Unknown exception occur!( RG RK R R RJ R0 R t removeR6 R7 Rb RI R R ( R RZ R[ Rl Rf Rh R ( ( s6 .\server\db\PyMongoDataServer\DBCommon\DBController.pyRr b s$ c C s | j s | j s t Sn | j | } y | j WnI t k
|
rV } | | _ t St k
|
rr } | | _ t Sd | _ t SXt S( Ns Unknown exception occur!( RG RK R R t dropRQ RI R R ( R RZ Rf RT R ( ( s6 .\server\db\PyMongoDataServer\DBCommon\DBController.pyRs x s c C s | j s" | j s" t d f Sn | j | } y | j } Wna t j j k
|
rj } | | _ t d f St k
|
r } | | _ t d f Sd | _ t d f SXt
|
| f S( Ni s Unknown exception occur!( RG RK R R t countR6 R7 Rb RI R R ( R RZ Rf t cntRh R ( ( s6 .\server\db\PyMongoDataServer\DBCommon\DBController.pyRt s
|
|
N( R R R RK RL RM R R_ R@ RY R Rj R R Rq Rr Rs Rt ( ( ( s6 .\server\db\PyMongoDataServer\DBCommon\DBController.pyRA ± s"
|
c C sÔ t j } | j } | j d d s0 d GHd Sd } d } | d } | j | | j d | t | | | d d \ } } | r | d k s t t | | | d d \ } } | rÊ | d k sÐ t d S( Nt sas auth failed!t
|
tagSeqTestt IDt tests %s_seqi i ( R6 RP RV RW t drop_collectionR t AssertionError( RH R RZ R R Rx ( ( s6 .\server\db\PyMongoDataServer\DBCommon\DBController.pyt test_seq s
|
c C sÒ t } g } | j | d } | | k s3 t | j | d } | | k sW t i d d 6g } | j | d } | | k s t | j | d } | | k s¯ t i d d 6g } | j | d } | rö | i t j d d 6g k sü t | j | d } | r | | k s&t i i d d 6i d d 6g d 6g } | j | d } | r| i i t j d d 6i t j d d 6g d 6g k s¤t | j | d } | rÈ| | k sÎt d S( Ni t at bt c( R2 R0 R R{ R R3 R4 ( RJ R R ( ( s6 .\server\db\PyMongoDataServer\DBCommon\DBController.pyt test_StringManipulator± s* +$Hc C sD d } t d d d d d } | j | } | s9 t | j | \ } } | r` | d k sf t i d d 6} | j | | } | s t | j | \ } } | r¾ t | d k sÄ t | d } | | k sà t i d d 6} i d d 6} i | d
|
6} | j | | | } | s(t | j | \ } } | rUt | d k s[t | d } | d =| | k s~t | j | } | st | j | \ } } | rÀ| g k sÆt i d d 6}
|
| j | |
|
} | sñt | j | \ } } | r t | d k s$t | d } | |
|
k s@t d S( Nt tagTestControllert localhostii Ry t 1i i R} i s $setR i R~ (
|
RA Rs R{ Rt Rj RY t lenR Rr Rq ( t testColNamet dbControllerR Ru Rn t recsR^ R[ t updateDoct updateDocWithModifiert saveDoc( ( s6 .\server\db\PyMongoDataServer\DBCommon\DBController.pyt test_DBControllerÌ sD
|
|
|
c C s t t t d GHd S( Ns test ok!( R| R R ( ( ( s6 .\server\db\PyMongoDataServer\DBCommon\DBController.pyRy ú s t __main__( ( R6 t pymongo.son_manipulatorR R3 t tracebackt functoolsR t timeR t CommonR R R R R2 R@ RA R| R R Ry R ( ( ( s6 .\server\db\PyMongoDataServer\DBCommon\DBController.pyt <module> s$ ;: ï .
|
