models.py 271 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245124612471248124912501251125212531254125512561257125812591260126112621263126412651266126712681269127012711272127312741275127612771278127912801281128212831284128512861287128812891290129112921293129412951296129712981299130013011302130313041305130613071308130913101311131213131314131513161317131813191320132113221323132413251326132713281329133013311332133313341335133613371338133913401341134213431344134513461347134813491350135113521353135413551356135713581359136013611362136313641365136613671368136913701371137213731374137513761377137813791380138113821383138413851386138713881389139013911392139313941395139613971398139914001401140214031404140514061407140814091410141114121413141414151416141714181419142014211422142314241425142614271428142914301431143214331434143514361437143814391440144114421443144414451446144714481449145014511452145314541455145614571458145914601461146214631464146514661467146814691470147114721473147414751476147714781479148014811482148314841485148614871488148914901491149214931494149514961497149814991500150115021503150415051506150715081509151015111512151315141515151615171518151915201521152215231524152515261527152815291530153115321533153415351536153715381539154015411542154315441545154615471548154915501551155215531554155515561557155815591560156115621563156415651566156715681569157015711572157315741575157615771578157915801581158215831584158515861587158815891590159115921593159415951596159715981599160016011602160316041605160616071608160916101611161216131614161516161617161816191620162116221623162416251626162716281629163016311632163316341635163616371638163916401641164216431644164516461647164816491650165116521653165416551656165716581659166016611662166316641665166616671668166916701671167216731674167516761677167816791680168116821683168416851686168716881689169016911692169316941695169616971698169917001701170217031704170517061707170817091710171117121713171417151716171717181719172017211722172317241725172617271728172917301731173217331734173517361737173817391740174117421743174417451746174717481749175017511752175317541755175617571758175917601761176217631764176517661767176817691770177117721773177417751776177717781779178017811782178317841785178617871788178917901791179217931794179517961797179817991800180118021803180418051806180718081809181018111812181318141815181618171818181918201821182218231824182518261827182818291830183118321833183418351836183718381839184018411842184318441845184618471848184918501851185218531854185518561857185818591860186118621863186418651866186718681869187018711872187318741875187618771878187918801881188218831884188518861887188818891890189118921893189418951896189718981899190019011902190319041905190619071908190919101911191219131914191519161917191819191920192119221923192419251926192719281929193019311932193319341935193619371938193919401941194219431944194519461947194819491950195119521953195419551956195719581959196019611962196319641965196619671968196919701971197219731974197519761977197819791980198119821983198419851986198719881989199019911992199319941995199619971998199920002001200220032004200520062007200820092010201120122013201420152016201720182019202020212022202320242025202620272028202920302031203220332034203520362037203820392040204120422043204420452046204720482049205020512052205320542055205620572058205920602061206220632064206520662067206820692070207120722073207420752076207720782079208020812082208320842085208620872088208920902091209220932094209520962097209820992100210121022103210421052106210721082109211021112112211321142115211621172118211921202121212221232124212521262127212821292130213121322133213421352136213721382139214021412142214321442145214621472148214921502151215221532154215521562157215821592160216121622163216421652166216721682169217021712172217321742175217621772178217921802181218221832184218521862187218821892190219121922193219421952196219721982199220022012202220322042205220622072208220922102211221222132214221522162217221822192220222122222223222422252226222722282229223022312232223322342235223622372238223922402241224222432244224522462247224822492250225122522253225422552256225722582259226022612262226322642265226622672268226922702271227222732274227522762277227822792280228122822283228422852286228722882289229022912292229322942295229622972298229923002301230223032304230523062307230823092310231123122313231423152316231723182319232023212322232323242325232623272328232923302331233223332334233523362337233823392340234123422343234423452346234723482349235023512352235323542355235623572358235923602361236223632364236523662367236823692370237123722373237423752376237723782379238023812382238323842385238623872388238923902391239223932394239523962397239823992400240124022403240424052406240724082409241024112412241324142415241624172418241924202421242224232424242524262427242824292430243124322433243424352436243724382439244024412442244324442445244624472448244924502451245224532454245524562457245824592460246124622463246424652466246724682469247024712472247324742475247624772478247924802481248224832484248524862487248824892490249124922493249424952496249724982499250025012502250325042505250625072508250925102511251225132514251525162517251825192520252125222523252425252526252725282529253025312532253325342535253625372538253925402541254225432544254525462547254825492550255125522553255425552556255725582559256025612562256325642565256625672568256925702571257225732574257525762577257825792580258125822583258425852586258725882589259025912592259325942595259625972598259926002601260226032604260526062607260826092610261126122613261426152616261726182619262026212622262326242625262626272628262926302631263226332634263526362637263826392640264126422643264426452646264726482649265026512652265326542655265626572658265926602661266226632664266526662667266826692670267126722673267426752676267726782679268026812682268326842685268626872688268926902691269226932694269526962697269826992700270127022703270427052706270727082709271027112712271327142715271627172718271927202721272227232724272527262727272827292730273127322733273427352736273727382739274027412742274327442745274627472748274927502751275227532754275527562757275827592760276127622763276427652766276727682769277027712772277327742775277627772778277927802781278227832784278527862787278827892790279127922793279427952796279727982799280028012802280328042805280628072808280928102811281228132814281528162817281828192820282128222823282428252826282728282829283028312832283328342835283628372838283928402841284228432844284528462847284828492850285128522853285428552856285728582859286028612862286328642865286628672868286928702871287228732874287528762877287828792880288128822883288428852886288728882889289028912892289328942895289628972898289929002901290229032904290529062907290829092910291129122913291429152916291729182919292029212922292329242925292629272928292929302931293229332934293529362937293829392940294129422943294429452946294729482949295029512952295329542955295629572958295929602961296229632964296529662967296829692970297129722973297429752976297729782979298029812982298329842985298629872988298929902991299229932994299529962997299829993000300130023003300430053006300730083009301030113012301330143015301630173018301930203021302230233024302530263027302830293030303130323033303430353036303730383039304030413042304330443045304630473048304930503051305230533054305530563057305830593060306130623063306430653066306730683069307030713072307330743075307630773078307930803081308230833084308530863087308830893090309130923093309430953096309730983099310031013102310331043105310631073108310931103111311231133114311531163117311831193120312131223123312431253126312731283129313031313132313331343135313631373138313931403141314231433144314531463147314831493150315131523153315431553156315731583159316031613162316331643165316631673168316931703171317231733174317531763177317831793180318131823183318431853186318731883189319031913192319331943195319631973198319932003201320232033204320532063207320832093210321132123213321432153216321732183219322032213222322332243225322632273228322932303231323232333234323532363237323832393240324132423243324432453246324732483249325032513252325332543255325632573258325932603261326232633264326532663267326832693270327132723273327432753276327732783279328032813282328332843285328632873288328932903291329232933294329532963297329832993300330133023303330433053306330733083309331033113312331333143315331633173318331933203321332233233324332533263327332833293330333133323333333433353336333733383339334033413342334333443345334633473348334933503351335233533354335533563357335833593360336133623363336433653366336733683369337033713372337333743375337633773378337933803381338233833384338533863387338833893390339133923393339433953396339733983399340034013402340334043405340634073408340934103411341234133414341534163417341834193420342134223423342434253426342734283429343034313432343334343435343634373438343934403441344234433444344534463447344834493450345134523453345434553456345734583459346034613462346334643465346634673468346934703471347234733474347534763477347834793480348134823483348434853486348734883489349034913492349334943495349634973498349935003501350235033504350535063507350835093510351135123513351435153516351735183519352035213522352335243525352635273528352935303531353235333534353535363537353835393540354135423543354435453546354735483549355035513552355335543555355635573558355935603561356235633564356535663567356835693570357135723573357435753576357735783579358035813582358335843585358635873588358935903591359235933594359535963597359835993600360136023603360436053606360736083609361036113612361336143615361636173618361936203621362236233624362536263627362836293630363136323633363436353636363736383639364036413642364336443645364636473648364936503651365236533654365536563657365836593660366136623663366436653666366736683669367036713672367336743675367636773678367936803681368236833684368536863687368836893690369136923693369436953696369736983699370037013702370337043705370637073708370937103711371237133714371537163717371837193720372137223723372437253726372737283729373037313732373337343735373637373738373937403741374237433744374537463747374837493750375137523753375437553756375737583759376037613762376337643765376637673768376937703771377237733774377537763777377837793780378137823783378437853786378737883789379037913792379337943795379637973798379938003801380238033804380538063807380838093810381138123813381438153816381738183819382038213822382338243825382638273828382938303831383238333834383538363837383838393840384138423843384438453846384738483849385038513852385338543855385638573858385938603861386238633864386538663867386838693870387138723873387438753876387738783879388038813882388338843885388638873888388938903891389238933894389538963897389838993900390139023903390439053906390739083909391039113912391339143915391639173918391939203921392239233924392539263927392839293930393139323933393439353936393739383939394039413942394339443945394639473948394939503951395239533954395539563957395839593960396139623963396439653966396739683969397039713972397339743975397639773978397939803981398239833984398539863987398839893990399139923993399439953996399739983999400040014002400340044005400640074008400940104011401240134014401540164017401840194020402140224023402440254026402740284029403040314032403340344035403640374038403940404041404240434044404540464047404840494050405140524053405440554056405740584059406040614062406340644065406640674068406940704071407240734074407540764077407840794080408140824083408440854086408740884089409040914092409340944095409640974098409941004101410241034104410541064107410841094110411141124113411441154116411741184119412041214122412341244125412641274128412941304131413241334134413541364137413841394140414141424143414441454146414741484149415041514152415341544155415641574158415941604161416241634164416541664167416841694170417141724173417441754176417741784179418041814182418341844185418641874188418941904191419241934194419541964197419841994200420142024203420442054206420742084209421042114212421342144215421642174218421942204221422242234224422542264227422842294230423142324233423442354236423742384239424042414242424342444245424642474248424942504251425242534254425542564257425842594260426142624263426442654266426742684269427042714272427342744275427642774278427942804281428242834284428542864287428842894290429142924293429442954296429742984299430043014302430343044305430643074308430943104311431243134314431543164317431843194320432143224323432443254326432743284329433043314332433343344335433643374338433943404341434243434344434543464347434843494350435143524353435443554356435743584359436043614362436343644365436643674368436943704371437243734374437543764377437843794380438143824383438443854386438743884389439043914392439343944395439643974398439944004401440244034404440544064407440844094410441144124413441444154416441744184419442044214422442344244425442644274428442944304431443244334434443544364437443844394440444144424443444444454446444744484449445044514452445344544455445644574458445944604461446244634464446544664467446844694470447144724473447444754476447744784479448044814482448344844485448644874488448944904491449244934494449544964497449844994500450145024503450445054506450745084509451045114512451345144515451645174518451945204521452245234524452545264527452845294530453145324533453445354536453745384539454045414542454345444545454645474548454945504551455245534554455545564557455845594560456145624563456445654566456745684569457045714572457345744575457645774578457945804581458245834584458545864587458845894590459145924593459445954596459745984599460046014602460346044605460646074608460946104611461246134614461546164617461846194620462146224623462446254626462746284629463046314632463346344635463646374638463946404641464246434644464546464647464846494650465146524653465446554656465746584659466046614662466346644665466646674668466946704671467246734674467546764677467846794680468146824683468446854686468746884689469046914692469346944695469646974698469947004701470247034704470547064707470847094710471147124713471447154716471747184719472047214722472347244725472647274728472947304731473247334734473547364737473847394740474147424743474447454746474747484749475047514752475347544755475647574758475947604761476247634764476547664767476847694770477147724773477447754776477747784779478047814782478347844785478647874788478947904791479247934794479547964797479847994800480148024803480448054806480748084809481048114812481348144815481648174818481948204821482248234824482548264827482848294830483148324833483448354836483748384839484048414842484348444845484648474848484948504851485248534854485548564857485848594860486148624863486448654866486748684869487048714872487348744875487648774878487948804881488248834884488548864887488848894890489148924893489448954896489748984899490049014902490349044905490649074908490949104911491249134914491549164917491849194920492149224923492449254926492749284929493049314932493349344935493649374938493949404941494249434944494549464947494849494950495149524953495449554956495749584959496049614962496349644965496649674968496949704971497249734974497549764977497849794980498149824983498449854986498749884989499049914992499349944995499649974998499950005001500250035004500550065007500850095010501150125013501450155016501750185019502050215022502350245025502650275028502950305031503250335034503550365037503850395040504150425043504450455046504750485049505050515052505350545055505650575058505950605061506250635064506550665067506850695070507150725073507450755076507750785079508050815082508350845085508650875088508950905091509250935094509550965097509850995100510151025103510451055106510751085109511051115112511351145115511651175118511951205121512251235124512551265127512851295130513151325133513451355136513751385139514051415142514351445145514651475148514951505151515251535154515551565157515851595160516151625163516451655166516751685169517051715172517351745175517651775178517951805181518251835184518551865187518851895190519151925193519451955196519751985199520052015202520352045205520652075208520952105211521252135214521552165217521852195220522152225223522452255226522752285229523052315232523352345235523652375238523952405241524252435244524552465247524852495250525152525253525452555256525752585259526052615262526352645265526652675268526952705271527252735274527552765277527852795280528152825283528452855286528752885289529052915292529352945295529652975298529953005301530253035304530553065307530853095310531153125313531453155316531753185319532053215322532353245325532653275328532953305331533253335334533553365337533853395340534153425343534453455346534753485349535053515352535353545355535653575358535953605361536253635364536553665367536853695370537153725373537453755376537753785379538053815382538353845385538653875388538953905391539253935394539553965397539853995400540154025403540454055406540754085409541054115412541354145415541654175418541954205421542254235424542554265427542854295430543154325433543454355436543754385439544054415442544354445445544654475448544954505451545254535454545554565457545854595460546154625463546454655466546754685469547054715472547354745475547654775478547954805481548254835484548554865487548854895490549154925493549454955496549754985499550055015502550355045505550655075508550955105511551255135514551555165517551855195520552155225523552455255526552755285529553055315532553355345535553655375538553955405541554255435544554555465547554855495550555155525553555455555556555755585559556055615562556355645565556655675568556955705571557255735574557555765577557855795580558155825583558455855586558755885589559055915592559355945595559655975598559956005601560256035604560556065607560856095610561156125613561456155616561756185619562056215622562356245625562656275628562956305631563256335634563556365637563856395640564156425643564456455646564756485649565056515652565356545655565656575658565956605661566256635664566556665667566856695670567156725673567456755676567756785679568056815682568356845685568656875688568956905691569256935694569556965697569856995700570157025703570457055706570757085709571057115712571357145715571657175718571957205721572257235724572557265727572857295730573157325733573457355736573757385739574057415742574357445745574657475748574957505751575257535754575557565757575857595760576157625763576457655766576757685769577057715772577357745775577657775778577957805781578257835784578557865787578857895790579157925793579457955796579757985799580058015802580358045805580658075808580958105811581258135814581558165817581858195820582158225823582458255826582758285829583058315832583358345835583658375838583958405841584258435844584558465847584858495850585158525853585458555856585758585859586058615862586358645865586658675868586958705871587258735874587558765877587858795880588158825883588458855886588758885889589058915892589358945895589658975898589959005901590259035904590559065907590859095910591159125913591459155916591759185919592059215922592359245925592659275928592959305931593259335934593559365937593859395940594159425943594459455946594759485949595059515952595359545955595659575958595959605961596259635964596559665967596859695970597159725973597459755976597759785979598059815982598359845985598659875988598959905991599259935994599559965997599859996000600160026003600460056006600760086009601060116012601360146015601660176018601960206021602260236024602560266027602860296030603160326033603460356036603760386039604060416042604360446045604660476048604960506051605260536054605560566057605860596060606160626063606460656066606760686069607060716072607360746075607660776078607960806081608260836084608560866087608860896090609160926093609460956096609760986099610061016102610361046105610661076108610961106111611261136114611561166117611861196120612161226123612461256126612761286129613061316132613361346135613661376138613961406141614261436144614561466147614861496150615161526153615461556156615761586159616061616162616361646165616661676168616961706171617261736174617561766177617861796180618161826183618461856186618761886189619061916192619361946195619661976198619962006201620262036204620562066207620862096210621162126213621462156216621762186219622062216222622362246225622662276228622962306231623262336234623562366237623862396240624162426243624462456246624762486249625062516252625362546255625662576258625962606261626262636264626562666267626862696270627162726273627462756276627762786279628062816282628362846285628662876288628962906291629262936294629562966297629862996300630163026303630463056306630763086309631063116312631363146315631663176318631963206321632263236324632563266327632863296330633163326333633463356336633763386339634063416342634363446345634663476348634963506351635263536354635563566357635863596360636163626363636463656366636763686369637063716372637363746375637663776378637963806381638263836384638563866387638863896390639163926393639463956396639763986399640064016402640364046405640664076408640964106411641264136414641564166417641864196420642164226423642464256426642764286429643064316432643364346435643664376438643964406441644264436444644564466447644864496450645164526453645464556456645764586459646064616462646364646465646664676468646964706471647264736474647564766477647864796480648164826483648464856486648764886489649064916492649364946495649664976498649965006501650265036504650565066507650865096510651165126513651465156516651765186519652065216522652365246525652665276528652965306531653265336534653565366537653865396540654165426543654465456546654765486549655065516552655365546555655665576558655965606561656265636564656565666567656865696570657165726573657465756576657765786579658065816582658365846585658665876588658965906591659265936594659565966597659865996600660166026603660466056606660766086609661066116612661366146615661666176618661966206621662266236624662566266627662866296630663166326633663466356636663766386639664066416642664366446645664666476648664966506651665266536654665566566657665866596660666166626663666466656666666766686669667066716672667366746675667666776678667966806681668266836684668566866687668866896690669166926693669466956696669766986699670067016702670367046705670667076708670967106711671267136714671567166717671867196720672167226723672467256726672767286729673067316732673367346735673667376738673967406741674267436744674567466747674867496750675167526753675467556756675767586759676067616762676367646765676667676768676967706771677267736774677567766777677867796780678167826783678467856786678767886789679067916792679367946795679667976798679968006801680268036804680568066807680868096810681168126813681468156816681768186819682068216822682368246825682668276828682968306831683268336834683568366837683868396840684168426843684468456846684768486849685068516852685368546855685668576858685968606861686268636864686568666867686868696870687168726873687468756876687768786879688068816882688368846885688668876888688968906891689268936894689568966897689868996900690169026903690469056906690769086909691069116912691369146915691669176918691969206921692269236924692569266927692869296930693169326933693469356936693769386939694069416942694369446945694669476948694969506951695269536954695569566957695869596960696169626963696469656966696769686969697069716972697369746975697669776978697969806981698269836984698569866987698869896990699169926993699469956996699769986999700070017002700370047005700670077008700970107011701270137014701570167017701870197020702170227023702470257026702770287029703070317032703370347035703670377038703970407041704270437044704570467047704870497050705170527053705470557056705770587059706070617062706370647065706670677068706970707071707270737074707570767077707870797080708170827083708470857086708770887089709070917092709370947095709670977098709971007101710271037104710571067107710871097110711171127113711471157116711771187119712071217122712371247125712671277128712971307131713271337134713571367137713871397140714171427143714471457146714771487149715071517152715371547155715671577158715971607161716271637164716571667167716871697170717171727173717471757176717771787179718071817182718371847185718671877188718971907191719271937194719571967197719871997200720172027203720472057206720772087209721072117212721372147215721672177218721972207221722272237224722572267227722872297230723172327233723472357236723772387239724072417242724372447245724672477248724972507251725272537254725572567257725872597260726172627263726472657266726772687269727072717272727372747275727672777278727972807281728272837284728572867287728872897290729172927293729472957296729772987299730073017302730373047305730673077308730973107311731273137314731573167317731873197320732173227323732473257326732773287329733073317332733373347335733673377338733973407341734273437344734573467347734873497350735173527353735473557356735773587359736073617362736373647365736673677368736973707371737273737374737573767377737873797380738173827383738473857386738773887389739073917392739373947395739673977398739974007401740274037404740574067407740874097410741174127413741474157416741774187419742074217422742374247425742674277428742974307431743274337434743574367437743874397440744174427443744474457446744774487449745074517452745374547455745674577458745974607461746274637464746574667467746874697470747174727473747474757476747774787479748074817482748374847485748674877488748974907491749274937494749574967497749874997500750175027503750475057506750775087509751075117512751375147515751675177518751975207521752275237524752575267527752875297530753175327533753475357536753775387539754075417542754375447545754675477548754975507551755275537554755575567557755875597560756175627563756475657566756775687569757075717572757375747575757675777578757975807581758275837584758575867587758875897590759175927593759475957596759775987599760076017602760376047605760676077608760976107611761276137614761576167617761876197620762176227623762476257626762776287629763076317632763376347635763676377638763976407641764276437644764576467647764876497650765176527653765476557656765776587659766076617662766376647665766676677668766976707671767276737674767576767677767876797680768176827683768476857686768776887689769076917692769376947695769676977698769977007701770277037704770577067707770877097710771177127713771477157716771777187719772077217722772377247725772677277728772977307731773277337734773577367737773877397740774177427743774477457746774777487749775077517752775377547755775677577758775977607761776277637764776577667767
  1. # -*- coding: utf-8 -*-
  2. # !/usr/bin/env python
  3. """
  4. web.user.models
  5. ~~~~~~~~~
  6. """
  7. import copy
  8. import datetime
  9. import logging
  10. import random
  11. import string
  12. import threading
  13. import time
  14. import uuid
  15. from collections import Counter, defaultdict
  16. from arrow import Arrow
  17. from bson.objectid import ObjectId
  18. from dateutil import relativedelta
  19. from django.conf import settings
  20. from django.utils.functional import cached_property
  21. from mongoengine import Q
  22. from mongoengine.document import EmbeddedDocument
  23. from mongoengine.errors import NotUniqueError, DoesNotExist
  24. from mongoengine.fields import (BooleanField, StringField, IntField, EmbeddedDocumentListField,
  25. DateTimeField, PointField, DictField, ObjectIdField, ListField, EmbeddedDocumentField,
  26. MapField, GenericLazyReferenceField, LazyReferenceField)
  27. from pymongo.errors import DuplicateKeyError
  28. from pymongo.results import UpdateResult
  29. from typing import Union, Dict, AnyStr, Optional, TYPE_CHECKING, List, Tuple
  30. from apilib.monetary import RMB, VirtualCoin, Ratio, Percent, Money
  31. from apilib.quantity import Quantity
  32. from apilib.systypes import IterConstant
  33. from apilib.utils import flatten
  34. from apilib.utils_datetime import generate_timestamp_ex, today_format_str, to_datetime, get_tomorrow_zero_time
  35. from apilib.utils_mongo import BulkHandlerEx
  36. from apilib.utils_string import get_random_str
  37. from apps import serviceCache
  38. from apps.web.agent.models import Agent, MoniApp
  39. from apps.web.common.models import OrderRecordBase
  40. from apps.web.common.transaction import OrderNoMaker, OrderMainType, UserPaySubType, UserConsumeSubType, RefundSubType
  41. from apps.web.common.validation import CARD_NO_RE
  42. from apps.web.constant import (Const,
  43. DEALER_CONSUMPTION_AGG_KIND, GLOSSARY_TRANSLATION,
  44. USER_RECHARGE_TYPE,
  45. RECHARGE_CARD_TYPE,
  46. AppPlatformType, APP_PLATFORM_TYPE_TRANSLATION,
  47. RechargeRecordVia, RECHARGE_RECORD_VIA_TRANSLATION, ErrorCode,
  48. DEVICE_INCOME_STRATEGY, support_policy_device, support_policy_weifule,
  49. PARTITION_ROLE)
  50. from apps.web.core import PayAppType, ROLE
  51. from apps.web.core.bridge import WechatClientProxy
  52. from apps.web.core.db import Searchable, MonetaryField, VirtualCoinField, RoleBaseDocument, BooleanIntField
  53. from apps.web.core.exceptions import InsufficientFundsError, ServiceException, ImproperlyOperatedBalance
  54. from apps.web.core.models import BoundOpenInfo
  55. from apps.web.core.models import WechatAuthApp
  56. from apps.web.dealer.models import Dealer, VirtualCard, DealerDict
  57. from apps.web.device.models import Group, Device, DeviceType
  58. from apps.web.exceptions import UserServerException, PostPayOrderError
  59. from apps.web.report.ledger import AgentLedgerFirst, PartnerLedgerFirst
  60. from apps.web.report.utils import record_consumption_stats
  61. from apps.web.user.constant2 import StartDeviceType, PackageCategory, CONSUME_ORDER_PAY_TIMEOUT, ConsumeOrderServiceItem, UserBalanceChangeCategory
  62. from apps.web.utils import concat_front_end_url, concat_user_login_entry_url, concat_user_center_entry_url, set_or_incr_cache, concat_count_down_page_url
  63. from library.idgen import IDGenService
  64. from apps.web.core.payment import PaymentGateway
  65. logger = logging.getLogger(__name__)
  66. if TYPE_CHECKING:
  67. from apps.web.device.models import FeedBack
  68. from apps.web.device.models import DeviceDict, GroupDict
  69. from apps.web.common.models import CapitalUser
  70. from apps.web.dealer.models import MonthlyPackageTemp
  71. from mongoengine.queryset import QuerySet
  72. from apps.web.dealer.models import DealerRechargeRecord
  73. from apps.web.user.utils2 import StartParamContext, check_consume_order_timeout
  74. class MyUserAuthBackend(object):
  75. """
  76. 部分接口需要用户鉴权,由于目前用户体系尚未完整建立,在这里首先实现MVP
  77. """
  78. # noinspection PyUnusedLocal
  79. def authenticate(self, **kwargs):
  80. return True
  81. def get_user(self, user_id):
  82. return self.user_document.objects.with_id(user_id)
  83. @property
  84. def user_document(self):
  85. self._user_doc = MyUser
  86. return self._user_doc
  87. class EndUserLocation(EmbeddedDocument):
  88. logicalCode = StringField(verbose_name = '')
  89. # point :: { "type" : "Point" , "coordinates" : [longitude, latitude]}
  90. point = PointField(default = None, verbose_name = u'用户的经纬度')
  91. type = StringField(verbose_name = u'经纬度类型')
  92. createdTime = DateTimeField(default = datetime.datetime.now)
  93. @property
  94. def coordinates(self):
  95. if not self.point:
  96. return ()
  97. else:
  98. return tuple(self.point['coordinates'])
  99. def __repr__(self):
  100. return '<EndUserLocation (logicalCode=%s, point=(lng=%s, lat=%s), type=%s)>' \
  101. % (self.logicalCode,
  102. self.point['coordinates'][0],
  103. self.point['coordinates'][1],
  104. self.type)
  105. class UserMoney(EmbeddedDocument):
  106. settled = MonetaryField(verbose_name = "已经分账给经销商的金额", default = RMB('0.00')) # type: RMB
  107. unsettled = MonetaryField(verbose_name = "未分账给经销商的金额", default = RMB('0.00')) # type: RMB
  108. total_recharged = MonetaryField(default = RMB('0.00')) # type: RMB
  109. total_consumed = MonetaryField(default = RMB('0.00')) # type: RMB
  110. def __repr__(self):
  111. return '<UserMoney settled=%s, unsettled=%s, total_recharged=%s, total_consumed=%s>' \
  112. % (self.settled, self.unsettled, self.total_recharged, self.total_consumed)
  113. def to_dict(self):
  114. return {
  115. 'settled': self.settled,
  116. 'unsettled': self.unsettled,
  117. 'total_recharged': self.total_recharged,
  118. 'total_consumed': self.total_consumed
  119. }
  120. class UserCoin(EmbeddedDocument):
  121. balance = VirtualCoinField(default = VirtualCoin('0.00'))
  122. total_recharged = VirtualCoinField(default = VirtualCoin('0.00'))
  123. total_consumed = VirtualCoinField(default = VirtualCoin('0.00'))
  124. def __repr__(self):
  125. return '<UserCoin balance=%s total_recharged=%s total_consumed=%s>' \
  126. % (self.balance, self.total_recharged, self.total_consumed)
  127. def to_dict(self):
  128. return {
  129. 'balance': self.balance,
  130. 'total_recharged': self.total_recharged,
  131. 'total_consumed': self.total_consumed
  132. }
  133. class UniqueUser(Searchable):
  134. openId = StringField(verbose_name='第三方用户ID(微信的OPENID,支付宝的UID)', null=False, unique=True)
  135. userId = StringField(verbose_name='用户平台UID')
  136. phone = StringField(verbose_name='电话号码', default="")
  137. meta = {
  138. 'collection': 'unique_user',
  139. 'db_alias': 'default',
  140. 'indexes': [
  141. {'fields': ['userId'], 'unique': True, 'sparse': True}
  142. ]
  143. }
  144. @classmethod
  145. def get_or_create(cls, openId): # type:(str) -> UniqueUser
  146. """
  147. 创建或者是获取用户
  148. """
  149. user = cls.objects(openId = openId).first()
  150. if user:
  151. return user
  152. try:
  153. return cls(openId = openId).save()
  154. except DuplicateKeyError:
  155. return cls.objects(openId = openId).first()
  156. def update_phone(self, phone): # type:(str) -> bool
  157. """
  158. 更新电话号码
  159. """
  160. return bool(self.update(phone=phone))
  161. class MyUser(RoleBaseDocument):
  162. """
  163. `EndUser`
  164. 终端用户模型
  165. """
  166. sex = IntField(verbose_name=u"性别", default=Const.USER_SEX.UNKNOWN)
  167. phoneOS = StringField(verbose_name=u"终端操作系统", default="")
  168. city = StringField(verbose_name=u"城市", default="")
  169. province = StringField(verbose_name=u"省份", default="")
  170. country = StringField(verbose_name=u"国家", default="")
  171. avatar = StringField(verbose_name=u"头像地址", default="")
  172. nickname = StringField(verbose_name=u"名称", max_length=255, default="")
  173. groupId = StringField(verbose_name=u"地址编号", default="")
  174. gateway = StringField(verbose_name=u"来自支付宝或微信", default='wechat')
  175. chargeBalance = MonetaryField(verbose_name=u"余额", default=RMB('0'))
  176. bestowBalance = MonetaryField(verbose_name=u"赠送余额", default=RMB('0'))
  177. total_recharged = MonetaryField(verbose_name="累计充值", default=RMB('0'))
  178. total_bestow = MonetaryField(verbose_name=u"累计赠送", default=RMB('0'))
  179. total_consumed = VirtualCoinField(verbose_name="累计消费", default=RMB('0'))
  180. phoneNumber = StringField(verbose_name=u"用户手机号码", default="")
  181. authAppId = StringField(verbose_name=u"鉴权appid", default="")
  182. openId = StringField(verbose_name=u"openID|支付宝buyerID", default="")
  183. unionId = StringField(verbose_name=u"统一用户ID", default="")
  184. managerialAppId = StringField(verbose_name=u"管理公众号AppId", default="")
  185. managerialOpenId = StringField(verbose_name=u"管理openId", default="")
  186. payAppId = StringField(verbose_name=u"最近登录微信授权appid", default="")
  187. payOpenId = StringField(verbose_name=u"openId", default="")
  188. payOpenIdMap = MapField(EmbeddedDocumentField(BoundOpenInfo))
  189. extra = DictField(verbose_name=u"多余字段", default={})
  190. dateTimeAdded = DateTimeField(default=datetime.datetime.now, verbose_name=u'添加进来的时间')
  191. last_login = DateTimeField(default=datetime.datetime.now, verbose_name=u'最近登录时间')
  192. lastLoginUserAgent = StringField(verbose_name=u"最后一次登录的userAgent", default="")
  193. locations = EmbeddedDocumentListField(document_type=EndUserLocation)
  194. agentId = StringField(verbose_name=u'当前用户绑定的直接上级代理商', default='')
  195. productAgentId = StringField(verbose_name=u'当前用户绑定的平台代理商', default='')
  196. promo = DictField(verbose_name=u'活动相关信息', default={})
  197. favoriteDeviceList = ListField(verbose_name=u'收藏的宝贝', default=[])
  198. smsVendor = StringField(verbose_name=u'sms提供商', default='')
  199. ongoingList = ListField(field=DictField(), verbose=u'冻结的金额')
  200. blacklistConfig = DictField(verbose_name=u'黑名单相关设置', default={})
  201. meta = {
  202. "collection": "MyUser2",
  203. "db_alias": "default",
  204. "indexes": [{
  205. 'fields': ['$openId', '$nickname'],
  206. 'weights': {'openId': 10, 'nickname': 2}
  207. },
  208. 'openId',
  209. 'nickname',
  210. 'dateTimeAdded',
  211. # 'groupId'
  212. ],
  213. # "shard_key": ('openId',)
  214. }
  215. search_fields = ('openId', 'nickname')
  216. _AGENT_GROUP_ID_PREFIX = 'agent_'
  217. __mgr_cache = serviceCache
  218. def __str__(self):
  219. return '{}<id={} openId={} groupId={}>'.format(self.__class__.__name__, str(self.id), self.openId, self.groupId)
  220. @property
  221. def balance(self):
  222. return self.chargeBalance + self.bestowBalance
  223. def pay(self, payment): # type:(PaymentInfo) -> PaymentInfo
  224. if not payment:
  225. return
  226. if not payment.deduct_list:
  227. return
  228. # 虽然是循环扣除 但是一般情况下只会扣除1次
  229. bulker = BulkHandlerEx(self.__class__.get_collection()) # type: BulkHandlerEx
  230. chargeBalanceField = self.__class__.chargeBalance.name
  231. bestowBalanceField = self.__class__.bestowBalance.name
  232. for deduct in payment.deduct_list:
  233. query = {
  234. '_id': ObjectId(deduct['id']),
  235. }
  236. update = {
  237. '$inc': {
  238. chargeBalanceField: (-RMB(deduct[chargeBalanceField])).mongo_amount,
  239. bestowBalanceField: (-RMB(deduct[bestowBalanceField])).mongo_amount
  240. }
  241. }
  242. bulker.update(query, update)
  243. result = bulker.execute()
  244. if result['success'] == 0 or len(result['info']['writeErrors']) != 0:
  245. logger.error("[user pay] pay error, result = {}".format(result))
  246. else:
  247. # 添加支付时间并返回
  248. payment.time = datetime.datetime.now().strftime("%Y-%m-%d %H:%M:%S")
  249. return payment
  250. def account_consume(self, order): # type:(ConsumeRecord) -> None
  251. payment = order.payment
  252. # 获取支付的钱
  253. consumeAmount = payment.actualAmount
  254. consumeBestowAmount = payment.totalAmount - consumeAmount
  255. # 获取平台余额
  256. charge, bestow, _, __ = self.filter_my_balance()
  257. UserBalanceLog.consume(
  258. order.user,
  259. afterAmount=charge,
  260. afterBestowAmount=bestow,
  261. consumeAmount=consumeAmount,
  262. consumeBestowAmount=consumeBestowAmount,
  263. order=order
  264. )
  265. def account_refund(self, order): # type:(ConsumeRecord) -> None
  266. refund = order.refund
  267. # 获取支付的钱
  268. refundAmount = refund.actualAmount
  269. refundBestowAmount = refund.totalAmount - refundAmount
  270. # 获取平台余额
  271. charge, bestow, _, __ = self.filter_my_balance()
  272. UserBalanceLog.refund(
  273. order.user,
  274. afterAmount=charge,
  275. afterBestowAmount=bestow,
  276. refundAmount=refundAmount,
  277. refundBestowAmount=refundBestowAmount,
  278. order=order
  279. )
  280. def account_recharge(self, order): # type:(RechargeRecord) -> None
  281. # 获取支付的钱
  282. chargeAmount = order.chargeAmount
  283. bestowAmount = order.bestowAmount
  284. # 获取平台余额
  285. charge, bestow, _, __ = self.filter_my_balance()
  286. UserBalanceLog.recharge(
  287. order.user,
  288. afterAmount=charge,
  289. afterBestowAmount=bestow,
  290. chargeAmount=chargeAmount,
  291. chargeBestowAmount=bestowAmount,
  292. order=order
  293. )
  294. def is_authenticated(self):
  295. if not self.authAppId:
  296. return False
  297. return True
  298. def get_promo_info(self, key):
  299. return self.promo.get(key)
  300. def set_promo_info(self, key, value):
  301. promo = self.promo
  302. promo[key] = value
  303. return self.update(promo = promo)
  304. @property
  305. def inhouse_promo_openId(self):
  306. return self.get_promo_info("inhouse_openId")
  307. def set_inhouse_promo_openId(self, openId):
  308. return self.set_promo_info(key = "inhouse_openId", value = openId)
  309. @property
  310. def feature_keys(self):
  311. return ['phoneOS', 'sex', 'gateway']
  312. @property
  313. def feature_map(self):
  314. return {'phoneOS': self.phoneOS, 'sex': self.sex_in_en, 'gateway': self.gateway}
  315. @property
  316. def sex_in_en(self):
  317. if self.is_female:
  318. return 'female'
  319. elif self.is_male:
  320. return 'male'
  321. else:
  322. return ''
  323. @property
  324. def is_female(self):
  325. return self.sex == Const.USER_SEX.FEMALE
  326. @property
  327. def is_male(self):
  328. return self.sex == Const.USER_SEX.MALE
  329. def get_bound_pay_openid(self, key):
  330. # type: (str)->str
  331. if self.gateway == AppPlatformType.ALIPAY:
  332. return self.openId
  333. else:
  334. pay_openid_map = self.payOpenIdMap # type: dict
  335. bound = pay_openid_map.get(key, BoundOpenInfo(openId = '')) # type: BoundOpenInfo
  336. return bound.openId
  337. def set_bound_pay_openid(self, key, **payload):
  338. # type: (str, Dict)->None
  339. self.payOpenIdMap[key] = BoundOpenInfo(**payload)
  340. @classmethod
  341. def _product_group_id(cls, agent_id):
  342. # type:(str)->str
  343. return '{}{}'.format(cls._AGENT_GROUP_ID_PREFIX, agent_id)
  344. @classmethod
  345. def _is_product_group_id(cls, group_id):
  346. # type: (str)->bool
  347. return group_id.startswith(cls._AGENT_GROUP_ID_PREFIX)
  348. def is_product_user(self):
  349. return not self.groupId or self.groupId.startswith(self._AGENT_GROUP_ID_PREFIX)
  350. @classmethod
  351. def get_product_users(cls, agent_id): # type: (str) -> QuerySet
  352. productGroupId = cls._product_group_id(agent_id)
  353. return cls.objects.filter(groupId=productGroupId)
  354. @classmethod
  355. def get_or_create(cls, app_platform_type, open_id, group_id=groupId.default, **kwargs):
  356. # type: (str, str, str, Dict)->MyUser
  357. logger.info('get_or_create cls = {}, app_platform_type = {}, open_id = {}, groupId = {}, kwargs = {}'.format(
  358. cls, app_platform_type, open_id, group_id, str(kwargs)))
  359. if app_platform_type in [AppPlatformType.ALIPAY, AppPlatformType.JD]:
  360. kwargs.pop('payOpenIdMap', None)
  361. if group_id == cls.groupId.default: # 平台个人中心登录
  362. format_group_id = cls._product_group_id(kwargs.get('productAgentId'))
  363. else:
  364. format_group_id = group_id
  365. user = cls.objects(openId = open_id, groupId = format_group_id).first() # type: MyUser
  366. if user is not None:
  367. need_update = False
  368. if 'payOpenIdMap' in kwargs:
  369. for key, value in kwargs['payOpenIdMap'].iteritems():
  370. if key in user.payOpenIdMap:
  371. continue
  372. user.set_bound_pay_openid(key = key, **(value.to_dict()))
  373. need_update = True
  374. if 'authAppId' in kwargs and kwargs['authAppId'] != user.authAppId:
  375. user.authAppId = kwargs['authAppId']
  376. need_update = True
  377. if 'agentId' in kwargs and kwargs['agentId'] != user.agentId:
  378. user.agentId = kwargs['agentId']
  379. need_update = True
  380. if 'productAgentId' in kwargs and kwargs['productAgentId'] != user.productAgentId:
  381. user.productAgentId = kwargs['productAgentId']
  382. need_update = True
  383. if 'avatar' in kwargs and kwargs['avatar'] != user.avatar:
  384. user.avatar = kwargs['avatar']
  385. need_update = True
  386. if need_update:
  387. user.save()
  388. return user
  389. else:
  390. kwargs.update({'gateway': app_platform_type})
  391. user = cls(openId = open_id, groupId = format_group_id, **kwargs)
  392. try:
  393. return user.save()
  394. except NotUniqueError:
  395. return cls.objects(openId = open_id, groupId = format_group_id).first()
  396. def group_pay(self, amount):
  397. # type: (VirtualCoin)->None
  398. """
  399. :param amount:
  400. :return:
  401. """
  402. #: 检查是否宇宙里未分配的余额和是否小于要付的钱
  403. valid_to_pay = bool(self.get_collection().count_documents(
  404. filter = {
  405. '_id': ObjectId(self.id),
  406. '$expr':
  407. {'$lt': [amount.mongo_amount, {'$sum': ["$universe.money.unsettled"]}]}
  408. }
  409. ))
  410. if valid_to_pay:
  411. raise NotImplementedError()
  412. def update_total_recharge(self, money):
  413. # type: (RMB)->int
  414. assert isinstance(money, RMB), 'money has to be RMB'
  415. updated = self.update(inc__total_recharged = money.amount)
  416. return updated
  417. def incr_balance(self, coins):
  418. # type: (VirtualCoin)->int
  419. assert isinstance(coins, VirtualCoin), 'coins had to be VirtualCoin'
  420. updated = self.update(inc__balance = coins.amount)
  421. return updated
  422. def recharge(self, money, bestowMoney): # type: (RMB, RMB)->MyUser
  423. """
  424. 给用户充值,同时记录累计充值数额
  425. :param money: 币额
  426. :param bestowMoney: 金额
  427. :return:
  428. """
  429. assert isinstance(money, RMB), 'coins had to be VirtualCoin'
  430. assert isinstance(bestowMoney, RMB), 'money has to be RMB'
  431. updated = self.update(
  432. inc__chargeBalance=money,
  433. inc__total_recharged=money,
  434. inc__bestowBalance=bestowMoney,
  435. inc__total_bestow=bestowMoney
  436. )
  437. if not updated:
  438. raise PostPayOrderError(u'余额和累计充值更新失败')
  439. return self.reload()
  440. @classmethod
  441. def deduct_balance(cls, deduct_list):
  442. # type:(dict)->None
  443. """
  444. 按照订单的扣款列表直接扣除用户相应balance
  445. :param deduct_list:
  446. :return:
  447. """
  448. bulker = BulkHandlerEx(cls.get_collection()) # type: BulkHandlerEx
  449. for deduct in deduct_list:
  450. query = {
  451. '_id': ObjectId(deduct['id'])
  452. }
  453. update = {
  454. '$inc': {
  455. 'balance': (-VirtualCoin(deduct['coins'])).mongo_amount
  456. }
  457. }
  458. bulker.update(query, update)
  459. result = bulker.execute()
  460. logger.debug(result['info'])
  461. if result['success'] == 0:
  462. raise ServiceException({'result': 0, 'description': u"扣款失败(1001)"})
  463. else:
  464. if len(result['info']['writeErrors']) != 0:
  465. raise ServiceException({'result': 0, 'description': u"扣款失败(1002)"})
  466. @classmethod
  467. def freeze_balance(cls, transaction_id, payment): # type:(str, PaymentInfo) -> bool
  468. """
  469. 按照扣款列表冻结用户相应的金额
  470. :param transaction_id:
  471. :param payment:
  472. :return:
  473. """
  474. bulker = BulkHandlerEx(cls.get_collection()) # type: BulkHandlerEx
  475. chargeBalanceField = cls.chargeBalance.name
  476. bestowBalanceField = cls.bestowBalance.name
  477. for deduct in payment.deduct_list:
  478. query = {
  479. '_id': ObjectId(deduct['id']),
  480. 'ongoingList.transaction_id': {
  481. '$ne': transaction_id
  482. }
  483. }
  484. update = {
  485. '$inc': {
  486. chargeBalanceField: (-RMB(deduct[chargeBalanceField])).mongo_amount,
  487. bestowBalanceField: (-RMB(deduct[bestowBalanceField])).mongo_amount
  488. },
  489. '$addToSet': {
  490. 'ongoingList': {
  491. 'transaction_id': transaction_id,
  492. chargeBalanceField: deduct[chargeBalanceField],
  493. bestowBalanceField: deduct[bestowBalanceField]
  494. }
  495. }
  496. }
  497. bulker.update(query, update)
  498. result = bulker.execute()
  499. logger.debug(result['info'])
  500. if result['success'] == 0:
  501. raise ServiceException({'result': 0, 'description': u"扣款失败(1001)"})
  502. else:
  503. if len(result['info']['writeErrors']) != 0:
  504. raise ServiceException({'result': 0, 'description': u"扣款失败(1002)"})
  505. else:
  506. return True
  507. @classmethod
  508. def clear_frozen_balance(cls, transaction_id, payment, refund): # type:(str, PaymentInfo, PaymentInfo)->bool
  509. """
  510. 清除冻结金额 实际上就是相当于退费
  511. 如果没有退费 也需要调用一次 目的是清理掉transactionId
  512. """
  513. try:
  514. bulker = BulkHandlerEx(cls.get_collection()) # type: BulkHandlerEx
  515. chargeBalanceField = cls.chargeBalance.name
  516. bestowBalanceField = cls.bestowBalance.name
  517. # 将总消费情况 更新到第一个地址里面去
  518. first = True
  519. totalDeduct = payment.totalAmount
  520. totalRefund = refund.totalAmount
  521. for deduct in refund.deduct_list:
  522. query = {
  523. '_id': ObjectId(deduct['id']),
  524. 'ongoingList': {
  525. '$elemMatch': {
  526. 'transaction_id': transaction_id
  527. }
  528. }
  529. }
  530. if first:
  531. update = {
  532. '$inc': {
  533. chargeBalanceField: deduct[chargeBalanceField],
  534. bestowBalanceField: deduct[bestowBalanceField],
  535. cls.total_consumed.name: (totalDeduct - totalRefund).mongo_amount
  536. },
  537. '$pull': {
  538. 'ongoingList': {
  539. 'transaction_id': transaction_id
  540. }
  541. }
  542. }
  543. else:
  544. update = {
  545. '$inc': {
  546. chargeBalanceField: deduct[chargeBalanceField],
  547. bestowBalanceField: deduct[bestowBalanceField],
  548. },
  549. '$pull': {
  550. 'ongoingList': {
  551. 'transaction_id': transaction_id
  552. }
  553. }
  554. }
  555. first = False
  556. bulker.update(query, update)
  557. result = bulker.execute()
  558. logger.debug(result['info'])
  559. if result['success'] == 0:
  560. return False
  561. else:
  562. if len(result['info']['writeErrors']) != 0:
  563. return False
  564. else:
  565. return True
  566. except Exception as e:
  567. logger.exception(e)
  568. return False
  569. @classmethod
  570. def recover_frozen_balance(cls, transaction_id, deduct_list):
  571. # type:(str, list)->bool
  572. """
  573. 回滚冻结的余额
  574. """
  575. bulker = BulkHandlerEx(cls.get_collection()) # type: BulkHandlerEx
  576. for deduct in deduct_list:
  577. query = {
  578. '_id': ObjectId(deduct['id']),
  579. 'ongoingList': {
  580. '$elemMatch': {'transaction_id': transaction_id}}}
  581. update = {
  582. '$inc': {
  583. 'balance': deduct['coins']
  584. },
  585. '$pull': {
  586. 'ongoingList': {
  587. 'transaction_id': transaction_id, 'frozenCoins': deduct['coins']
  588. }}
  589. }
  590. bulker.update(query, update)
  591. result = bulker.execute()
  592. if result['success'] == 0:
  593. logger.error(result['info'])
  594. return False
  595. else:
  596. if len(result['info']['writeErrors']) != 0:
  597. logger.error(result['info'])
  598. return False
  599. else:
  600. return True
  601. def to_dict(self, default_avatar = settings.DEFAULT_AVATAR_URL):
  602. avatar = self.avatar or default_avatar or Agent.get_agent(self.agentId).get("productLogo")
  603. return {
  604. 'avatar': avatar,
  605. 'nickname': self.nickname,
  606. 'openId': self.openId,
  607. 'groupId': self.groupId,
  608. 'sex': self.sex,
  609. 'balance': self.balance,
  610. 'total_recharged': self.total_recharged,
  611. 'total_consumed': self.total_consumed,
  612. }
  613. @property
  614. def product_users(self):
  615. return MyUser.objects.filter(
  616. __raw__ = {
  617. 'openId': self.openId,
  618. '$or': [
  619. {
  620. 'agentId': self.agentId
  621. },
  622. {
  623. 'productAgentId': self.productAgentId
  624. }
  625. ]
  626. })
  627. @property
  628. def total_balance(self):
  629. """
  630. 个人中心获取balance. 需要按照agentId过滤
  631. :return:
  632. """
  633. totalCharge, totalBestow, total, dataList = self.filter_my_balance()
  634. return totalCharge + totalBestow
  635. def filter_my_balance(self, currentGroup=None):
  636. """
  637. 个人中心获取balance列表. 需要按照agentId过滤
  638. """
  639. dataList = []
  640. total = 0
  641. totalCharge = RMB(0)
  642. totalBestow = RMB(0)
  643. chargeBalanceField = self.__class__.chargeBalance.name
  644. bestowBalanceField = self.__class__.bestowBalance.name
  645. # dealer_partition = defaultdict(lambda: {
  646. # chargeBalanceField: RMB(0),
  647. # bestowBalanceField: RMB(0)
  648. # })
  649. users = self.product_users
  650. for user in users: # type: MyUser
  651. groupId = user.groupId
  652. if user.is_product_user():
  653. logger.debug('groupId <{}> is agent group id.'.format(groupId))
  654. continue
  655. group = Group.get_group(groupId) # type: GroupDict
  656. if not group:
  657. logger.error('no group. id = %s' % groupId)
  658. continue
  659. dealer = Dealer.get_dealer(group['ownerId']) # type: DealerDict
  660. if dealer is None:
  661. logger.error('no dealer. id = %s' % group['ownerId'])
  662. continue
  663. # dealer_partition[dealer['id']][chargeBalanceField] += user.chargeBalance
  664. # dealer_partition[dealer['id']][bestowBalanceField] += user.bestowBalance
  665. # 过滤掉为0的情况
  666. if user.balance == RMB(0):
  667. continue
  668. else:
  669. total += 1
  670. totalCharge += user.chargeBalance
  671. totalBestow += user.bestowBalance
  672. address = group.address
  673. rv = {
  674. 'address': address,
  675. 'groupId': groupId,
  676. 'groupName': group.groupName,
  677. 'dealerId': group.ownerId,
  678. 'balance': {
  679. chargeBalanceField: user.chargeBalance,
  680. bestowBalanceField: user.bestowBalance
  681. },
  682. 'currency': False
  683. }
  684. if currentGroup and dealer.is_currency(currentGroup, group):
  685. rv['currency'] = True
  686. dataList.append(rv)
  687. return totalCharge, totalBestow, total, dataList
  688. def get_balance_in_dealer(self, dealer, group): # type: (Dealer, GroupDict)->Tuple[RMB, RMB, RMB]
  689. if self.groupId != group.groupId:
  690. payload = self.cloneable_user_info
  691. payload['agentId'] = dealer.agentId
  692. user = MyUser.get_or_create(
  693. app_platform_type = self.gateway, open_id = self.openId, group_id = group.groupId, **payload)
  694. else:
  695. user = self
  696. users = user.product_users
  697. overall, dealer_balance, currency_balance = RMB(0), RMB(0), RMB(0)
  698. dealer_group_ids = Group.get_group_ids_of_dealer(str(dealer.id)) # type: List[GroupDict]
  699. groups = Group.get_groups_by_group_ids(dealer_group_ids).values()
  700. share_group_ids = dealer.get_currency_group_ids(group, groups)
  701. share_group_ids.append(group.groupId)
  702. for user in users:
  703. overall += user.balance
  704. if user.groupId in dealer_group_ids:
  705. dealer_balance += user.balance
  706. if user.groupId in share_group_ids:
  707. currency_balance += user.balance
  708. return overall, dealer_balance, currency_balance
  709. @classmethod
  710. def get_new_user_count(cls, groupIds, start, end):
  711. # type: (list, datetime.datetime, datetime.datetime)->int
  712. result = cls.get_collection().aggregate(
  713. [
  714. {
  715. '$match': {
  716. "groupId": {"$in": groupIds},
  717. "dateTimeAdded": {"$gte": start, "$lte": end}
  718. }
  719. },
  720. {
  721. '$group': {
  722. '_id': '$openId'
  723. }
  724. },
  725. {'$count': 'count'},
  726. ]
  727. )
  728. return next(result, {}).get("count", 0)
  729. @classmethod
  730. def get_user_count_by_filter(cls, groupIds, filterDict = {}):
  731. # type: (list)->int
  732. filter = {"groupId": {"$in": groupIds}}
  733. filter.update(filterDict)
  734. result = cls.get_collection().aggregate(
  735. [
  736. {
  737. '$match': filter
  738. },
  739. {
  740. '$group': {
  741. '_id': '$openId'
  742. }
  743. },
  744. {'$count': 'count'},
  745. ]
  746. )
  747. return next(result, {}).get("count", 0)
  748. def get_extra_info(self, key):
  749. return self.extra.get(key)
  750. def set_extra_info(self, key, value):
  751. extra = self.extra
  752. extra[key] = value
  753. return self.update(extra = extra)
  754. @staticmethod
  755. def get_active_info(openId, **kwargs):
  756. """
  757. 获取用户的激活状态 安骑换电用户的激活状态在一个经销商下只会有一个激活状态
  758. """
  759. users = MyUser.objects(openId = openId, **kwargs)
  760. for user in users:
  761. if user.get_extra_info("active"):
  762. curUser = user
  763. break
  764. else:
  765. return dict() # 没找到用户的激活信息
  766. return curUser.get_extra_info("active")
  767. @staticmethod
  768. def del_active_info(openId, **kwargs):
  769. users = MyUser.objects(openId = openId, **kwargs)
  770. for user in users:
  771. if user.get_extra_info("active"):
  772. user.set_extra_info("active", {})
  773. user.phoneNumber = ""
  774. user.save()
  775. break
  776. @staticmethod
  777. def set_active_info(setInfo, openId, **kwargs):
  778. users = MyUser.objects(openId = openId, **kwargs)
  779. for user in users:
  780. if user.get_extra_info("active"):
  781. curUser = user
  782. break
  783. else: # 用户第一次注册
  784. curUser = user
  785. active = curUser.get_extra_info("active")
  786. if not active:
  787. active = dict()
  788. active.update(setInfo)
  789. curUser.set_extra_info("active", active)
  790. if setInfo.get("isMember"): # 被经销商激活的同时,为用户添加phoneNumber字段
  791. curUser.phoneNumber = active.get("phoneNumber")
  792. curUser.save()
  793. return
  794. @property
  795. def cloneable_user_info(self):
  796. """
  797. 对于微信, 必须是相同的平台, 用户信息才能一致;
  798. 对于京东和支付宝, 用户信息都是一致的
  799. :return:
  800. """
  801. return {
  802. 'authAppId': self.authAppId,
  803. 'avatar': self.avatar,
  804. 'nickname': self.nickname,
  805. 'managerialAppId': self.managerialAppId,
  806. 'managerialOpenId': self.managerialOpenId,
  807. 'payOpenIdMap': self.payOpenIdMap,
  808. 'gateway': self.gateway,
  809. 'productAgentId': self.productAgentId
  810. }
  811. @property
  812. def my_product_user(self):
  813. payload = self.cloneable_user_info
  814. payload['agentId'] = self.productAgentId
  815. return MyUser.get_or_create(self.gateway, self.openId, **payload)
  816. @property
  817. def collectedDeviceList(self):
  818. """
  819. 用户收藏的设备
  820. :return:
  821. """
  822. return self.my_product_user.favoriteDeviceList
  823. @collectedDeviceList.setter
  824. def collectedDeviceList(self, value):
  825. cur_user = self.my_product_user
  826. cur_user.favoriteDeviceList = value
  827. cur_user.save()
  828. @property
  829. def phone(self):
  830. """
  831. 用户的电话号码
  832. 之前是存储在productUser上面 现在需要做一次转换适配
  833. """
  834. uniqueUser = UniqueUser.get_or_create(self.openId)
  835. # 唯一用户存储了电话的情况下 直接返回唯一用户
  836. if uniqueUser.phone:
  837. return uniqueUser.phone
  838. # 从以前的存储电话号码的地方找出存储的电话 如有 转移到uniquerUser上面
  839. phoneNumber = self.my_product_user.phoneNumber
  840. if phoneNumber:
  841. uniqueUser.update_phone(phoneNumber)
  842. return phoneNumber
  843. else:
  844. return phoneNumber
  845. @phone.setter
  846. def phone(self, value):
  847. """
  848. 保存用户的电话号码 将直接保存到uniqueUser上面去
  849. """
  850. uniqueUser = UniqueUser.get_or_create(self.openId)
  851. uniqueUser.update_phone(value)
  852. @property
  853. def user_id(self):
  854. """
  855. 用户系统分配ID
  856. """
  857. if settings.ID_SERVICE_IP:
  858. uniqueUser = UniqueUser.get_or_create(self.openId)
  859. if not uniqueUser.userId:
  860. try:
  861. user_id = IDGenService(server_ip = settings.ID_SERVICE_IP).get_id()
  862. uniqueUser.userId = user_id
  863. uniqueUser.save()
  864. except Exception as e:
  865. logger.error('update user id failure.')
  866. logger.exception(e)
  867. return uniqueUser.userId
  868. else:
  869. return None
  870. def record_and_check_access_day(self, name, limit):
  871. dayTime = datetime.datetime.now().strftime(Const.DATE_FMT)
  872. key = '%s-%s-%s' % (self.openId, name, dayTime)
  873. value = serviceCache.get(key, 0)
  874. if value >= limit:
  875. return False
  876. value += 1
  877. serviceCache.set(key, value, 24 * 60 * 60)
  878. return True
  879. @property
  880. def cards_num(self):
  881. """
  882. 统计该用户的该平台下的实体卡数量
  883. :return:
  884. """
  885. filters = {"openId": self.openId, "agentId": self.productAgentId}
  886. return Card.objects.filter(**filters).count()
  887. @property
  888. def many_cards(self):
  889. """
  890. 是否允许 该用户拥有多张实体卡
  891. :return:
  892. """
  893. if "manyCards" in self.extra and self.get_extra_info("manyCards"):
  894. return True
  895. return False
  896. @classmethod
  897. def get_dealer_ids(cls, openId, productAgentId):
  898. users = cls.objects(openId = openId, productAgentId = productAgentId)
  899. group_ids = []
  900. for user in users:
  901. if not user.is_product_user():
  902. group_ids.append(user.groupId)
  903. groups = Group.get_groups_by_group_ids(group_ids)
  904. dealers = set()
  905. for group in groups.values():
  906. dealers.add(group.ownerId)
  907. return list(dealers)
  908. def calc_currency_balance(self, dealer, group, groups = None):
  909. # type:(Dealer, GroupDict, List[GroupDict])->RMB
  910. share_group_ids = dealer.get_currency_group_ids(group, groups)
  911. share_group_ids.append(group.groupId)
  912. users = MyUser.objects(openId =self.openId, groupId__in=share_group_ids)
  913. return sum((u.balance for u in users), RMB(0))
  914. def calc_currency_total_recharge(self, dealer, group):
  915. # type:(Dealer, GroupDict)->RMB
  916. share_group_ids = dealer.get_currency_group_ids(group)
  917. share_group_ids.append(group.groupId)
  918. users = MyUser.objects(openId = self.openId, groupId__in = share_group_ids)
  919. if users.count() == 0:
  920. return RMB(0)
  921. else:
  922. return RMB(users.sum('total_recharged'))
  923. @property
  924. def group(self):
  925. if self.is_product_user():
  926. return None
  927. if hasattr(self, '__group__'):
  928. return getattr(self, '__group__')
  929. group = Group.get_group(self.groupId) # type: GroupDict
  930. if group:
  931. setattr(self, '__group__', group)
  932. return group
  933. @property
  934. def username(self):
  935. return self.nickname
  936. @property
  937. def request_limit_key(self):
  938. return self.openId
  939. @classmethod
  940. def get_day_used_cache(cls, openId):
  941. """获取用户的每日使用次数信息"""
  942. key = '{}_day_used'.format(openId)
  943. result = cls.__mgr_cache.get(key, 0)
  944. return result
  945. @classmethod
  946. def update_day_used_cache(cls, openId):
  947. """
  948. 更新设备的每日使用次数信息
  949. :param openId:
  950. :return:
  951. """
  952. try:
  953. key = '{}_day_used'.format(openId)
  954. set_or_incr_cache(cls.__mgr_cache, key, 1, 24 * 60 * 60)
  955. except Exception:
  956. pass
  957. @classmethod
  958. def clear_day_used_cache(cls, openId):
  959. """清除用户的每日使用次数信息"""
  960. key = '{}_day_used'.format(openId)
  961. cls.__mgr_cache.set(key, 0)
  962. @classmethod
  963. def get_today_can_use(cls, openId):
  964. usedCount = MyUser.get_day_used_cache(openId)
  965. if usedCount > serviceCache.get('{}_day_used_limit'.format(openId), 10):
  966. return False
  967. return True
  968. def prepare_refund_cash(self, refund_order, minus_total_consume): # type:(RefundMoneyRecord, VirtualCoin)->bool
  969. query = {
  970. '_id': self.id,
  971. 'ongoingList.transaction_id': {
  972. '$ne': 'r_{}'.format(str(refund_order.id))
  973. }
  974. }
  975. update = {
  976. '$inc': {
  977. 'chargeBalance': (-refund_order.coins).mongo_amount,
  978. 'total_recharged': (-refund_order.money).mongo_amount,
  979. },
  980. '$addToSet': {
  981. 'ongoingList': {
  982. 'transaction_id': 'r_{}'.format(str(refund_order.id)),
  983. 'chargeBalance': refund_order.money.mongo_amount,
  984. 'minus_total_consume': minus_total_consume.mongo_amount
  985. }
  986. }
  987. }
  988. result = self.get_collection().update_one(query, update, upsert = False) # type: UpdateResult
  989. return bool(result.modified_count == 1)
  990. def commit_refund_cash(self, refund_order):
  991. # type:(RefundMoneyRecord)->bool
  992. query = {
  993. '_id': self.id,
  994. 'ongoingList': {
  995. '$elemMatch': {
  996. 'transaction_id': 'r_{}'.format(str(refund_order.id))
  997. }
  998. }
  999. }
  1000. update = {
  1001. '$pull': {
  1002. 'ongoingList': {'transaction_id': 'r_{}'.format(str(refund_order.id))}
  1003. }
  1004. }
  1005. result = self.get_collection().update_one(query, update, upsert = False) # type: UpdateResult
  1006. return bool(result.modified_count == 1)
  1007. @property
  1008. def isNormal(self):
  1009. return True
  1010. class MyUserDetail(Searchable):
  1011. openId = StringField(verbose_name="Myuser中的id", default="")
  1012. dealerId = StringField(verbose_name="经销商Id", default="")
  1013. userName = StringField(verbose_name="用户姓名", default="")
  1014. userPhone = StringField(verbose_name="用户电话", default="")
  1015. userUnit = StringField(verbose_name="用户地址单元", default="")
  1016. userFloor = StringField(verbose_name="用户地址楼层", default="")
  1017. userRoom = StringField(verbose_name="用户地址房间", default="")
  1018. meta = {
  1019. "collection": "MyUserDetail",
  1020. "db_alias": "logdata"}
  1021. @classmethod
  1022. def get_collection(cls):
  1023. # type: ()->Collection
  1024. return cls._get_collection()
  1025. class RechargeRecordDict(dict):
  1026. def __repr__(self):
  1027. return '<RechargeRecordDict id=%s>' % (self.get('id'),)
  1028. @property
  1029. def v(self):
  1030. return dict(self)
  1031. @property
  1032. def ownerId(self):
  1033. return self.get('ownerId')
  1034. @property
  1035. def groupId(self):
  1036. return self.get('groupId')
  1037. @property
  1038. def money(self):
  1039. return RMB(self['money'])
  1040. @property
  1041. def coins(self):
  1042. return VirtualCoin(self['coins'])
  1043. @property
  1044. def attachParas(self):
  1045. return self.get('attachParas')
  1046. @property
  1047. def extraInfo(self):
  1048. return self.get('extraInfo')
  1049. @property
  1050. def group(self):
  1051. # type:()->GroupDict
  1052. _attr_name = '__my_group__'
  1053. if hasattr(self, _attr_name):
  1054. return getattr(self, _attr_name)
  1055. else:
  1056. group = Group.get_group(self.groupId)
  1057. setattr(self, _attr_name, group)
  1058. return group
  1059. @property
  1060. def owner(self):
  1061. _attr_name = '__my_owner__'
  1062. if hasattr(self, _attr_name):
  1063. return getattr(self, _attr_name)
  1064. else:
  1065. dealer = Dealer.objects(id = self.ownerId).first()
  1066. setattr(self, _attr_name, dealer)
  1067. return dealer
  1068. @property
  1069. def my_gateway(self):
  1070. return self.get('gateway')
  1071. @property
  1072. def pay_app_type(self):
  1073. return self.get('payAppType')
  1074. @property
  1075. def pay_gateway_key(self):
  1076. return self.get('payGatewayKey')
  1077. @property
  1078. def withdraw_source_key(self):
  1079. return self.get('withdrawSourceKey')
  1080. @property
  1081. def dev_type_name(self):
  1082. if 'devTypeName' in self:
  1083. return self.get('devTypeName')
  1084. elif 'devType' in self:
  1085. return self.get('devType')
  1086. else:
  1087. return u'其他'
  1088. @property
  1089. def logicalCode(self):
  1090. return self.get('logicalCode')
  1091. @property
  1092. def address(self):
  1093. return self.get('address')
  1094. @property
  1095. def via(self):
  1096. return self.get('via')
  1097. @property
  1098. def partition_map(self):
  1099. """
  1100. 新订单模型下, 创建订单就会生成partitionMap. 所以直接取
  1101. :return:
  1102. """
  1103. return self.extraInfo.get('partitionMap', None)
  1104. def calc_income_partitions(self, partition_map):
  1105. totalMoney = self.money
  1106. leftMoney = totalMoney
  1107. agent_partitions = partition_map.get(PARTITION_ROLE.AGENT, list())
  1108. if agent_partitions:
  1109. for agentShare in agent_partitions:
  1110. shareMoney = totalMoney * Percent(agentShare['share']).as_ratio
  1111. if shareMoney > leftMoney:
  1112. shareMoney = leftMoney
  1113. leftMoney = leftMoney - shareMoney
  1114. agentShare['money'] = shareMoney.mongo_amount
  1115. partner_partitions = partition_map.get(PARTITION_ROLE.PARTNER, list())
  1116. if partner_partitions:
  1117. for partnerShare in partner_partitions:
  1118. shareMoney = totalMoney * Percent(partnerShare['share']).as_ratio
  1119. if shareMoney > leftMoney:
  1120. shareMoney = leftMoney
  1121. leftMoney = leftMoney - shareMoney
  1122. partnerShare['money'] = shareMoney.mongo_amount
  1123. owner_partitions = partition_map.get(PARTITION_ROLE.OWNER, list())
  1124. assert len(owner_partitions) == 1, u'经销商只能有一个'
  1125. ownerShare = owner_partitions[0]
  1126. ownerShare['money'] = leftMoney.mongo_amount
  1127. return partition_map
  1128. def calc_refund_partitions(self, pay_split_map):
  1129. totalMoney = self.money
  1130. leftMoney = totalMoney
  1131. share_key = lambda _role, _id: '{}_{}'.format(_role, _id)
  1132. old_shares_map = {}
  1133. partition_map = {
  1134. PARTITION_ROLE.AGENT: [],
  1135. PARTITION_ROLE.PARTNER: [],
  1136. PARTITION_ROLE.OWNER: []
  1137. }
  1138. agent_partitions = pay_split_map.get(PARTITION_ROLE.AGENT, list())
  1139. if agent_partitions:
  1140. for agentShare in agent_partitions:
  1141. old_shares_map[share_key(agentShare['role'], agentShare['id'])] = RMB(agentShare['money'])
  1142. newAgentShare = copy.deepcopy(agentShare)
  1143. shareMoney = totalMoney * Percent(agentShare['share']).as_ratio
  1144. if abs(shareMoney) > abs(leftMoney):
  1145. shareMoney = leftMoney
  1146. leftMoney = leftMoney - shareMoney
  1147. newAgentShare['money'] = shareMoney.mongo_amount
  1148. partition_map[PARTITION_ROLE.AGENT].append(newAgentShare)
  1149. partner_partitions = pay_split_map.get(PARTITION_ROLE.PARTNER, list())
  1150. if partner_partitions:
  1151. for partnerShare in partner_partitions:
  1152. old_shares_map[share_key(partnerShare['role'], partnerShare['id'])] = RMB(partnerShare['money'])
  1153. newPartnerShare = copy.deepcopy(partnerShare)
  1154. shareMoney = totalMoney * Percent(partnerShare['share']).as_ratio
  1155. if abs(shareMoney) > abs(leftMoney):
  1156. shareMoney = leftMoney
  1157. leftMoney = leftMoney - shareMoney
  1158. newPartnerShare['money'] = shareMoney.mongo_amount
  1159. partition_map[PARTITION_ROLE.PARTNER].append(newPartnerShare)
  1160. owner_partitions = pay_split_map.get(PARTITION_ROLE.OWNER, list())
  1161. assert len(owner_partitions) == 1, u'经销商只能有一个'
  1162. ownerShare = owner_partitions[0]
  1163. newOwnerShare = copy.deepcopy(ownerShare)
  1164. old_share_money = RMB(ownerShare['money'])
  1165. if abs(leftMoney) <= old_share_money:
  1166. newOwnerShare['money'] = leftMoney.mongo_amount
  1167. else:
  1168. diff = abs(leftMoney) - old_share_money
  1169. shares = list(flatten(partition_map.values()))
  1170. shares.sort(key = lambda x: Percent(x['share']), reverse = True)
  1171. for share in shares:
  1172. if RMB(share['money']) - RMB('0.01') + RMB(
  1173. old_shares_map[share_key(share['role'], share['id'])]) >= RMB(0):
  1174. share['money'] = (RMB(share['money']) - RMB('0.01')).mongo_amount
  1175. diff -= RMB('0.01')
  1176. leftMoney += RMB('0.01')
  1177. if diff <= RMB(0):
  1178. break
  1179. assert diff == RMB(0), u'分账金额错误'
  1180. newOwnerShare['money'] = leftMoney.mongo_amount
  1181. partition_map[PARTITION_ROLE.OWNER] = [newOwnerShare]
  1182. for role, items in partition_map.iteritems():
  1183. for item in items:
  1184. if RMB(item['money']) == RMB(0):
  1185. item['money'] = RMB(0).mongo_amount
  1186. return partition_map
  1187. class RechargeRecord(OrderRecordBase):
  1188. class PayResult(IterConstant):
  1189. UNPAY = 'unPay'
  1190. SUCCESS = 'success'
  1191. FAILED = 'failed'
  1192. REFUNDING = 'refunding'
  1193. CANCEL = 'cancel'
  1194. orderNo = StringField(verbose_name=u"订单号", unique=True)
  1195. wxOrderNo = StringField(verbose_name=u"渠道订单号")
  1196. transactionId = StringField(verbose_name=u"微信或者支付宝订单号", default=None)
  1197. ownerId = StringField(verbose_name = u"设备的owner", default = "")
  1198. #: 充值的金钱数额 incr(user.balance)
  1199. money = MonetaryField(verbose_name = u"充值", default = RMB('0.00'))
  1200. #: 用户购得的金币,如果是卡,就是充值的钱
  1201. coins = VirtualCoinField(verbose_name = u"金币数目", default = VirtualCoin('0.00'))
  1202. subject = StringField(verbose_name = u"订单产品或者服务描述", default = "")
  1203. result = StringField(verbose_name = u"充值或者服务结果", default = PayResult.UNPAY)
  1204. extraInfo = DictField(verbose_name = u"支付订单模型信息", default = {})
  1205. description = StringField(verbose_name = u"结果描述,一般为第三方错误码", default = None)
  1206. isQuickPay = BooleanField(verbose_name = u"是否直接支付使用,默认为否", default = False)
  1207. selectedQuickPayPackageId = StringField(verbose_name = u"快捷支付所选的投币套餐", default = None)
  1208. via = StringField(verbose_name = u"充值途径 =: (recharge|sendcoin|refund|chargeCard|swap)", default='recharge')
  1209. finishedTime = DateTimeField(verbose_name=u"支付完成时间")
  1210. # 派币需要记录信息
  1211. operator = StringField(verbose_name=u"操作员")
  1212. attachParas = DictField(verbose_name=u"消费订单信息", default={})
  1213. gateway = StringField(verbose_name=u'支付网关类型', default='')
  1214. payAppType = StringField(verbose_name=u'支付应用类型', default=None)
  1215. payGatewayKey = StringField(verbose_name=u'支付网关key', default=None)
  1216. withdrawSourceKey = StringField(verbose_name=u'提现商户平台标识,资金池模式下和代理商相同')
  1217. isAllocatedCardMoney = BooleanField(verbose_name=u'是否分钱了', default=False)
  1218. search_fields = ('orderNo', 'wxOrderNo')
  1219. _shard_key = ('ownerId', 'dateTimeAdded')
  1220. _origin_meta = {
  1221. "collection": "RechargeRecord",
  1222. "db_alias": "default"
  1223. }
  1224. meta = _origin_meta
  1225. def __repr__(self):
  1226. return '<RechargeRecord id=%s, orderNo=%s, wxOrderNo=%s, via=%s>' % (str(self.id), self.orderNo, self.wxOrderNo, self.via)
  1227. @cached_property
  1228. def payGateway(self):
  1229. return PaymentGateway.clone_from_order(self)
  1230. @cached_property
  1231. def user(self):
  1232. return MyUser.objects.filter(openId=self.openId, groupId=self.groupId).first()
  1233. @cached_property
  1234. def dealer(self):
  1235. return Dealer.objects(id=self.ownerId).get()
  1236. @property
  1237. def goods(self): # type:()->dict
  1238. return {}
  1239. @property
  1240. def timeout(self): # type:()-> bool
  1241. """支付超时"""
  1242. return (datetime.datetime.now() - self.dateTimeAdded).total_seconds() > 60 * 3
  1243. @property
  1244. def chargeAmount(self):
  1245. return self.money
  1246. @property
  1247. def bestowAmount(self):
  1248. return RMB(self.coins) - RMB(self.money)
  1249. @property
  1250. def my_gateway(self):
  1251. return self.gateway
  1252. @property
  1253. def pay_app_type(self):
  1254. return self.payAppType
  1255. @property
  1256. def pay_gateway_key(self):
  1257. return self.payGatewayKey
  1258. @property
  1259. def owner(self):
  1260. return self.dealer
  1261. @property
  1262. def payOrder(self):
  1263. """
  1264. 用户的实际支付的订单 由于业务需要 可能被拆分为多笔
  1265. # TODO 需要搞定历史数据库的问题
  1266. """
  1267. if self.isSubOrder:
  1268. _id = self.attachParas.get("tradeOrderId", None) or self.extraInfo.get('tradeOrderId', None)
  1269. return self.__class__.objects.get(id = _id)
  1270. return self
  1271. @property
  1272. def isSubOrder(self):
  1273. return "tradeOrderId" in self.extraInfo or "tradeOrderId" in self.attachParas
  1274. @property
  1275. def card_no(self):
  1276. return self.attachParas.get('cardNo', None)
  1277. @property
  1278. def my_description(self):
  1279. if self.description:
  1280. return self.description
  1281. else:
  1282. return getattr(self, 'desc', '')
  1283. @property
  1284. def my_via(self):
  1285. """
  1286. 老的订单模型, 对于recharge, 根据是否快速支付做一个调整
  1287. :return:
  1288. """
  1289. return self.via
  1290. @property
  1291. def withdraw_source_key(self):
  1292. return self.withdrawSourceKey
  1293. @property
  1294. def partition_map(self):
  1295. return self.to_dict_obj.partition_map
  1296. @property
  1297. def is_cancel(self):
  1298. return self.result == self.PayResult.CANCEL
  1299. @property
  1300. def fen_total_fee(self):
  1301. return int(self.money * 100)
  1302. @property
  1303. def amount(self):
  1304. # type:()->RMB
  1305. return self.money
  1306. @property
  1307. def my_amount(self):
  1308. if self.via in [RechargeRecordVia.RefundCash,
  1309. RechargeRecordVia.Cash,
  1310. RechargeRecordVia.VirtualCard,
  1311. RechargeRecordVia.MonthlyPackage]:
  1312. return '{}元'.format(abs(self.money))
  1313. else:
  1314. return self.coins
  1315. @property
  1316. def mongo_amount(self):
  1317. return self.amount.mongo_amount
  1318. @property
  1319. def extra_detail_info(self):
  1320. rv = {}
  1321. # TODO: 需要根据RechargeRecord建立各VIA下的订单详细信息
  1322. if self.via == 'chargeCard':
  1323. cardId = str(self.attachParas.get('cardId'))
  1324. card = Card.objects(id = cardId).first()
  1325. if card:
  1326. rv.update({'cardNo': card.cardNo, 'cardType': card.cardType})
  1327. elif self.via == 'chargeVirtualCard':
  1328. if 'type' not in self.attachParas:
  1329. # 老模型在new的情况下, 是没有type字段的
  1330. cardNo = self.attachParas.get('cardNo', None)
  1331. if cardNo:
  1332. rv.update({'cardNo': cardNo})
  1333. cardId = self.attachParas.get('cardId', None)
  1334. if cardId:
  1335. cardTmpl = VirtualCard.objects(id = cardId).first() # type: VirtualCard
  1336. if cardTmpl:
  1337. rv.update({'cardName': cardTmpl.cardName})
  1338. else:
  1339. if 'cardId' in self.attachParas:
  1340. virtualCard = UserVirtualCard.objects(
  1341. id = self.attachParas['cardId']).first() # type: UserVirtualCard
  1342. rv.update({
  1343. 'cardNo': virtualCard.cardNo, 'cardName': virtualCard.cardName
  1344. })
  1345. return rv
  1346. @property
  1347. def summary_info(self):
  1348. rv = {
  1349. 'id': str(self.id),
  1350. 'createdTime': self.to_datetime_str(self.dateTimeAdded),
  1351. 'money': self.money,
  1352. 'coins': self.coins,
  1353. 'groupName': self.groupName,
  1354. 'groupNumber': self.groupNumber,
  1355. 'address': self.address,
  1356. 'logicalCode': self.logicalCode,
  1357. 'devTypeName': self.dev_type_name,
  1358. 'devTypeCode': self.dev_type_code,
  1359. 'ownerId': self.ownerId,
  1360. 'via': self.via
  1361. }
  1362. if self.via == 'chargeVirtualCard':
  1363. rv.pop('coins')
  1364. return rv
  1365. @property
  1366. def created_date(self):
  1367. return self.dateTimeAdded
  1368. @property
  1369. def finished_time(self):
  1370. if self.result != self.PayResult.SUCCESS:
  1371. return ''
  1372. if self.finishedTime:
  1373. return self.to_datetime_str(self.finishedTime)
  1374. return self.to_datetime_str(self.dateTimeAdded)
  1375. @property
  1376. def ledger_enable(self):
  1377. return self.result in [self.PayResult.SUCCESS, self.PayResult.REFUNDING]
  1378. @property
  1379. def is_ledgered(self):
  1380. """
  1381. 判断该订单是否已经 分账获取了收益
  1382. :return:
  1383. """
  1384. if self.result != self.PayResult.SUCCESS:
  1385. # 失败订单肯定是没有分账的
  1386. return False
  1387. if self.isAllocatedCardMoney:
  1388. return True
  1389. # 没有该字段的订单 直接进行一次检查 看看有没有分账
  1390. from apps.web.common.proxy import ClientDealerIncomeModelProxy
  1391. try:
  1392. ledgerRecord = ClientDealerIncomeModelProxy.get_one(ref_id = self.id)
  1393. except Exception as e:
  1394. logger.exception(e)
  1395. return False
  1396. return bool(ledgerRecord)
  1397. @property
  1398. def notLedgerDesc(self):
  1399. """
  1400. 没有分账的描述值 12小时 72小时分界线
  1401. :return:
  1402. """
  1403. # 未超过 12 小时的情况 可以当成是订单尚未结束
  1404. if self.dateTimeAdded + datetime.timedelta(hours = 12) > datetime.datetime.now():
  1405. desc = u"订单尚未获取收益,可能设备正在运行"
  1406. # 超过 12 小时但是 没有到 72 小时的情况 告知订单可能可能出现了某种问题 等待系统监测自动结账
  1407. elif self.dateTimeAdded + datetime.timedelta(hours = 72) > datetime.datetime.now():
  1408. desc = u"订单尚未获取收益,可能由于网络原因造成订单未关闭,系统将于{}小时后强制分账,请勿担心!".format(
  1409. 72 - (datetime.datetime.now().hour - self.dateTimeAdded.hour))
  1410. # 超过72小时还没有分账的 问题单
  1411. else:
  1412. desc = u"订单尚未获取收益,请联系平台运营人员进行处理"
  1413. return desc
  1414. @property
  1415. def used_port(self):
  1416. if self.attachParas and 'chargeIndex' in self.attachParas and self.attachParas['chargeIndex']:
  1417. return str(self.attachParas['chargeIndex'])
  1418. return ''
  1419. @property
  1420. def startKey(self):
  1421. return self.attachParas.get('startKey', None)
  1422. @property
  1423. def refundOrders(self):
  1424. return RefundMoneyRecord.objects.filter(rechargeObjId = self.id)
  1425. @property
  1426. def has_refund_order(self):
  1427. if self.refundOrders:
  1428. return True
  1429. else:
  1430. return False
  1431. @property
  1432. def is_temp_package(self):
  1433. return 'isTempPackage' in self.attachParas and self.attachParas['isTempPackage'] is True
  1434. @property
  1435. def to_dict_obj(self):
  1436. return RechargeRecordDict({
  1437. 'via': self.via,
  1438. 'money': self.money,
  1439. 'coins': self.coins,
  1440. 'attachParas': self.attachParas,
  1441. 'extraInfo': self.extraInfo,
  1442. 'gateway': self.my_gateway,
  1443. 'payAppType': self.pay_app_type,
  1444. 'payGatewayKey': self.pay_gateway_key
  1445. })
  1446. @classmethod
  1447. def get_record(cls, order_no, dealer_id = None):
  1448. # type: (str, str)->Optional[RechargeRecord]
  1449. # TODO 数据库分片修改
  1450. # record = RechargeRecord.objects(ownerId = dealer_id, orderNo = order_no).first()
  1451. record = cls.objects(orderNo = order_no).first()
  1452. return record
  1453. @classmethod
  1454. def calc_count(cls, filter, group = None, limit = None, sort = None, allowDiskUse = False):
  1455. # type: (dict, dict, int, dict, bool)->int
  1456. pipeline = []
  1457. if filter:
  1458. pipeline.append({
  1459. '$match': filter
  1460. })
  1461. if group:
  1462. pipeline.append({
  1463. '$group': group
  1464. })
  1465. if limit:
  1466. pipeline.append({
  1467. '$limit': limit
  1468. })
  1469. if sort:
  1470. pipeline.append({
  1471. '$sort': sort
  1472. })
  1473. pipeline.append({'$count': 'count'})
  1474. result = cls.get_collection().aggregate(pipeline, allowDiskUse = allowDiskUse)
  1475. return next(result, {}).get('count', 0)
  1476. @classmethod
  1477. def get_recharged_records(cls, **kwargs):
  1478. return cls.objects(via = 'recharge', result = cls.PayResult.SUCCESS, **kwargs).order_by('-dateTimeAdded')
  1479. @classmethod
  1480. def get_card_recharged_records(cls, **kwargs):
  1481. return cls.objects(via = 'chargeCard', result = cls.PayResult.SUCCESS, **kwargs).order_by('-dateTimeAdded')
  1482. @classmethod
  1483. def issue_pay_order(cls, context, gateway, **payload):
  1484. payload.update({
  1485. "openId": context.user.openId,
  1486. "nickname": context.user.nickname,
  1487. 'gateway': gateway.gateway_type,
  1488. 'payAppType': gateway.pay_app_type,
  1489. 'payGatewayKey': gateway.gateway_key,
  1490. 'withdrawSourceKey': gateway.withdraw_source_key()
  1491. })
  1492. if gateway.pay_app_type in [PayAppType.WECHAT, PayAppType.ALIPAY]:
  1493. payload['extraInfo'].update({
  1494. 'payOpenId': context.user.get_bound_pay_openid(gateway.bound_openid_key)
  1495. })
  1496. if "result" not in payload:
  1497. payload["result"] = cls.PayResult.UNPAY
  1498. order = cls(payload)
  1499. return order.save()
  1500. @classmethod
  1501. def issue_from_auto_sim_order(cls, owner, sim_recharge, device, group):
  1502. # type:(Dealer, DealerRechargeRecord, Device, GroupDict)->RechargeRecord
  1503. payload = {
  1504. 'openId': owner.username,
  1505. 'nickname': owner.nickname,
  1506. 'orderNo': sim_recharge.orderNo,
  1507. 'ownerId': sim_recharge.dealerId,
  1508. 'money': -RMB(sim_recharge.totalFee/100),
  1509. 'coins': VirtualCoin(0),
  1510. 'subject': sim_recharge.subject,
  1511. 'result': cls.PayResult.SUCCESS,
  1512. 'via': RechargeRecordVia.AutoSim,
  1513. 'finishedTime': sim_recharge.finishedTime,
  1514. 'gateway': sim_recharge.my_gateway,
  1515. 'payAppType': sim_recharge.pay_app_type,
  1516. 'payGatewayKey': sim_recharge.pay_gateway_key,
  1517. 'isAllocatedCardMoney': True,
  1518. 'devNo': device.devNo,
  1519. 'devType': device.devTypeName,
  1520. 'devTypeName': device.devTypeName,
  1521. 'devTypeCode': device.devTypeCode,
  1522. 'logicalCode': device.logicalCode,
  1523. 'groupId': device.groupId,
  1524. 'address': group.address,
  1525. 'groupNumber': device.groupNumber,
  1526. 'groupName': group.groupName
  1527. }
  1528. record = cls(**payload).save()
  1529. record.group = group
  1530. return record
  1531. @classmethod
  1532. def viaText(cls, via, quickPay):
  1533. if via == RechargeRecordVia.Balance:
  1534. if quickPay:
  1535. return u'快捷支付'
  1536. else:
  1537. return u'优惠充值'
  1538. else:
  1539. return RECHARGE_RECORD_VIA_TRANSLATION[via]
  1540. @classmethod
  1541. def gatewayText(cls, gateway):
  1542. return APP_PLATFORM_TYPE_TRANSLATION.get(gateway, u'未知')
  1543. @classmethod
  1544. def from_feedback(cls, feedback, refundCoins):
  1545. # type: (FeedBack, VirtualCoin)->RechargeRecord
  1546. return cls(
  1547. orderNo = str(uuid.uuid4()),
  1548. coins = refundCoins,
  1549. money = RMB(0),
  1550. openId = feedback.openId,
  1551. groupId = feedback.groupId,
  1552. devNo = feedback.devNo,
  1553. ownerId = feedback.ownerId,
  1554. groupName = feedback.groupName,
  1555. groupNumber = feedback.groupNumber,
  1556. address = feedback.address,
  1557. wxOrderNo = u'老板退币',
  1558. devTypeName = feedback.devTypeName,
  1559. nickname = feedback.nickname,
  1560. result = cls.PayResult.SUCCESS,
  1561. via = 'refund')
  1562. @classmethod
  1563. def issue_pay_record(cls, context, **payload):
  1564. # type: (OrderBuilderContext, Dict)->RechargeRecord
  1565. user = context.user
  1566. payment_gateway = context.pay_gateway # type: PaymentGateway
  1567. payload.update({
  1568. 'openId': user.openId,
  1569. 'nickname': user.nickname,
  1570. 'gateway': payment_gateway.gateway_type,
  1571. 'payAppType': payment_gateway.pay_app_type,
  1572. 'payGatewayKey': payment_gateway.gateway_key,
  1573. 'withdrawSourceKey': payment_gateway.withdraw_source_key()
  1574. })
  1575. if 'attachParas' not in payload:
  1576. payload['attachParas'] = {}
  1577. if 'extraInfo' not in payload:
  1578. payload['extraInfo'] = {}
  1579. if payment_gateway.pay_app_type in [PayAppType.WECHAT, PayAppType.ALIPAY]:
  1580. payload['extraInfo'].update({
  1581. 'payOpenId': user.get_bound_pay_openid(payment_gateway.bound_openid_key)
  1582. })
  1583. if "result" not in payload:
  1584. payload['result'] = cls.PayResult.UNPAY
  1585. record = cls(**payload) # type: RechargeRecord
  1586. return record.save()
  1587. def to_dict(self, json_safe=False):
  1588. money = str(self.money) if json_safe else self.money
  1589. return {
  1590. 'ownerId': self.ownerId,
  1591. 'nickname': self.nickname,
  1592. 'money': money,
  1593. 'gateway': self.my_gateway,
  1594. 'via': self.via,
  1595. }
  1596. def to_json_dict(self):
  1597. return self.to_dict(json_safe = True)
  1598. def succeed(self, wxOrderNo, **kwargs):
  1599. # 这里和数据库强相关, 通过判断是否正确更新记录
  1600. payload = {
  1601. 'wxOrderNo': wxOrderNo
  1602. }
  1603. if kwargs:
  1604. payload.update(kwargs)
  1605. payload.update({'result': self.PayResult.SUCCESS})
  1606. result = self.get_collection().update_one(
  1607. filter = {'_id': ObjectId(self.id), 'result': {'$ne': self.PayResult.SUCCESS}},
  1608. update = {'$set': payload},
  1609. upsert = False)
  1610. return result.matched_count == 1
  1611. def fail(self, **kwargs):
  1612. self.update(result = self.PayResult.FAILED, **kwargs)
  1613. return self.reload()
  1614. def cancel(self):
  1615. payload = {
  1616. 'result': self.PayResult.CANCEL,
  1617. 'finishedTime': datetime.datetime.now()
  1618. }
  1619. result = self.get_collection().update_one(
  1620. filter = {'_id': ObjectId(self.id), 'result': self.PayResult.UNPAY},
  1621. update = {'$set': payload},
  1622. upsert = False)
  1623. return result.matched_count == 1
  1624. def is_success(self):
  1625. return self.result == self.PayResult.SUCCESS
  1626. def is_fail(self):
  1627. return self.result == self.PayResult.FAILED
  1628. def to_detail(self):
  1629. # type: ()->dict
  1630. """
  1631. :return:
  1632. """
  1633. return {
  1634. 'id': self.id,
  1635. 'rechargeTradeNo': self.orderNo,
  1636. 'payResult': self.result,
  1637. 'ownerId': self.ownerId,
  1638. 'gateway': self.my_gateway,
  1639. 'openId': self.openId,
  1640. 'groupId': self.groupId,
  1641. 'userNickname': self.nickname,
  1642. 'orderAmount': self.money,
  1643. 'coins': self.coins,
  1644. 'devNo': self.devNo,
  1645. 'logicalCode': self.logicalCode,
  1646. 'devTypeName': self.dev_type_name,
  1647. 'groupName': self.groupName,
  1648. 'groupNumber': self.groupNumber,
  1649. 'address': self.address,
  1650. 'createdTime': self.dateTimeAdded,
  1651. 'completionTime': self.finished_time,
  1652. 'via': self.via,
  1653. 'isQuickPay': self.isQuickPay,
  1654. 'startKey': self.attachParas.get('startKey', None),
  1655. # 以下字段和以前做兼容
  1656. 'outTradeNo': self.orderNo,
  1657. 'finishedTime': self.finished_time,
  1658. 'gatewayTradeNo': self.wxOrderNo,
  1659. 'result': self.result
  1660. }
  1661. def to_dict_for_super_admin(self):
  1662. from apps.web.common.proxy import ClientConsumeModelProxy
  1663. orderResult = ''
  1664. isNormalDesc = ''
  1665. via = ''
  1666. if self.result == 'unPay':
  1667. orderResult = u'未付款'
  1668. elif self.result == 'failed':
  1669. orderResult = u'充值失败'
  1670. elif self.result == 'success':
  1671. orderResult = u'充值成功'
  1672. if self.via == '':
  1673. via = u'充值金币'
  1674. elif self.via == '':
  1675. via = u'充值卡'
  1676. if 'consumeRecordId' in self.attachParas:
  1677. # 部分后支付的设备类型, 在支付的时候会塞入消费单的id
  1678. consumeRcd = ClientConsumeModelProxy.get_one(
  1679. shard_filter = {'ownerId': self.ownerId},
  1680. id = self.attachParas.get("consumeRecordId")) # type: ConsumeRecord
  1681. elif 'startKey' in self.attachParas and self.attachParas['startKey']:
  1682. consumeRcd = ClientConsumeModelProxy.get_one(
  1683. shard_filter = {'ownerId': self.ownerId},
  1684. foreign_id = str(self.id),
  1685. startKey = self.attachParas['startKey']) # type: ConsumeRecord
  1686. else:
  1687. consumeRcd = None
  1688. if consumeRcd is not None:
  1689. if consumeRcd.isNormal is True:
  1690. isNormalDesc = u'消费成功'
  1691. else:
  1692. isNormalDesc = u'消费失败'
  1693. return {
  1694. 'wxOrderNo': self.wxOrderNo,
  1695. 'orderNo': self.orderNo,
  1696. 'orderResult': orderResult,
  1697. 'rechargeUser': self.nickname + ' ' + self.openId,
  1698. 'rechargeType': via,
  1699. 'money': str(self.money),
  1700. 'coins': str(self.coins),
  1701. 'orderDetail': self.attachParas,
  1702. 'orderCreateTime': self.dateTimeAdded.strftime('%Y-%m-%d %H:%M:%S'),
  1703. 'logicalCode': self.logicalCode,
  1704. 'devNo': self.devNo,
  1705. 'devTypeCode': self.dev_type_code,
  1706. 'consumeOrderNo': consumeRcd.orderNo if consumeRcd else '',
  1707. 'consumeUser': consumeRcd.nickname + ' ' + consumeRcd.openId if consumeRcd else '',
  1708. 'consumeCoins': str(consumeRcd.coin) if consumeRcd else '',
  1709. 'consumeResult': isNormalDesc if consumeRcd else '',
  1710. 'failedReason': consumeRcd.errorDesc if consumeRcd else '',
  1711. 'consumeOrderCreateTime': consumeRcd.dateTimeAdded.strftime('%Y-%m-%d %H:%M:%S') if consumeRcd else '',
  1712. 'consumeDetail': consumeRcd.attachParas if consumeRcd else {},
  1713. 'consumeDict': consumeRcd.servicedInfo if consumeRcd else {}
  1714. }
  1715. def set_ledgered(self):
  1716. try:
  1717. modified = self.update(isAllocatedCardMoney = True)
  1718. if not modified:
  1719. logger.error('failed to update isAllocatedCardMoney for record=%r' % (self,))
  1720. except Exception as e:
  1721. logger.exception(e)
  1722. def new_refund_cash_order(self, refund_order): # type: (RefundMoneyRecord)->RechargeRecord
  1723. payload = {}
  1724. for field in self.__class__._fields_ordered:
  1725. if field in ['id', 'orderNo', 'wxOrderNo', 'transactionId', 'dateTimeAdded', 'ledgerStatus',
  1726. 'finishedTime', 'notifiedTime', 'operator', 'devTypeName', 'money', 'coins',
  1727. 'selectedQuickPayPackageId', 'devType', 'attachParas', 'isAllocatedCardMoney', 'subject']:
  1728. continue
  1729. payload.update({
  1730. field: getattr(self.payOrder, field)
  1731. })
  1732. payload.update({
  1733. 'orderNo': refund_order.orderNo,
  1734. 'money': -refund_order.money,
  1735. 'coins': -refund_order.coins,
  1736. 'devTypeName': self.dev_type_name,
  1737. 'via': RechargeRecordVia.RefundCash,
  1738. 'isAllocatedCardMoney': False,
  1739. 'gateway': self.my_gateway,
  1740. 'payGatewayKey': self.pay_gateway_key,
  1741. 'payAppType': self.pay_app_type,
  1742. 'withdrawSourceKey': self.withdraw_source_key,
  1743. 'result': self.PayResult.REFUNDING,
  1744. 'extraInfo': {
  1745. 'refPay': {'objId': str(self.id)}
  1746. },
  1747. 'subject': '{} {}'.format(self.subject, u'退费')
  1748. })
  1749. refund_order_record = self.__class__(**payload).save() # type: RechargeRecord
  1750. refund_order.refund_order_record = refund_order_record
  1751. self.update(push__extraInfo__refRefund = {
  1752. 'objId': str(refund_order_record.id),
  1753. 'money': refund_order.money.mongo_amount,
  1754. 'coins': refund_order.coins.mongo_amount
  1755. })
  1756. return refund_order_record
  1757. def is_refund_available(self, customer): # type:(CapitalUser) -> bool
  1758. if customer.role == ROLE.dealer:
  1759. if self.ownerId != str(customer.id):
  1760. logger.warning('is not my order. {} != {}'.format(self.ownerId, str(customer.id)))
  1761. return False
  1762. if self.via not in [USER_RECHARGE_TYPE.RECHARGE, USER_RECHARGE_TYPE.RECHARGE_CASH]:
  1763. return False
  1764. return True
  1765. return False
  1766. class Dispute(EmbeddedDocument):
  1767. feedback_id = ObjectIdField()
  1768. #
  1769. # class ConsumeRecord(OrderRecordBase):
  1770. # """
  1771. # 用户的消费订单记录
  1772. # """
  1773. #
  1774. # if TYPE_CHECKING:
  1775. # state = None
  1776. #
  1777. # def is_waitPay(self): pass
  1778. #
  1779. # def p_to_f(self): pass
  1780. #
  1781. # def c_to_s(self): pass
  1782. #
  1783. # def s_to_e(self): pass
  1784. #
  1785. # def is_end(self): pass
  1786. #
  1787. # def e_to_p(self): pass
  1788. #
  1789. # class Status(IterConstant):
  1790. # CREATED = 'created'
  1791. # WAITING = 'waiting'
  1792. #
  1793. # RUNNING = 'running'
  1794. # END = "end" # 结束运行的状态
  1795. # FINISHED = 'finished'
  1796. # WAIT = 'waitPay'
  1797. # TIMEOUT = 'timeout'
  1798. # FAILURE = 'failure'
  1799. # UNKNOWN = 'unknown'
  1800. #
  1801. # # TODO: 以下三个状态蓝牙使用过, 后续整改
  1802. # INIT = 'init'
  1803. # CHARGED = 'Charged'
  1804. # REFUND = 'Refund'
  1805. #
  1806. # orderNo = StringField(verbose_name = "订单号", default = "")
  1807. # ownerId = StringField(verbose_name = '经销商ID', default = "")
  1808. # price = MonetaryField(verbose_name = "消耗的钱(硬币)", default = RMB('0.00'))
  1809. #
  1810. # #: 在部分场合,设备会失败启动,但这时应该添加该字段为False方便追溯
  1811. # isNormal = BooleanField(verbose_name = "设备是否正常启动", default = True)
  1812. # errorDesc = StringField(verbose_name = u"错误描述", default = '')
  1813. #
  1814. # remarks = StringField(verbose_name = u"备注(默认值不能为None, 否则在使用in的时候出错)", default = '')
  1815. #
  1816. # dateTimeCompletion = DateTimeField(verbose_name = '结单时间', default = None)
  1817. #
  1818. # sequanceNo = StringField(verbose_name = u'消费的流水号,用于和设备上的订单关联', default = None)
  1819. #
  1820. # status = StringField(verbose_name = u'状态', default = '')
  1821. #
  1822. # startKey = StringField(verbose_name = u'启动key')
  1823. #
  1824. # attachParas = DictField(verbose_name = u'用户选择的额外参数', default = {}) # 比如端口、电池类型等信息
  1825. #
  1826. # startTime = DateTimeField(verbose_name = u'设备启动时间', default = None)
  1827. # finishedTime = DateTimeField(verbose_name = u'设备结束时间', default = None)
  1828. #
  1829. # servicedInfo = DictField(verbose_name = u'服务内容', default = {})
  1830. # desc = StringField(verbose_name = u'描述', default = None)
  1831. #
  1832. # feedbackId = ObjectIdField(verbose_name = u'用户反馈ID', default = None)
  1833. #
  1834. # refunded = BooleanField(verbose_name = u'是否已经处理完退费', default = None)
  1835. #
  1836. # aggInfo = DictField(verbose_name = u'聚合信息', default = {})
  1837. #
  1838. # rechargeRcdId = StringField(verbose_name = u'如果是快捷支付,可以关联充值订单(比如昌原,没有金币的概念,需要查询每单的花费)', default = None)
  1839. #
  1840. # # {via: coin/card/virtualCard/free/netPay itemId: 相应的id money: 现金金额 coins: 金币支付金额}
  1841. # payInfo = ListField(verbose_name = u'支付的相关途径', default = [])
  1842. # paymentInfo = DictField(verbose_name = u'支付的相关途径', default = {})
  1843. #
  1844. # package = DictField(verbose_name = u'套餐', default = {})
  1845. #
  1846. # # role: 'main', list<id>; rule: 'sub', id
  1847. # association = DictField(verbose_name = u'关联单', default = {})
  1848. #
  1849. # # 以下字段不在使用
  1850. # leftBalance = StringField(verbose_name = u"心跳机制返回的剩余金额", default = None)
  1851. # rechargeRcdIds = ListField(verbose_name = u'现金充值的订单id', default = None)
  1852. #
  1853. # search_fields = ('openId', 'devNo', 'orderNo', 'remarks', 'address', 'groupName', 'logicalCode')
  1854. #
  1855. # _shard_key = ('ownerId', 'dateTimeAdded')
  1856. #
  1857. # _origin_meta = {
  1858. # "collection": "ConsumeRecord",
  1859. # "db_alias": "default"
  1860. # }
  1861. #
  1862. # meta = _origin_meta
  1863. #
  1864. # # 订单的5种基础状态 后续有需要可以添加
  1865. # states = ["created", "wait", "running", "finished", "waitPay", "end"]
  1866. #
  1867. # # 两种支付流程各个状态的切换
  1868. # # created ---> start ---> end ---> waitPay ---> finished
  1869. # # created ---> waitPay ---> start ---> end ---> finished
  1870. #
  1871. # def __init__(self, *args, **kwargs):
  1872. # # super(ConsumeRecord, self).__init__(*args, **kwargs)
  1873. # Searchable.__init__(self, *args, **kwargs)
  1874. #
  1875. # # 这个目前只有后付费的订单支持这个 做了一个简易的开关
  1876. # # mongoDB 使用save的时候 会有created 以及 update的区别 已知的情况是 在save_created的时候,如果运行了_add_state_machine 会造成数据插入异常
  1877. # if self.id and self.devTypeCode and self.status in ConsumeRecord.states:
  1878. # self._add_state_machine()
  1879. #
  1880. # def __str__(self):
  1881. # return '{}<id={} orderNo={}>'.format(self.__class__.__name__, str(self.id), self.orderNo)
  1882. #
  1883. # @property
  1884. # def owner(self):
  1885. # from apps.web.dealer.models import Dealer
  1886. #
  1887. # _attr_name = '__my_owner__'
  1888. #
  1889. # if hasattr(self, _attr_name):
  1890. # return getattr(self, _attr_name)
  1891. # else:
  1892. # if not self.ownerId:
  1893. # return None
  1894. # else:
  1895. # dealer = Dealer.objects(id = self.ownerId).first()
  1896. # setattr(self, _attr_name, dealer)
  1897. # return dealer
  1898. #
  1899. # def _add_state_machine(self):
  1900. # """
  1901. # 为订单对象创建状态机 并初始化状态机
  1902. # :return:
  1903. # """
  1904. # self._machine = Machine(model = self, states = ConsumeRecord.states, initial = self.status,
  1905. # after_state_change = ["_after_state_change"], queued = True, auto_transitions = False)
  1906. #
  1907. # # 订单创建之后 需要将订单转换到下一个状态,订单的转换状态是根据订单是否正常来切换的
  1908. # self._machine.add_transition("c_to_s", "created", "running", conditions = ['isNormal'])
  1909. # self._machine.add_transition("c_to_s", "created", "finished", unless = ["isNormal"])
  1910. #
  1911. # # 订单运行状态到订单停止运行状态
  1912. # self._machine.add_transition("s_to_e", "running", "end", after = ["_pay_coins", "e_to_f"])
  1913. # # 订单虚拟货币支付, 支付成功之后将订单状态切换为结束,结束之后需要有数据汇总的操作 支付不成功智能等待用户主动现金支付
  1914. # self._machine.add_transition("e_to_f", "end", "finished", conditions = ["_isPaid"],
  1915. # after = ["_stats_info", "_do_dev_callback"])
  1916. # self._machine.add_transition("e_to_e", "end", "end", unless = ["_isPaid"])
  1917. # self._machine.add_transition("e_to_p", "end", "waitPay", after = ["_add_pay_time_point"])
  1918. #
  1919. # # 订单支后,轮询的回调函数,一旦轮询到订单支付成功, 则将状态改变 由于不知道end状态是会首先变成paying再变成finished 还是直接由end变为finished, 接受的初始状态实体应该是一个列表
  1920. # self._machine.add_transition("p_to_f", "waitPay", "finished", conditions = ["_isRecharge", "_isNotPaid"],
  1921. # after = ["_pay_cash", "_do_dev_callback"])
  1922. # self._machine.add_transition("p_to_f", "waitPay", "end", unless = ["_isRecharge", "_isPaid"],
  1923. # after = ["_clear_recharge"])
  1924. #
  1925. # def _after_state_change(self):
  1926. # """
  1927. # 订单使用状态机切换状态之后,将订单的status相应修改,并且重载一下
  1928. # :return:
  1929. # """
  1930. # self.update(status = self.state)
  1931. # self.reload()
  1932. #
  1933. # def _pay_coins(self):
  1934. # """
  1935. # 订单运行结束之后,直接尝试使用虚拟货币对订单进行支付
  1936. # :return:
  1937. # """
  1938. #
  1939. # # 已经支付了或者已经有充值订单相关联
  1940. # if self._isPaid or self._isRecharge:
  1941. # return
  1942. #
  1943. # # 尝试使用包月卡支付
  1944. # dev = Device.get_dev(self.devNo)
  1945. # package = dev.get("washConfig", dict()).get(self.attachParas.get("packageId", "1"))
  1946. #
  1947. # monthlyPackage = MonthlyPackage.get_enable_one(self.openId, self.groupId)
  1948. # if monthlyPackage and monthlyPackage.is_suit_package(package):
  1949. # monthlyPackage.deduct(self)
  1950. # self.paymentInfo = { # zjl [] -> {}
  1951. # "via": "monthlyPackage",
  1952. # "itemId": str(monthlyPackage.id),
  1953. # "money": RMB(0).mongo_amount,
  1954. # "coins": VirtualCoin(0).mongo_amount
  1955. # } # zjl [] -> {}
  1956. # self.save()
  1957. # return
  1958. #
  1959. # dev = Device.get_dev(self.devNo) # type: DeviceDict
  1960. # group = Group.get_group(self.groupId)
  1961. # dealer = dev.owner # type: Dealer
  1962. #
  1963. # devTypeCode = dev.get("devType", dict()).get("code", "")
  1964. #
  1965. # # 尝试使用虚拟卡支付
  1966. # # package = {"unit": u"币", "count": self.coin}
  1967. # vCards = UserVirtualCard.get_user_cards(self.openId, self.groupId, self.ownerId)
  1968. # for vCard in vCards:
  1969. # if vCard.support_dev_type(devTypeCode) and vCard.can_use_today(package):
  1970. # # TODO zjl 先简陋处理一下 后续将其更换位置
  1971. # from apps.web.user.utils import vCard_pay_coin
  1972. # vCard_pay_coin(self, vCard, dev, group, package, self.attachParas) # type: VCardConsumeRecord
  1973. # return
  1974. #
  1975. # user = MyUser.objects(openId = self.openId, groupId = self.groupId).first() # type: MyUser
  1976. # balance = user.calc_currency_balance(dealer, group)
  1977. #
  1978. # if balance >= VirtualCoin(self.coin):
  1979. # # 金币足够支付的情况下 直接使用金币支付
  1980. # from apps.web.user.utils import user_pay_coin
  1981. # user_pay_coin(self.openId, self.devNo, self.groupId, self.ownerId, self.coin)
  1982. # self.paymentInfo = {
  1983. # "via": "coin",
  1984. # "itemId": "",
  1985. # "money": RMB(0).mongo_amount,
  1986. # "coins": VirtualCoin(self.coin).mongo_amount
  1987. # }
  1988. # self.save()
  1989. #
  1990. # def _clear_recharge(self):
  1991. # self.rechargeRcdId = ""
  1992. # self.save()
  1993. #
  1994. # def _pay_cash(self):
  1995. # # 这一步的时候实际上已经支付成功了,所需要做的步骤就是将支付的信息加载进 payInfo 字段中
  1996. # self.paymentInfo = {
  1997. # "via": "cash",
  1998. # "itemId": self.rechargeRcdId,
  1999. # "money": RMB(self.money).mongo_amount,
  2000. # "coins": VirtualCoin(0).mongo_amount
  2001. # }
  2002. # self.save()
  2003. #
  2004. # @property
  2005. # def _isPaid(self):
  2006. # """
  2007. # 检查用户是否已经为此订单付了钱 无论是此订单关联的recharge 还是使用金币支付 或者是使用虚拟卡支付
  2008. # :return:
  2009. # """
  2010. # return any([self.payInfo, self.paymentInfo])
  2011. #
  2012. # @property
  2013. # def _isNotPaid(self):
  2014. # return not self._isPaid
  2015. #
  2016. # @property
  2017. # def _isRecharge(self):
  2018. # """
  2019. # 检测订单是否已经与充值订单相关联
  2020. # :return:
  2021. # """
  2022. # rechargeRcdId = self.recharge_record_id
  2023. #
  2024. # if not rechargeRcdId:
  2025. # return False
  2026. #
  2027. # try:
  2028. # from apps.web.common.proxy import ClientRechargeModelProxy
  2029. # record = ClientRechargeModelProxy.get_one(
  2030. # shard_filter = {'ownerId': self.ownerId},
  2031. # id = str(rechargeRcdId)) # type: RechargeRecord
  2032. # except DoesNotExist:
  2033. # return False
  2034. # except Exception as e:
  2035. # logger.exception(e)
  2036. # return False
  2037. #
  2038. # if record.result == RechargeRecord.PayResult.SUCCESS:
  2039. # # 下面还要用的,防止重复查询
  2040. # setattr(self, "_rechargeRcd", record)
  2041. # return True
  2042. # else:
  2043. # return False
  2044. #
  2045. # def _do_dev_callback(self):
  2046. # dev = Device.get_dev(self.devNo)
  2047. # box = ActionDeviceBuilder.create_action_device(dev)
  2048. # if box.isHaveCallback():
  2049. # try:
  2050. # box.do_callback(self)
  2051. # except Exception:
  2052. # self.attachParas.update({"callbackResult": False})
  2053. # self.save()
  2054. # self.attachParas.update({"callbackResult": True})
  2055. # self.save()
  2056. #
  2057. # def _add_pay_time_point(self):
  2058. #
  2059. # self.attachParas['payTimePoint'] = datetime.datetime.now()
  2060. # self.save()
  2061. #
  2062. # @property
  2063. # def is_from_api(self):
  2064. # return False
  2065. #
  2066. # @property
  2067. # def is_on_point(self):
  2068. # return False
  2069. #
  2070. # @property
  2071. # def is_from_user(self):
  2072. # return True
  2073. #
  2074. # @property
  2075. # def used_port(self):
  2076. # port = self.attachParas['chargeIndex'] if (
  2077. # self.attachParas and self.attachParas.has_key('chargeIndex')
  2078. # and self.attachParas['chargeIndex']) else -1
  2079. # try:
  2080. # return int(port)
  2081. # except:
  2082. # return port
  2083. #
  2084. # @property
  2085. # def created_date(self):
  2086. # return self.dateTimeAdded
  2087. #
  2088. # @property
  2089. # def completion_date(self):
  2090. # if self.dateTimeCompletion:
  2091. # return self.dateTimeCompletion
  2092. # else:
  2093. # return self.finishedTime
  2094. #
  2095. # @property
  2096. # def device_start_time(self):
  2097. # if self.startTime:
  2098. # return self.startTime
  2099. # else:
  2100. # return self.dateTimeAdded
  2101. #
  2102. # @property
  2103. # def device_finished_time(self):
  2104. # if self.finishedTime:
  2105. # return self.finishedTime
  2106. # else:
  2107. # return ''
  2108. #
  2109. # @classmethod
  2110. # def new_one(cls, order_no, user, device, group, package, attach_paras, pay_info = None,
  2111. # start_key = None, sequence_no = None):
  2112. # # type: (str, MyUser, DeviceDict, GroupDict, Dict, Dict, Dict, str, str)->ConsumeRecord
  2113. #
  2114. # order = cls(
  2115. # orderNo = order_no,
  2116. # time = str(datetime.datetime.now().strftime("%Y-%m-%d %H:%M:%S")),
  2117. # openId = user.openId,
  2118. # nickname = user.nickname,
  2119. # coin = VirtualCoin(0),
  2120. # money = RMB(0),
  2121. # devNo = device.devNo,
  2122. # logicalCode = device.logicalCode,
  2123. # devTypeCode = device.devTypeCode,
  2124. # devTypeName = device.devTypeName,
  2125. # groupId = device.groupId,
  2126. # ownerId = device.ownerId,
  2127. # address = group.address,
  2128. # groupNumber = device.groupNumber,
  2129. # groupName = group.groupName,
  2130. # startKey = start_key,
  2131. # isNormal = False,
  2132. # errorDesc = '',
  2133. # sequanceNo = sequence_no,
  2134. # attachParas = attach_paras,
  2135. # status = cls.Status.CREATED,
  2136. # paymentInfo = pay_info,
  2137. # package = package)
  2138. #
  2139. # remarks = None
  2140. # money = RMB(0)
  2141. # coins = VirtualCoin(0)
  2142. #
  2143. # if pay_info:
  2144. # if pay_info['via'] == 'free':
  2145. # remarks = u'免费'
  2146. # elif pay_info['via'] == 'virtualCard':
  2147. # remarks = u'虚拟卡券'
  2148. # elif pay_info['via'] == 'coins':
  2149. # coins = pay_info['coins']
  2150. # remarks = u'余额支付'
  2151. # elif pay_info['via'] == 'card':
  2152. # coins = pay_info['coins']
  2153. # money = pay_info['coins']
  2154. # remarks = u'刷卡消费'
  2155. # else:
  2156. # money = pay_info['money']
  2157. # coins = pay_info['coins']
  2158. # remarks = u'网络支付'
  2159. #
  2160. # order.paymentInfo = pay_info
  2161. # order.remarks = remarks
  2162. # order.money = money
  2163. # order.coin = coins
  2164. #
  2165. # order.save()
  2166. #
  2167. # return order
  2168. #
  2169. # def update_service_info(self, consumeDict, isFinished=True):
  2170. # # type:(dict, bool)->None
  2171. #
  2172. # try:
  2173. # trans_consume_dict = {}
  2174. #
  2175. # for key, value in consumeDict.iteritems():
  2176. # if value is None:
  2177. # logger.debug('{} is none.'.format(key))
  2178. # continue
  2179. #
  2180. # if hasattr(value, 'mongo_amount'):
  2181. # trans_consume_dict[key] = value.mongo_amount
  2182. # else:
  2183. # trans_consume_dict[key] = value
  2184. #
  2185. # self.servicedInfo.update(trans_consume_dict)
  2186. # self.save()
  2187. #
  2188. # if not isFinished:
  2189. # return
  2190. #
  2191. # status = self.update_agg_info(trans_consume_dict)
  2192. # if status:
  2193. # record_consumption_stats(self)
  2194. # else:
  2195. # logger.error(
  2196. # '[update_progress_and_consume_rcd] failed to update_agg_info record=%r' % (self,))
  2197. # except Exception, e:
  2198. # logger.exception('update agg info error=%s' % e)
  2199. #
  2200. # def update_for_end(self, serviceInfo = None, finishedTime=datetime.datetime.now()):
  2201. # """
  2202. # 订单切换到设备运行结束的状态
  2203. # :param serviceInfo:
  2204. # :param finishedTime
  2205. # :return:
  2206. # """
  2207. #
  2208. # self.status = self.Status.FINISHED
  2209. # self.finishedTime = to_datetime(finishedTime)
  2210. #
  2211. # serviceInfo = serviceInfo or dict()
  2212. #
  2213. # self.update_service_info(consumeDict=serviceInfo)
  2214. #
  2215. # @property
  2216. # def receiptDesc(self):
  2217. # rv = {
  2218. # 'orderNo': self.orderNo,
  2219. # 'createdTime': self.dateTimeAdded,
  2220. # 'payment': u'{} 金币'.format(str(VirtualCoin(self.coin)))
  2221. # }
  2222. #
  2223. # if self.paymentInfo:
  2224. # if self.paymentInfo['via'] == 'free':
  2225. # rv.update({'payment': u'本次免费'})
  2226. # return rv
  2227. # elif self.paymentInfo['via'] == 'cash':
  2228. # rv.update({'payment': u'{} 元'.format(self.paymentInfo['money'])})
  2229. # return rv
  2230. # elif self.paymentInfo['via'] == 'virtualCard':
  2231. # rv.update({'payment': u'虚拟卡抵扣'})
  2232. # return rv
  2233. # elif self.paymentInfo['via'] == 'yuchuanCard':
  2234. # rv.update({'payment': u'{} 元 宇川一卡通'.format(self.paymentInfo['money'])})
  2235. # return rv
  2236. #
  2237. # if self.remarks == u'虚拟卡消费':
  2238. # rv.update({'payment': u'虚拟卡抵扣'})
  2239. # return rv
  2240. #
  2241. # if self.remarks == u'包月卡消费':
  2242. # rv.update({'payment': u'包月卡抵扣'})
  2243. # return rv
  2244. #
  2245. # if self.remarks == 'YuChuanCard':
  2246. # rv.update({'payment': u'{} 元 宇川一卡通'.format(self.paymentInfo['money'])})
  2247. # return rv
  2248. #
  2249. # return rv
  2250. #
  2251. # def _stats_info(self):
  2252. # """
  2253. # 订单切换为最终完成状态之后 将订单的信息上报到数据中心进行统计
  2254. # :return:
  2255. # """
  2256. # duration = self._duration
  2257. # if duration:
  2258. # agg_info = {
  2259. # DEALER_CONSUMPTION_AGG_KIND.DURATION: duration,
  2260. # DEALER_CONSUMPTION_AGG_KIND.COIN: self.coin
  2261. # }
  2262. # else:
  2263. # agg_info = {
  2264. # DEALER_CONSUMPTION_AGG_KIND.COIN: self.coin
  2265. # }
  2266. #
  2267. # status = self.update_agg_info(agg_info)
  2268. # if status:
  2269. # record_consumption_stats(self)
  2270. #
  2271. # @property
  2272. # def _duration(self):
  2273. # """
  2274. # 计算订单的持续时长
  2275. # :return:
  2276. # """
  2277. # if self.state != "finished" or not self.finishedTime:
  2278. # return 0
  2279. #
  2280. # if 'duration' in self.servicedInfo and isinstance(self.servicedInfo['duration'], (float, int)):
  2281. # duration = self.servicedInfo['duration']
  2282. #
  2283. # else:
  2284. # # 计算时长不需要很精确,直接使用round算了
  2285. # duration = round((self.finishedTime - self.dateTimeAdded).total_seconds() / 60.0)
  2286. #
  2287. # return duration
  2288. #
  2289. # @property
  2290. # def virtual_card_id(self):
  2291. # if not self.paymentInfo:
  2292. # return None
  2293. #
  2294. # via = self.paymentInfo.get('via', None)
  2295. #
  2296. # if via == 'virtualCard':
  2297. # return self.paymentInfo['itemId']
  2298. # else:
  2299. # return None
  2300. #
  2301. # @property
  2302. # def recharge_record_id(self):
  2303. # if self.rechargeRcdId and str(self.rechargeRcdId) != 'None':
  2304. # return str(self.rechargeRcdId)
  2305. #
  2306. # if not self.paymentInfo:
  2307. # return None
  2308. #
  2309. # via = self.paymentInfo.get('via', None)
  2310. # if via in ['netPay', 'cash', 'coins']:
  2311. # if 'itemId' in self.paymentInfo and self.paymentInfo['itemId']:
  2312. # return str(self.paymentInfo['itemId'])
  2313. # else:
  2314. # return None
  2315. # else:
  2316. # return None
  2317. #
  2318. # @property
  2319. # def payTypeInfo(self):
  2320. # cardId = self.attachParas.get("cardId")
  2321. # vCardId = self.attachParas.get("vCardId")
  2322. # data = dict()
  2323. #
  2324. # if cardId:
  2325. # try:
  2326. # cardNo = Card.objects.get(id = cardId).cardNo
  2327. # except Exception as e:
  2328. # logger.exception(e)
  2329. # else:
  2330. # data.update({"cardId": cardId, "cardNo": cardNo})
  2331. #
  2332. # if vCardId:
  2333. # try:
  2334. # vCardNo = UserVirtualCard.objects.get(id = vCardId).cardNo
  2335. # except Exception as e:
  2336. # logger.exception(e)
  2337. # else:
  2338. # data.update({"vCardId": vCardId, "vCardNo": vCardNo})
  2339. #
  2340. # return data
  2341. #
  2342. # @staticmethod
  2343. # def make_no(*args, **kwargs):
  2344. # timestamp = generate_timestamp_ex()
  2345. # random_int = random.randint(1, 1000)
  2346. # return "%d%03d" % (timestamp, random_int)
  2347. #
  2348. # @staticmethod
  2349. # def make_order_no(identifier, sub_type):
  2350. # return OrderNoMaker.make_order_no_32(identifier, OrderMainType.CONSUME, sub_type)
  2351. #
  2352. # @property
  2353. # def consume_type(self):
  2354. # if self.remarks:
  2355. # if u'刷卡' in self.remarks:
  2356. # return 'card'
  2357. # elif u'虚拟卡' in self.remarks:
  2358. # return 'mobile_vcard'
  2359. # elif u'投币或者刷卡消费' in self.remarks:
  2360. # return 'offline'
  2361. # elif u'包月卡' in self.remarks:
  2362. # return 'monthlyPackage'
  2363. #
  2364. # if self.recharge_record_id:
  2365. # return 'cash'
  2366. # else:
  2367. # return 'balance'
  2368. #
  2369. # def to_user_detail(self):
  2370. # """
  2371. # 消费记录对于用户端显示的详细信息
  2372. # :return:
  2373. # """
  2374. #
  2375. # data = {
  2376. # 'id': str(self.id),
  2377. #
  2378. # 'consumeType': self.consume_type,
  2379. #
  2380. # 'orderNo': self.orderNo,
  2381. #
  2382. # 'ownerId': self.ownerId,
  2383. #
  2384. # 'money': self.money,
  2385. #
  2386. # 'coins': self.coin,
  2387. #
  2388. # 'createdTime': self.created_date,
  2389. # 'completionTime': self.completion_date,
  2390. #
  2391. # 'startResult': 'success' if self.isNormal else 'failed',
  2392. # 'errorDesc': self.errorDesc,
  2393. #
  2394. # 'deviceStatTime': self.device_start_time,
  2395. # 'deviceFinishedTime': self.device_finished_time,
  2396. #
  2397. # 'reason': self.servicedInfo.get("reason", "")
  2398. # }
  2399. #
  2400. # data.update(self.device_identity_info)
  2401. #
  2402. # data.update(self.refund_info)
  2403. #
  2404. # if self.recharge_record_id:
  2405. # recharge_record = RechargeRecord.objects(id = self.recharge_record_id).first() # type: RechargeRecord
  2406. # if recharge_record:
  2407. # data.update({
  2408. # 'rechargeRcdId': str(recharge_record.id),
  2409. # 'rechargeTradeNo': recharge_record.orderNo,
  2410. # })
  2411. #
  2412. # # 兼容之前的订单 凡是没有订单状态机的订单,一律视为 之前的订单
  2413. # # 简陋 的实现以下订单状态改变
  2414. # if self.devTypeCode in [
  2415. # Const.DEVICE_TYPE_CODE_CABINET,
  2416. # Const.DEVICE_TYPE_CODE_CHARGE_XIAOKEDOU,
  2417. # Const.DEVICE_TYPE_CODE_CABINET_NEW,
  2418. # Const.DEVICE_TYPE_CODE_CHARGE_WEIFULE_CAR,
  2419. # Const.DEVICE_TYPE_CODE_CAR_WEIFULE_CHARGING_DOUB,
  2420. # Const.DEVICE_TYPE_CODE_CAR_WEIFULE_21KW,
  2421. # Const.DEVICE_TYPE_CODE_CAR_WEIFILE_HOME_JFPG,
  2422. # Const.DEVICE_TYPE_CODE_CAR_WEIFILE_HOME_DOUB_JFPG,
  2423. # ] + support_policy_device + support_policy_weifule:
  2424. # self.charge_recharge_status()
  2425. # try:
  2426. # data.update({"orderStatus": self.state})
  2427. # except Exception as e:
  2428. # logger.exception(e)
  2429. #
  2430. # if self.servicedInfo:
  2431. # self.servicedInfo.pop('refundedMoney', None)
  2432. # self.servicedInfo.pop('refundedCash', None)
  2433. # data.update(self.servicedInfo)
  2434. #
  2435. # data.update(self.refund_info)
  2436. #
  2437. # return data
  2438. #
  2439. # @property
  2440. # def summary(self):
  2441. # return {
  2442. # 'id': str(self.id),
  2443. #
  2444. # 'consumeType': self.consume_type,
  2445. #
  2446. # 'ownerId': self.ownerId,
  2447. #
  2448. # 'logicalCode': self.logicalCode,
  2449. #
  2450. # 'groupName': self.groupName,
  2451. # 'devTypeName': self.dev_type_name,
  2452. #
  2453. # 'money': self.money,
  2454. # 'coins': self.coin,
  2455. #
  2456. # 'createdTime': self.dateTimeAdded.strftime(Const.DATETIME_FMT),
  2457. # }
  2458. #
  2459. # def to_detail(self, source = None, hides = []):
  2460. # # type: (Optional[str], Optional[list])->dict
  2461. #
  2462. # data = {
  2463. # 'id': str(self.id),
  2464. #
  2465. # 'orderNo': self.orderNo,
  2466. #
  2467. # 'createdTime': self.created_date,
  2468. # 'completionTime': self.completion_date,
  2469. #
  2470. # 'deviceStatTime': self.device_start_time,
  2471. # 'deviceFinishedTime': self.device_finished_time,
  2472. #
  2473. # 'orderAmount': self.coin,
  2474. #
  2475. # 'amount': self.coin,
  2476. # 'coins': self.coin,
  2477. #
  2478. # 'openId': self.openId,
  2479. # 'groupId': self.groupId,
  2480. # 'userNickname': u'用户' if not self.nickname else self.nickname,
  2481. #
  2482. # 'ownerId': self.ownerId,
  2483. #
  2484. # 'devNo': self.devNo,
  2485. # 'logicalCode': self.logicalCode,
  2486. # 'groupName': self.groupName,
  2487. # 'address': self.address,
  2488. # 'groupNumber': self.groupNumber,
  2489. # 'devType': self.dev_type_name, # 兼容
  2490. # 'devTypeName': self.dev_type_name,
  2491. #
  2492. # 'remarks': self.remarks,
  2493. #
  2494. # 'startResult': 'success' if self.isNormal else 'failed',
  2495. #
  2496. # 'errorDesc': self.errorDesc,
  2497. #
  2498. # 'servicedInfo': [u'%s: %s'.encode('utf-8')
  2499. # % (GLOSSARY_TRANSLATION.get(k, k), v)
  2500. # for k, v in self.servicedInfo.iteritems()
  2501. # if (k not in self.aggInfo and k in GLOSSARY_TRANSLATION)],
  2502. #
  2503. # 'aggInfo': self.consumption_stats(source = source, hides = hides),
  2504. #
  2505. # # 以下字段和以前做兼容
  2506. # 'finishedTime': self.device_finished_time,
  2507. # 'port': self.used_port
  2508. # }
  2509. #
  2510. # if self.rechargeRcdId:
  2511. # recharge_record = RechargeRecord.objects(id = self.rechargeRcdId).first() # type: RechargeRecord
  2512. # if recharge_record:
  2513. # data.update({
  2514. # 'gateway': recharge_record.gateway,
  2515. # 'rechargeRcdId': str(recharge_record.id),
  2516. # 'rechargeTradeNo': recharge_record.orderNo,
  2517. # 'gatewayTradeNo': recharge_record.wxOrderNo # 兼容
  2518. # })
  2519. #
  2520. # if self.redpackId:
  2521. # redPack = Redpack.get_one(self.redpackId)
  2522. # if redPack:
  2523. # package = self.package
  2524. # coins = package.get('coins')
  2525. # price = package.get('price')
  2526. # redpackMoney = redPack['money']
  2527. #
  2528. # if RMB(redpackMoney) >= RMB(price): # 消费中红包抵扣 做一个保护
  2529. # redpackAmount = coins
  2530. # else:
  2531. # redpackAmount = round(float(RMB(redpackMoney)) / price * coins, 2)
  2532. #
  2533. # data.update({'redpackAmount': redpackAmount})
  2534. #
  2535. # if self.devTypeCode in ["1008031", "100803"] and self.status:
  2536. # data.update({"status": self.status})
  2537. #
  2538. # payTypeInfo = self.payTypeInfo
  2539. # if payTypeInfo:
  2540. # data.update(payTypeInfo)
  2541. #
  2542. # return data
  2543. #
  2544. # def charge_recharge_status(self):
  2545. # if not hasattr(self, "is_waitPay") or not self.is_waitPay():
  2546. # return
  2547. #
  2548. # if self.recharge_record_id:
  2549. # from apps.web.common.proxy import ClientRechargeModelProxy
  2550. #
  2551. # rechargeRecord = ClientRechargeModelProxy.get_one(
  2552. # shard_filter = {'ownerId': self.ownerId},
  2553. # id = self.recharge_record_id) # type: RechargeRecord
  2554. # if int((datetime.datetime.now() - rechargeRecord.dateTimeAdded).total_seconds()) > 60 * 3:
  2555. # self.p_to_f()
  2556. #
  2557. # def consumption_stats(self, source = None, hides = []):
  2558. # # type:(Optional[str], Optional[list])->list
  2559. # return translate_consumption_stats(self.aggInfo.iteritems(), source, hides)
  2560. #
  2561. # def to_dict(self, source = None, hides = []):
  2562. # # type: (Optional[str], Optional[list])->Dict[str, Union[AnyStr, float]]
  2563. #
  2564. # title = u'设备{logicalCode}消费 {agg_info} '.format(logicalCode = self.logicalCode,
  2565. # agg_info = '-'.join(self.consumption_stats(source, hides)))
  2566. # return {
  2567. # 'id': str(self.id),
  2568. # 'title': title,
  2569. # 'amount': self.coin,
  2570. # 'createdTime': self.created_date,
  2571. # 'source': source,
  2572. # 'ownerId': self.ownerId,
  2573. # 'startResult': 'success' if self.isNormal else 'failed',
  2574. #
  2575. # 'logicalCode': self.logicalCode,
  2576. # 'devTypeName': self.dev_type_name,
  2577. # 'userNickname': self.nickname
  2578. # }
  2579. #
  2580. # def update_agg_info(self, mapping):
  2581. # # type: (dict)->bool
  2582. # """
  2583. #
  2584. # :param mapping:
  2585. # :return:
  2586. # """
  2587. # try:
  2588. # trans_consume_dict = {}
  2589. # for kind, value in mapping.iteritems():
  2590. # if kind not in DEALER_CONSUMPTION_AGG_KIND.choices():
  2591. # logger.debug('{} is not consume stat kind.'.format(kind))
  2592. # continue
  2593. #
  2594. # if value is None:
  2595. # logger.debug('{} is none.'.format(kind))
  2596. # continue
  2597. #
  2598. # if hasattr(value, 'mongo_amount'):
  2599. # trans_consume_dict[kind] = value.mongo_amount
  2600. # else:
  2601. # trans_consume_dict[kind] = value
  2602. #
  2603. # self.aggInfo.update(trans_consume_dict)
  2604. # self.save()
  2605. #
  2606. # except Exception as e:
  2607. # logger.exception(e)
  2608. # return False
  2609. #
  2610. # return True
  2611. #
  2612. # @classmethod
  2613. # def get_consumed_records(cls, **kwargs):
  2614. # return cls.objects(isNormal = True, **kwargs).order_by('-dateTimeAdded')
  2615. #
  2616. # @property
  2617. # def is_temp_package(self):
  2618. # if 'isTempPackage' in self.attachParas:
  2619. # return self.attachParas['isTempPackage']
  2620. #
  2621. # if 'isTempPackage' in self.package:
  2622. # return self.package['isTempPackage']
  2623. #
  2624. # return False
  2625. #
  2626. # @property
  2627. # def is_free(self):
  2628. # if 'isFree' in self.attachParas:
  2629. # return self.attachParas['isFree']
  2630. #
  2631. # if 'isFree' in self.package:
  2632. # return self.package['isFree']
  2633. #
  2634. # return False
  2635. #
  2636. # @property
  2637. # def my_package(self):
  2638. # return PackageDict(self.package)
  2639. #
  2640. # @property
  2641. # def statistic_type(self):
  2642. # return "consumption"
  2643. #
  2644. # @property
  2645. # def dealerIds(self):
  2646. # return [ObjectId(self.ownerId)] + [ObjectId(_['id']) for _ in
  2647. # Group.get_group(self.groupId)['partnerDict'].values()]
  2648. #
  2649. # def get_statistic_update_info(self, amount = None):
  2650. # """
  2651. # 获取 更新的数据
  2652. # :param amount: 设置的金币的金额
  2653. # :return:
  2654. # """
  2655. # kindMap = self.aggInfo or dict()
  2656. #
  2657. # kindMap.setdefault("coin", VirtualCoin(amount or self.coin))
  2658. # hour = self.dateTimeAdded.hour
  2659. #
  2660. # updateOrInsertData = {
  2661. # "add_to_set__origin__{}".format(self.statistic_type): self.id,
  2662. # "inc__daily__totalConsumptionCount": 1
  2663. # }
  2664. #
  2665. # for _kind, _value in kindMap.items():
  2666. # updateOrInsertData["inc__daily__{}__{}".format(self.statistic_type, _kind)] = \
  2667. # to_quantity(_kind, _value).mongo_amount
  2668. # updateOrInsertData["inc__hourly__{}__{}__{}".format(hour, self.statistic_type, _kind)] = \
  2669. # to_quantity(_kind, _value).mongo_amount
  2670. #
  2671. # return updateOrInsertData
  2672. #
  2673. # def deail_link(self, device):
  2674. # # type:(DeviceDict)->str
  2675. #
  2676. # if device.channelType == DeviceChannelType.Channel_BT:
  2677. # if device.deviceAdapter.support_count_down():
  2678. # return concat_bt_count_down_page_url(l = self.logicalCode, port = self.used_port)
  2679. # else:
  2680. # if device.devTypeCode in ['1002121', '1002122']:
  2681. # return concat_front_end_url(uri = '/user/index.html#/user/deviceStatus')
  2682. # else:
  2683. # if device.deviceAdapter.support_count_down(self.openId, self.used_port):
  2684. # return concat_count_down_page_url(devNo = self.devNo, port = self.used_port)
  2685. # else:
  2686. # return concat_front_end_url(uri = '/user/index.html#/user/consumeDetail?id={}'.format(str(self.id)))
  2687. #
  2688. # def update_paid(self, **kwargs):
  2689. # # type: (dict)->None
  2690. #
  2691. # if not self.paymentInfo:
  2692. # logger.debug('consume record<id={}> is old version.'.format(str(self.id)))
  2693. # return
  2694. #
  2695. # self.paymentInfo['payTime'] = datetime.datetime.now()
  2696. #
  2697. # vCardConsumeRecord = kwargs.get('vCardConsumeRecord', None)
  2698. # if vCardConsumeRecord:
  2699. # assert self.paymentInfo['via'] == 'virtualCard', 'is not virtual card pay type.'
  2700. #
  2701. # self.paymentInfo['vCardConsumeRcdId'] = str(vCardConsumeRecord.id)
  2702. # self.paymentInfo['vCardConsumeData'] = {
  2703. # 'consumeData': vCardConsumeRecord.consumeData,
  2704. # 'consumeDayData': vCardConsumeRecord.consumeDayData
  2705. # }
  2706. #
  2707. # self.save()
  2708. #
  2709. # def update_failure(self, errorDesc):
  2710. # self.isNormal = False
  2711. # self.errorDesc = errorDesc
  2712. # if not self.dateTimeCompletion:
  2713. # self.dateTimeCompletion = datetime.datetime.now()
  2714. #
  2715. # self.save()
  2716. #
  2717. # @property
  2718. # def user(self):
  2719. # return MyUser.objects.filter(openId = self.openId, groupId = self.groupId).first()
  2720. #
  2721. # @property
  2722. # def redpackId(self):
  2723. # return self.attachParas.get('redpackId', "")
  2724. #
  2725. # @property
  2726. # def insId(self):
  2727. # return self.attachParas.get("insId", "")
  2728. #
  2729. # @property
  2730. # def monthlyPackageId(self):
  2731. # return self.attachParas.get('monthlyPackageId', "")
  2732. #
  2733. # @property
  2734. # def refund_info(self):
  2735. # rv = {
  2736. # 'isRefund': False
  2737. # }
  2738. #
  2739. # if self.isNormal:
  2740. # if self.servicedInfo.get('refundedMoney') and VirtualCoin(
  2741. # self.servicedInfo.get('refundedMoney')) > VirtualCoin('0.00'):
  2742. # rv['isRefund'] = True
  2743. # rv['refundedMoney'] = str(self.servicedInfo.get('refundedMoney'))
  2744. # elif self.servicedInfo.get('refundedCash') and RMB(self.servicedInfo.get('refundedCash')) > RMB('0.00'):
  2745. # rv['isRefund'] = True
  2746. # rv['refundedCash'] = str(self.servicedInfo.get('refundedCash'))
  2747. # else:
  2748. # if self.status == ConsumeRecord.Status.REFUND: # 蓝牙会设置退费状态
  2749. # if VirtualCoin(self.coin) > VirtualCoin('0.00'):
  2750. # rv['isRefund'] = True
  2751. # rv['refundedMoney'] = str(self.coin)
  2752. # elif self.recharge_record_id:
  2753. # refund_order = RefundMoneyRecord.objects(rechargeObjId = str(self.recharge_record_id)).first()
  2754. # if refund_order:
  2755. # rv['isRefund'] = True
  2756. # rv['refundedCash'] = str(refund_order.money)
  2757. # else:
  2758. # if VirtualCoin(self.coin) > VirtualCoin('0.00'): # 失败的情况下根本不会扣费
  2759. # rv['isRefund'] = True
  2760. # rv['refundedMoney'] = str(self.coin)
  2761. #
  2762. # return rv
  2763. #
  2764. # @property
  2765. # def info_for_feedback(self):
  2766. # rv = {
  2767. # 'orderId': str(self.id),
  2768. # 'orderNo': self.orderNo,
  2769. # 'coins': self.coin,
  2770. # 'deviceStartTime': self.device_start_time,
  2771. # 'deviceFinishedTime': self.device_finished_time
  2772. # }
  2773. #
  2774. # rv.update(self.device_identity_info)
  2775. #
  2776. # rv.update(self.refund_info)
  2777. #
  2778. # if self.used_port != -1:
  2779. # rv.update({
  2780. # 'port': self.used_port
  2781. # })
  2782. #
  2783. # if self.recharge_record_id:
  2784. # from apps.web.common.proxy import ClientRechargeModelProxy
  2785. # recharge_record = ClientRechargeModelProxy.get_one(
  2786. # shard_filter = {'ownerId': self.ownerId}, id = self.recharge_record_id) # type: RechargeRecord
  2787. #
  2788. # if recharge_record:
  2789. # rv.update({
  2790. # 'rechargeTradeNo': recharge_record.orderNo,
  2791. # 'money': self.money
  2792. # })
  2793. #
  2794. # return rv
  2795. #
  2796. class CardRechargeOrder(Searchable):
  2797. """
  2798. 卡充值订单
  2799. """
  2800. orderNo = StringField(verbose_name=u"订单号", default="")
  2801. cardId = StringField(verbose_name=u"实体卡ID", default="")
  2802. cardNo = StringField(verbose_name=u"实体卡NO", default="")
  2803. # 卡绑定的发行地址
  2804. groupId = StringField(verbose_name=u"设备地址编号", default="")
  2805. address = StringField(verbose_name=u"设备地址", default="")
  2806. groupName = StringField(verbose_name=u"交易场地", default="")
  2807. openId = StringField(verbose_name=u"openId", default="")
  2808. money = MonetaryField(verbose_name=u"付款金额", default=RMB('0.00'))
  2809. coins = MonetaryField(verbose_name=u"充值金额", default=RMB('0.00'))
  2810. remarks = StringField(verbose_name=u"备注", default="")
  2811. dateTimeAdded = DateTimeField(default=datetime.datetime.now, verbose_name=u'生成时间')
  2812. sequanceNo = StringField(verbose_name=u"交易流水号", default="")
  2813. status = StringField(verbose_name=u"状态", default="")
  2814. rechargeType = StringField(verbose_name=u"充值类型,一般分网络充值、线下设备充值", default="") # netpay/device
  2815. dealerId = StringField(verbose_name=u'订单ownerId, 集群需要', default=None)
  2816. rechargeNo = ObjectIdField(verbose_name=u"订单ID")
  2817. operationLog = ListField(verbose_name=u'历史处理结果', default=[])
  2818. processingLog = DictField(verbose_name=u'')
  2819. meta = {
  2820. "collection": "cardRechargeOrder",
  2821. "db_alias": "default",
  2822. }
  2823. def __repr__(self):
  2824. return '<CardRechargeOrder(id=%s, orderNo=%s, cardNo=%s)>' % (str(self.id), self.orderNo, self.cardNo)
  2825. @cached_property
  2826. def rechargeOrder(self):
  2827. return RechargeRecord.objects.filter(id=self.rechargeNo).first()
  2828. @property
  2829. def chargeAmount(self):
  2830. return self.money
  2831. @property
  2832. def bestowAmount(self):
  2833. return RMB(self.coins) - RMB(self.money)
  2834. # 获取卡的订单,但是没有支付的
  2835. @staticmethod
  2836. def get_last_to_do_one(cardId):
  2837. obj = CardRechargeOrder.objects(
  2838. cardId=cardId, status='finishedPay',
  2839. rechargeType__in=["netpay", "sendCoin"]
  2840. ).order_by('-dateTimeAdded').first()
  2841. return obj
  2842. # 获取卡的订单,但是没有支付的
  2843. @staticmethod
  2844. def get_to_do_list(cardId):
  2845. objs = CardRechargeOrder.objects.filter(
  2846. cardId=cardId, status='finishedPay',
  2847. rechargeType__in=['netpay', "sendCoin"]
  2848. )
  2849. money, coins = RMB(0), RMB(0)
  2850. orderNos = []
  2851. cardOrderNos = []
  2852. for obj in objs:
  2853. money += obj.money
  2854. coins += obj.coins
  2855. orderNos.append(obj.rechargeNo)
  2856. cardOrderNos.append(obj.orderNo)
  2857. return money, coins, orderNos, cardOrderNos
  2858. # 更新卡的为到账订单,更新为已经支付
  2859. @staticmethod
  2860. def update_card_order_has_finished(cardId):
  2861. CardRechargeOrder.get_collection().update(
  2862. {'cardId': cardId, 'status': 'finishedPay', 'rechargeType': {"$in": ["netpay", "sendCoin"]}},
  2863. {'$set': {'status': 'finished'}},
  2864. multi = True
  2865. )
  2866. def init_processing_log(self, device, sendMoney, preBalance):
  2867. # type: (DeviceDict, RMB, RMB)->CardRechargeOrder
  2868. self.processingLog = {
  2869. 'devNo': device.devNo,
  2870. 'logicalCode': device.logicalCode,
  2871. 'sendMoney': sendMoney.mongo_amount,
  2872. 'preBalance': preBalance.mongo_amount,
  2873. 'code': device.get('devType', {}).get('code', ''),
  2874. 'time': datetime.datetime.now()
  2875. }
  2876. # return self.save()
  2877. def update_after_recharge_ic_card(self, device, sendMoney, preBalance,
  2878. result = ErrorCode.SUCCESS, description = '', syncBalance = None):
  2879. # type: (DeviceDict, RMB, RMB, int, basestring, Optional[RMB])->CardRechargeOrder
  2880. log = {
  2881. 'devNo': device.devNo,
  2882. 'logicalCode': device.logicalCode,
  2883. 'sendMoney': sendMoney.mongo_amount,
  2884. 'preBalance': preBalance.mongo_amount,
  2885. 'code': device.get('devType', {}).get('code', ''),
  2886. 'result': result,
  2887. 'description': description,
  2888. 'time': datetime.datetime.now()
  2889. }
  2890. if syncBalance:
  2891. log['syncBalance'] = syncBalance.mongo_amount
  2892. self.operationLog.append(log)
  2893. if result == ErrorCode.SUCCESS:
  2894. self.status = 'finished'
  2895. return self.save()
  2896. def update_after_recharge_id_card(self, device, balance, preBalance):
  2897. # type: (Optional[DeviceDict], RMB, RMB)->CardRechargeOrder
  2898. if device:
  2899. log = {
  2900. 'devNo': device.devNo,
  2901. 'logicalCode': device.logicalCode,
  2902. 'balance': balance.mongo_amount,
  2903. 'preBalance': preBalance.mongo_amount,
  2904. 'code': device.get('devType', {}).get('code', ''),
  2905. 'time': datetime.datetime.now()
  2906. }
  2907. else:
  2908. log = {
  2909. 'balance': balance.mongo_amount,
  2910. 'preBalance': preBalance.mongo_amount,
  2911. 'time': datetime.datetime.now()
  2912. }
  2913. self.operationLog.append(log)
  2914. self.status = 'finished'
  2915. return self.save()
  2916. @staticmethod
  2917. def new_one(openId, cardId, cardNo, money, coins, group, rechargeId, rechargeType = 'netpay'):
  2918. # type:(str, str, str, RMB, RMB, GroupDict, ObjectId, str)->CardRechargeOrder
  2919. newRcd = CardRechargeOrder(
  2920. orderNo = OrderNoMaker.make_order_no_32(identifier = cardNo,
  2921. main_type = OrderMainType.PAY,
  2922. sub_type = UserPaySubType.CARD_RECHARGE_RECORD),
  2923. cardId = str(cardId),
  2924. cardNo = cardNo,
  2925. openId = openId,
  2926. money = money,
  2927. coins = RMB(coins.amount),
  2928. status = 'finishedPay',
  2929. rechargeType = rechargeType,
  2930. dealerId = group.ownerId,
  2931. groupId = group.groupId,
  2932. address = group.address,
  2933. groupName = group.groupName,
  2934. rechargeNo = rechargeId
  2935. )
  2936. logger.info('make card recharge order = %s' % repr(newRcd))
  2937. try:
  2938. return newRcd.save()
  2939. except Exception as e:
  2940. logger.exception('save cardId = %s; cardNo = %s; recharge error = %s' % (cardId, cardNo, e))
  2941. return None
  2942. def to_detail(self):
  2943. # type: ()->dict
  2944. """
  2945. :return:
  2946. """
  2947. return {
  2948. 'orderNo': self.orderNo,
  2949. 'cardNo': self.cardNo,
  2950. 'openId': self.openId,
  2951. 'groupId': self.groupId,
  2952. 'amount': self.money,
  2953. 'devNo': '',
  2954. 'logicalCode': '',
  2955. 'devType': '',
  2956. 'groupNumber': '',
  2957. 'groupName': self.groupName,
  2958. 'address': self.address
  2959. }
  2960. @classmethod
  2961. def get_by_rechargeRecord(cls, record):
  2962. # type:(RechargeRecord)->CardRechargeOrder
  2963. return cls.objects(rechargeNo = record.id).first()
  2964. class CardRechargeRecord(Searchable):
  2965. orderNo = StringField(verbose_name = "关联订单号", default = "") # 关联订单处理卡退费
  2966. cardId = StringField(verbose_name = "实体卡ID", default = "")
  2967. cardNo = StringField(verbose_name = "实体卡号", default = "")
  2968. openId = StringField(verbose_name = "openId", default = "")
  2969. ownerId = ObjectIdField(verbose_name = "dealerId")
  2970. money = MonetaryField(verbose_name = "付款金额", default = RMB('0.00'))
  2971. coins = MonetaryField(verbose_name = "充值金额", default = VirtualCoin('0.00'))
  2972. chargeBalance = MonetaryField(verbose_name="充值余额", default=RMB(0))
  2973. bestowBalance = MonetaryField(verbose_name="赠送余额", default=RMB(0))
  2974. preChargeBalance = MonetaryField(verbose_name="之前充值余额", default=RMB(0))
  2975. preBestowBalance = MonetaryField(verbose_name="之前赠送余额", default=RMB(0))
  2976. devNo = StringField(verbose_name = "设备ID", default = "")
  2977. devTypeCode = StringField(verbose_name = "设备类型编码", default = "")
  2978. logicalCode = StringField(verbose_name = "设备逻辑编码", default = "")
  2979. groupId = StringField(verbose_name = "设备地址编号", default="")
  2980. address = StringField(verbose_name = "设备地址", default = "")
  2981. groupNumber = StringField(verbose_name = "设备", default = "")
  2982. groupName = StringField(verbose_name = "交易场地", default = "")
  2983. #: 在部分场合,设备会失败启动,但这时应该添加该字段为False方便追溯
  2984. remarks = StringField(verbose_name="备注", default="")
  2985. dateTimeAdded = DateTimeField(default=datetime.datetime.now, verbose_name='生成时间')
  2986. sequanceNo = StringField(verbose_name="交易流水号,有些刷卡的会上报", default="")
  2987. status = StringField(verbose_name="状态", default="")
  2988. rechargeType = StringField(verbose_name="充值类型,一般分网络充值、线下设备充值", default="")
  2989. meta = {
  2990. "collection": "cardRechargeRecord",
  2991. "db_alias": "default",
  2992. # "shard_key":('cardId',),
  2993. }
  2994. @property
  2995. def amount(self):
  2996. return self.money
  2997. @property
  2998. def balance(self):
  2999. return self.chargeBalance + self.bestowBalance
  3000. @property
  3001. def preBalance(self):
  3002. return self.preChargeBalance + self.preBestowBalance
  3003. @property
  3004. def dealer(self):
  3005. return Dealer.objects(id=self.ownerId).get()
  3006. @staticmethod
  3007. def make_no():
  3008. timestamp = generate_timestamp_ex()
  3009. random_int = random.randint(1, 1000)
  3010. return "%d%03d" % (timestamp, random_int)
  3011. @classmethod
  3012. def add_self_card_record(cls, cardId, group, money, device):
  3013. newRcd = CardRechargeRecord(
  3014. cardId = cardId,
  3015. ownerId = ObjectId(group.ownerId),
  3016. money = money,
  3017. coins = money,
  3018. groupId = group.groupId,
  3019. address = group.address,
  3020. groupName = group.groupName,
  3021. status = 'success',
  3022. rechargeType = 'dealer_oper',
  3023. devNo = device.devNo,
  3024. devTypeCode = device.get('devType', {}).get('code', ''),
  3025. logicalCode = device.logicalCode,
  3026. groupNumber = device.groupNumber
  3027. )
  3028. try:
  3029. return newRcd.save()
  3030. except Exception as e:
  3031. logger.exception('save card consume rcd error=%s' % e)
  3032. @staticmethod
  3033. def add_record(card, group, order, device=None):
  3034. # type:(Card, GroupDict, CardRechargeOrder, Optional[DeviceDict])->CardRechargeRecord
  3035. newRcd = CardRechargeRecord(
  3036. cardId=str(card.id),
  3037. cardNo=card.cardNo,
  3038. openId=card.openId,
  3039. ownerId=ObjectId(group.ownerId),
  3040. money=order.money,
  3041. coins=order.coins,
  3042. chargeBalance=order.chargeAmount + card.chargeBalance,
  3043. bestowBalance=order.bestowAmount + card.bestowBalance,
  3044. preChargeBalance=card.chargeBalance,
  3045. preBestowBalance=card.bestowBalance,
  3046. groupId=group.groupId,
  3047. address=group.address,
  3048. groupName=group.groupName,
  3049. status='success',
  3050. rechargeType=order.rechargeType
  3051. )
  3052. if device:
  3053. newRcd.devNo = device.devNo
  3054. newRcd.logicalCode = device.logicalCode
  3055. newRcd.devTypeCode = device.devTypeCode
  3056. newRcd.groupNumber = device.groupNumber
  3057. try:
  3058. return newRcd.save()
  3059. except Exception as e:
  3060. logger.exception('save card consume rcd error=%s' % e)
  3061. class CardConsumeRecord(Searchable):
  3062. orderNo = StringField(verbose_name = "订单号", default = "")
  3063. openId = StringField(verbose_name = "微信ID", default = "")
  3064. cardId = StringField(verbose_name = "实体卡ID", default = "")
  3065. money = MonetaryField(verbose_name = "消耗的钱(硬币)", default = RMB('0.00'))
  3066. balance = MonetaryField(verbose_name = "当前余额(硬币)", default = RMB('0.00'))
  3067. devNo = StringField(verbose_name = "设备ID", default = "")
  3068. devType = StringField(verbose_name = "设备类型", default = "")
  3069. logicalCode = StringField(verbose_name = "设备逻辑编码", default = "")
  3070. groupId = StringField(verbose_name = "设备地址编号", default = "")
  3071. address = StringField(verbose_name = "设备地址", default = "")
  3072. groupNumber = StringField(verbose_name = "设备", default = "")
  3073. groupName = StringField(verbose_name = "交易场地", default = "")
  3074. result = StringField(verbose_name = "消费结果", default = "")
  3075. remarks = StringField(verbose_name = "备注", default = "")
  3076. desc = StringField(verbose_name = "描述", default = "")
  3077. dateTimeAdded = DateTimeField(verbose_name = '生成时间', default = datetime.datetime.now)
  3078. finishedTime = DateTimeField(verbose_name = '结束时间', default = datetime.datetime.now)
  3079. servicedInfo = DictField(verbose_name = '服务内容', default = {})
  3080. sid = StringField(verbose_name = "从设备上获取的流水号", default = "")
  3081. linkedConsumeRcdOrderNo = StringField(verbose_name = u'关联的消费订单号.发布后去掉, 两者一致', default = '')
  3082. meta = {
  3083. "collection": "CardConsumeRecord",
  3084. "db_alias": "default",
  3085. # "shard_key":('orderNo',),
  3086. }
  3087. @staticmethod
  3088. def make_no():
  3089. timestamp = generate_timestamp_ex()
  3090. random_int = random.randint(1, 1000)
  3091. return "%d%03d" % (timestamp, random_int)
  3092. class ServiceProgress(Searchable):
  3093. open_id = StringField(verbose_name = 'open id', default = '')
  3094. device_imei = StringField(verbose_name = 'device imei', default = '')
  3095. port = IntField(verbose_name = 'port', default = 0)
  3096. isFinished = BooleanField(verbose_name = 'isFinished', default = False)
  3097. devTypeCode = StringField(verbose_name = 'devTypeCode', default = '')
  3098. cardId = StringField(verbose_name = 'cardNo', default = '')
  3099. attachParas = DictField(verbose_name = 'attachParas', default = {})
  3100. consumeOrder = DictField(verbose_name = 'consumeOrder', default = {})
  3101. finished_time = IntField(verbose_name = 'finished time', default = 0)
  3102. start_time = IntField(verbose_name = 'start_time', default = 0)
  3103. consumes = ListField(verbose_name = "对应的所有消费记录单号", default = [])
  3104. # 应该用一个状态表示服务的执行情况,包括等待服务、服务正在进行、服务结束。只有我们才需要此字段
  3105. # 但是只有我们自己的主板支持每个订单的执行情况反馈以及查询,所以没有办法,以前都是裹在一起,现在分开逻辑更清晰。
  3106. status = StringField(verbose_name = u'执行状态', default = 'running') # waiting/working/finished/failed
  3107. weifuleOrderNo = StringField(verbose_name = u'微付乐订单编号,设备也会产生订单', default = '')
  3108. runningTime = DateTimeField(verbose_name = u'开始运行的时间')
  3109. datetimeAdded = DateTimeField(verbose_name = "时间", default = datetime.datetime.now)
  3110. expireAt = DateTimeField(verbose_name = u"TTL到期时间", default = None)
  3111. meta = {
  3112. 'collection': 'service_progress',
  3113. 'db_alias': 'logdata'
  3114. }
  3115. @property
  3116. def used_port(self):
  3117. """
  3118. 兼容性属性. 以前是整形, 后面全部为字符串型
  3119. :return:
  3120. """
  3121. return str(self.port)
  3122. @property
  3123. def order_no(self):
  3124. return self.attachParas.get('orderNo', None)
  3125. # 更新progress,以及关联的消费日志记录。这个函数一般是服务结束的时候调用,以刷新progress,有时候,结束时
  3126. # 才扣费的场景,扣费的时候会直接把使用后的消耗,记录到消费日志中,就不需要刷新消费日志
  3127. @staticmethod
  3128. def update_progress_and_consume_rcd(ownerId, queryDict, consumeDict, updateConsume = True, progressDict = None):
  3129. progressDict = {
  3130. 'isFinished': True, 'finished_time': int(time.time())
  3131. } if progressDict is None else progressDict
  3132. if 'isFinished' in progressDict and progressDict['isFinished']:
  3133. progressDict.update({'expireAt': datetime.datetime.now()})
  3134. rcds = ServiceProgress.get_collection().find(queryDict).sort('start_time')
  3135. if rcds.count() == 0:
  3136. logger.error('can not find the progress info')
  3137. return
  3138. orderIdList, orderNoList, cardOrderNoList = [], [], []
  3139. for rcd in rcds: # type: Dict
  3140. consumeOrder = rcd.get('consumeOrder', {})
  3141. if 'consumeRecordId' in consumeOrder and consumeOrder['consumeRecordId']:
  3142. orderIdList.append(consumeOrder['consumeRecordId'])
  3143. elif 'orderNo' in consumeOrder and consumeOrder['orderNo']:
  3144. orderNoList.append(consumeOrder['orderNo'])
  3145. if 'cardOrderNo' in consumeOrder:
  3146. cardOrderNoList.append(consumeOrder['cardOrderNo'])
  3147. # 更新进程的数据
  3148. try:
  3149. ServiceProgress.get_collection().update(queryDict, {'$set': progressDict}, multi = True)
  3150. except Exception, e:
  3151. logger.exception('update service progress e=%s' % e)
  3152. if not updateConsume: # 不需要更新就直接返回
  3153. return
  3154. # 更新订单中的时间以及服务信息。如果是卡的信息,因为找不到order,会直接pass
  3155. # 如果是多条消费单,但是最终只有一条结束事件的场景,统计以及消费信息只记录到第一单上
  3156. isFinished = progressDict and progressDict.get('isFinished', False)
  3157. consume_orders = []
  3158. for orderNo in orderNoList:
  3159. rcd = ConsumeRecord.objects(ownerId = ownerId, orderNo = orderNo).first() # type: ConsumeRecord
  3160. consume_orders.append(rcd)
  3161. for orderId in orderIdList:
  3162. rcd = ConsumeRecord.objects(ownerId = ownerId, id = orderId).first() # type: ConsumeRecord
  3163. consume_orders.append(rcd)
  3164. count = 0
  3165. uart_data = consumeDict.pop('uartData', None)
  3166. for rcd in consume_orders: # type: ConsumeRecord
  3167. try:
  3168. count += 1
  3169. pre_aggInfo = rcd.aggInfo
  3170. # 如果是结束更新,相应的信息只需要更新到第一条中。非结束的,全部都刷新。
  3171. if not isFinished:
  3172. updated = rcd.update(
  3173. finishedTime = datetime.datetime.now(),
  3174. servicedInfo = consumeDict,
  3175. status = ConsumeRecord.Status.FINISHED)
  3176. elif count == 1:
  3177. if uart_data:
  3178. updated = rcd.update(
  3179. finishedTime = datetime.datetime.now(),
  3180. servicedInfo = consumeDict,
  3181. status = ConsumeRecord.Status.FINISHED,
  3182. attachParas__uartData = uart_data)
  3183. else:
  3184. updated = rcd.update(
  3185. finishedTime = datetime.datetime.now(),
  3186. servicedInfo = consumeDict,
  3187. status = ConsumeRecord.Status.FINISHED)
  3188. else:
  3189. continue
  3190. if not updated:
  3191. logger.error('%r updated failed' % (rcd,))
  3192. if isFinished and count == 1: # 只需要汇总一次使用的数据即可
  3193. try:
  3194. valueDict = {}
  3195. for kind, value in consumeDict.items():
  3196. if kind in DEALER_CONSUMPTION_AGG_KIND.choices():
  3197. if kind in pre_aggInfo:
  3198. #: 会在发起service阶段更新一次报表
  3199. #: 此处为结束时更新, 应该更新新数据 - 初始更新的数据
  3200. valueDict[kind] = float(str(value)) - float(str(pre_aggInfo[kind]))
  3201. else:
  3202. valueDict[kind] = value
  3203. status = rcd.update_agg_info(valueDict)
  3204. if status:
  3205. record_consumption_stats(rcd)
  3206. else:
  3207. logger.error(
  3208. '[update_progress_and_consume_rcd] failed to update_agg_info record=%r' % (rcd,))
  3209. except Exception, e:
  3210. logger.exception('update agg info error=%s' % e)
  3211. except Exception as e:
  3212. logger.exception('update consume rcd e=%s' % e)
  3213. continue
  3214. for orderNo in cardOrderNoList:
  3215. if orderNo == '':
  3216. continue
  3217. try:
  3218. CardConsumeRecord.get_collection().update_one({'orderNo': orderNo}, {
  3219. '$set': {'finished_time': int(time.time()), 'servicedInfo': consumeDict}})
  3220. except Exception, e:
  3221. logger.exception('update consume rcd e=%s' % e)
  3222. continue
  3223. @staticmethod
  3224. def register_card_service(dev, port, card, consumeOrder = None, finishedTime = None):
  3225. # type:(DeviceDict, int, Card, dict, int)->None
  3226. consumeOrder = {} if consumeOrder is None else consumeOrder
  3227. finishedTime = int(time.time()) + 24 * 60 * 60 if finishedTime is None else finishedTime
  3228. ServiceProgress.get_collection().insert({
  3229. 'device_imei': dev.devNo,
  3230. 'devTypeCode': dev.devTypeCode,
  3231. 'cardId': str(card.id),
  3232. 'port': port,
  3233. 'finished_time': finishedTime,
  3234. 'start_time': int(time.time()),
  3235. 'open_id': card.openId,
  3236. 'consumeOrder': consumeOrder,
  3237. 'isFinished': False,
  3238. 'attachParas': {},
  3239. 'expireAt': datetime.datetime.now() + datetime.timedelta(days = 91)
  3240. })
  3241. @staticmethod
  3242. def register_card_service_for_weifule(dev, port, card, consumeOrder):
  3243. ServiceProgress.get_collection().insert({
  3244. 'device_imei': dev['devNo'],
  3245. 'devTypeCode': dev['devType']['code'],
  3246. 'cardId': str(card.id),
  3247. 'port': port,
  3248. 'finished_time': int(time.time()) + 24 * 60 * 60,
  3249. 'start_time': int(time.time()),
  3250. 'open_id': card.openId,
  3251. 'consumeOrder': consumeOrder,
  3252. 'isFinished': False,
  3253. 'attachParas': {},
  3254. 'status': 'waiting',
  3255. 'weifuleOrderNo': consumeOrder['orderNo'],
  3256. 'expireAt': datetime.datetime.now() + datetime.timedelta(days = 91)
  3257. })
  3258. @classmethod
  3259. def new_progress_for_order(cls, order, device, cache_info):
  3260. # type:(ConsumeRecord, DeviceDict, dict)->None
  3261. try:
  3262. consumeOrder = {
  3263. 'consumeRecordId': str(order.id),
  3264. 'orderNo': order.orderNo,
  3265. 'coin': VirtualCoin(cache_info['coins']).mongo_amount,
  3266. 'consumeType': cache_info.get('consumeType') or (
  3267. 'mobile_vcard' if u'虚拟卡' in order.remarks else 'mobile')
  3268. }
  3269. if 'money' in cache_info:
  3270. consumeOrder.update({'money': RMB(cache_info['money']).mongo_amount})
  3271. if 'unit' in cache_info:
  3272. consumeOrder.update({'unit': cache_info['unit']})
  3273. if 'subOrderNos' in cache_info:
  3274. consumeOrder.update({'subOrderNos': cache_info['subOrderNos']})
  3275. if 'needKind' in cache_info and 'needValue' in cache_info:
  3276. consumeOrder.update({
  3277. cache_info['needKind']: cache_info['needValue']
  3278. })
  3279. if 'cardId' in cache_info:
  3280. consumeOrder.update({'cardId': cache_info['cardId']})
  3281. update_dict = {
  3282. 'open_id': cache_info['openId'],
  3283. 'device_imei': device.devNo,
  3284. 'devTypeCode': device['devType']['code'],
  3285. 'port': order.used_port,
  3286. 'attachParas': order.attachParas,
  3287. 'start_time': int(
  3288. Arrow.fromdatetime(order.startTime, tzinfo = settings.TIME_ZONE).timestamp),
  3289. 'finished_time': int(cache_info['estimatedTs']),
  3290. 'consumeOrder': consumeOrder,
  3291. 'isFinished': False,
  3292. 'expireAt': datetime.datetime.now() + datetime.timedelta(days = 91)
  3293. }
  3294. cls.objects(open_id = order.openId,
  3295. device_imei = device.devNo,
  3296. port = order.used_port).update_one(upsert = True, **update_dict)
  3297. except Exception as e:
  3298. logger.exception(e)
  3299. @classmethod
  3300. def finish_progress_for_order(cls, order, device):
  3301. # type:(ConsumeRecord, DeviceDict)->None
  3302. cls.get_collection().update_one(filter = {
  3303. 'open_id': order.openId,
  3304. 'device_imei': device.devNo,
  3305. 'port': int(order.used_port),
  3306. 'consumeOrder.consumeRecordId': str(order.id)
  3307. }, update = {
  3308. '$set': {
  3309. 'isFinished': True,
  3310. 'finished_time': Arrow.fromdatetime(order.finishedTime,
  3311. tzinfo = settings.TIME_ZONE).timestamp,
  3312. 'expireAt': datetime.datetime.now()
  3313. }
  3314. }, upsert = False)
  3315. class Card(Searchable):
  3316. cardNo = StringField(verbose_name="卡号", default="")
  3317. cardType = StringField(verbose_name="卡类型", default="")
  3318. openId = StringField(verbose_name="绑定的微信", default="")
  3319. nickName = StringField(verbose_name="用户昵称", default="")
  3320. productAgentId = StringField(verbose_name="当前用户绑定的平台代理商", default = "")
  3321. cardName = StringField(verbose_name="持卡人姓名", default="")
  3322. phone = StringField(verbose_name="手机号码", default="")
  3323. # 绑定的管理APP信息
  3324. managerialAppId = StringField(verbose_name="管理公众号AppId", default="")
  3325. managerialOpenId = StringField(verbose_name="管理openId", default="")
  3326. dealerId = StringField(verbose_name="经销商ID", default="")
  3327. groupId = StringField(verbose_name="设备组ID", default="")
  3328. agentId = StringField(verbose_name="代理商ID", default="")
  3329. # 以下信息为卡固有信息
  3330. remarks = StringField(verbose_name="备注", default="")
  3331. chargeBalance = MonetaryField(verbose_name="充值余额", default=RMB(0))
  3332. bestowBalance = MonetaryField(verbose_name=u"赠送余额", default=RMB(0))
  3333. isHaveBalance = BooleanField(verbose_name="是否能够获取到卡的余额数据", default=True)
  3334. lastMaxBalance = MonetaryField(verbose_name="最后一次充卡金额", default=RMB('0.00'))
  3335. boundVirtualCardId = ObjectIdField(verbose_name=u"绑定的虚拟卡券ID")
  3336. showBalance = MonetaryField(verbose_name=u"显示到设备上的卡余额,只有微付乐才会用此字段", default=RMB('0.00'))
  3337. # 最近一次刷卡的设备
  3338. devNo = StringField(verbose_name="设备IMEI", default="")
  3339. devTypeCode = StringField(verbose_name="设备类型代码", default="")
  3340. frozen = BooleanField(verbose_name="卡挂失状态", default=False)
  3341. status = StringField(verbose_name="卡状态", default='active')
  3342. dateTimeAdded = DateTimeField(verbose_name=u"添加卡的时间", default=datetime.datetime.now)
  3343. attachParas = DictField(verbose_name='附加参数', default={})
  3344. ongoingList = ListField(field=DictField(), verbose=u'冻结的金额')
  3345. search_fields = ('cardNo', 'cardName', 'phone')
  3346. meta = {
  3347. "collection": "recharge_card",
  3348. "db_alias": "default",
  3349. }
  3350. def __repr__(self):
  3351. return '<Card(id={}, cardNo={})>'.format(str(self.id), self.cardNo)
  3352. @cached_property
  3353. def group(self):
  3354. if not self.groupId:
  3355. return None
  3356. return Group.get_group(self.groupId)
  3357. @cached_property
  3358. def user(self):
  3359. if not self.isBinded:
  3360. return None
  3361. return MyUser.objects.filter(openId=self.openId, groupId=self.groupId).first()
  3362. @property
  3363. def balance(self):
  3364. return self.chargeBalance + self.bestowBalance
  3365. @property
  3366. def nickname(self):
  3367. return self.nickName
  3368. @property
  3369. def is_id_card(self):
  3370. # type: ()->bool
  3371. return self.cardType == RECHARGE_CARD_TYPE.ID
  3372. @property
  3373. def is_ic_card(self):
  3374. # type: ()->bool
  3375. return self.cardType == RECHARGE_CARD_TYPE.IC
  3376. @property
  3377. def isBinded(self):
  3378. """
  3379. 用户的绑定状态
  3380. """
  3381. return True if self.openId and self.openId != Const.DEFAULT_CARD_OPENID else False
  3382. def recharge(self, money, bestowMoney):
  3383. assert isinstance(money, RMB), 'coins had to be VirtualCoin'
  3384. assert isinstance(bestowMoney, RMB), 'money has to be RMB'
  3385. updated = self.update(
  3386. inc__chargeBalance=money,
  3387. inc__bestowBalance=bestowMoney,
  3388. )
  3389. if not updated:
  3390. raise PostPayOrderError(u'余额和累计充值更新失败')
  3391. return self.reload()
  3392. def account_consume(self, order): # type:(ConsumeRecord) -> None
  3393. self.reload()
  3394. payment = order.payment
  3395. # 获取支付的钱
  3396. consumeAmount = payment.actualAmount
  3397. consumeBestowAmount = payment.totalAmount - consumeAmount
  3398. # 获取当前的余额
  3399. CardBalanceLog.consume(
  3400. self,
  3401. afterAmount=self.chargeBalance,
  3402. afterBestowAmount=self.bestowBalance,
  3403. consumeAmount=consumeAmount,
  3404. consumeBestowAmount=consumeBestowAmount,
  3405. order=order
  3406. )
  3407. def account_refund(self, order): # type:(ConsumeRecord) -> None
  3408. self.reload()
  3409. refund = order.refund
  3410. # 获取支付的钱
  3411. refundAmount = refund.actualAmount
  3412. refundBestowAmount = refund.totalAmount - refundAmount
  3413. CardBalanceLog.refund(
  3414. self,
  3415. afterAmount=self.chargeBalance,
  3416. afterBestowAmount=self.bestowBalance,
  3417. refundAmount=refundAmount,
  3418. refundBestowAmount=refundBestowAmount,
  3419. order=order
  3420. )
  3421. def account_recharge(self, order): # type:(RechargeRecord) -> None
  3422. # 获取支付的钱
  3423. chargeAmount = order.chargeAmount
  3424. bestowAmount = order.bestowAmount
  3425. CardBalanceLog.recharge(
  3426. self,
  3427. afterAmount=self.chargeBalance,
  3428. afterBestowAmount=self.bestowBalance,
  3429. chargeAmount=chargeAmount,
  3430. chargeBestowAmount=bestowAmount,
  3431. order=order
  3432. )
  3433. @classmethod
  3434. def freeze_balance(cls, transaction_id, payment):
  3435. bulker = BulkHandlerEx(cls.get_collection()) # type: BulkHandlerEx
  3436. chargeBalanceField = cls.chargeBalance.name
  3437. bestowBalanceField = cls.bestowBalance.name
  3438. for deduct in payment.deduct_list:
  3439. query = {
  3440. '_id': ObjectId(deduct['id']),
  3441. 'ongoingList.transaction_id': {
  3442. '$ne': transaction_id
  3443. }
  3444. }
  3445. update = {
  3446. '$inc': {
  3447. chargeBalanceField: (-RMB(deduct[chargeBalanceField])).mongo_amount,
  3448. bestowBalanceField: (-RMB(deduct[bestowBalanceField])).mongo_amount
  3449. },
  3450. '$addToSet': {
  3451. 'ongoingList': {
  3452. 'transaction_id': transaction_id,
  3453. chargeBalanceField: deduct[chargeBalanceField],
  3454. bestowBalanceField: deduct[bestowBalanceField]
  3455. }
  3456. }
  3457. }
  3458. bulker.update(query, update)
  3459. result = bulker.execute()
  3460. logger.debug(result['info'])
  3461. if result['success'] == 0:
  3462. raise ServiceException({'result': 0, 'description': u"扣款失败(1001)"})
  3463. else:
  3464. if len(result['info']['writeErrors']) != 0:
  3465. raise ServiceException({'result': 0, 'description': u"扣款失败(1002)"})
  3466. else:
  3467. return True
  3468. @classmethod
  3469. def clear_frozen_balance(cls, transaction_id, payment, refund):
  3470. try:
  3471. bulker = BulkHandlerEx(cls.get_collection()) # type: BulkHandlerEx
  3472. chargeBalanceField = cls.chargeBalance.name
  3473. bestowBalanceField = cls.bestowBalance.name
  3474. for deduct in refund.deduct_list:
  3475. query = {
  3476. '_id': ObjectId(deduct['id']),
  3477. 'ongoingList': {
  3478. '$elemMatch': {
  3479. 'transaction_id': transaction_id
  3480. }
  3481. }
  3482. }
  3483. update = {
  3484. '$inc': {
  3485. chargeBalanceField: deduct[chargeBalanceField],
  3486. bestowBalanceField: deduct[bestowBalanceField],
  3487. },
  3488. '$pull': {
  3489. 'ongoingList': {
  3490. 'transaction_id': transaction_id
  3491. }
  3492. }
  3493. }
  3494. bulker.update(query, update)
  3495. result = bulker.execute()
  3496. logger.debug(result['info'])
  3497. if result['success'] == 0:
  3498. return False
  3499. else:
  3500. if len(result['info']['writeErrors']) != 0:
  3501. return False
  3502. else:
  3503. return True
  3504. except Exception as e:
  3505. logger.exception(e)
  3506. return False
  3507. @classmethod
  3508. def recover_frozen_balance(cls, transaction_id, deduct_list):
  3509. bulker = BulkHandlerEx(cls.get_collection()) # type: BulkHandlerEx
  3510. for deduct in deduct_list:
  3511. query = {
  3512. '_id': ObjectId(deduct['id']),
  3513. 'ongoingList': {
  3514. '$elemMatch': {'transaction_id': transaction_id}}}
  3515. update = {
  3516. '$inc': {
  3517. 'balance': deduct['coins']
  3518. },
  3519. '$pull': {
  3520. 'ongoingList': {
  3521. 'transaction_id': transaction_id, 'frozenCoins': deduct['coins']
  3522. }}
  3523. }
  3524. bulker.update(query, update)
  3525. result = bulker.execute()
  3526. if result['success'] == 0:
  3527. logger.error(result['info'])
  3528. return False
  3529. else:
  3530. if len(result['info']['writeErrors']) != 0:
  3531. logger.error(result['info'])
  3532. return False
  3533. else:
  3534. return True
  3535. @classmethod
  3536. def fake_one(cls, groupId, cardId='fake_id', cardNo='fake'):
  3537. return cls(id=cardId, cardNo=cardNo, openId='', productAgentId='', groupId=groupId)
  3538. @staticmethod
  3539. def get_card_status(cardId):
  3540. status = serviceCache.get(cardId, 'idle')
  3541. return status
  3542. @staticmethod
  3543. def set_card_status(cardId, status):
  3544. serviceCache.set(cardId, status, 10)
  3545. @staticmethod
  3546. def get_dev_cur_card(devNo):
  3547. return serviceCache.get('%s_cardId' % devNo, None)
  3548. @staticmethod
  3549. def set_dev_cur_card(devNo, cardInfo):
  3550. serviceCache.set('%s_cardId' % devNo, cardInfo, 90)
  3551. @property
  3552. def bound_virtual_card(self):
  3553. # type: ()->Optional[UserVirtualCard]
  3554. if self.boundVirtualCardId:
  3555. now = datetime.datetime.now()
  3556. virtual_card = UserVirtualCard.objects(id = self.boundVirtualCardId,
  3557. expiredTime__gte = now,
  3558. startTime__lte = now).first()
  3559. return virtual_card
  3560. else:
  3561. return None
  3562. def bind_virtual_card(self, card):
  3563. # type:(UserVirtualCard)->None
  3564. return self.update(boundVirtualCardId = card.id)
  3565. def unbind_virtual_card(self, card):
  3566. # type:(UserVirtualCard)->None
  3567. """
  3568. 以后也许会支持多张卡
  3569. :param card:
  3570. :return:
  3571. """
  3572. return self.update(boundVirtualCardId = None)
  3573. # 这个那种有余额的卡,才会调用
  3574. @staticmethod
  3575. def update_balance(cardId, balance):
  3576. card = Card.objects(id=cardId).get()
  3577. result = card.update(balance = balance)
  3578. if not result:
  3579. logger.error('update error, cardId=%s, balance=%s' % (cardId, balance))
  3580. return result
  3581. @staticmethod
  3582. def check_card_no(cardNo):
  3583. """
  3584. 检查实体卡卡号的合法性
  3585. :param cardNo:
  3586. :return:
  3587. """
  3588. if CARD_NO_RE.match(cardNo):
  3589. return True
  3590. return False
  3591. @classmethod
  3592. def check_swap_card_no(cls, cardNo, dealerId, agentId):
  3593. try:
  3594. card = cls.objects.get(cardNo=cardNo, agentId=agentId)
  3595. except DoesNotExist:
  3596. return True, ""
  3597. if card.dealerId and card.dealerId != dealerId:
  3598. return False, u"该卡已经被其他经销商绑定,请确认卡号无误"
  3599. if card.openId:
  3600. return False, u"该卡号已经被其他用户绑定,请确认卡号无误"
  3601. if CardConsumeRecord.objects.filter(cardId=str(card.id)):
  3602. return False, u"该卡号存在用户使用记录, 请确认卡号无误"
  3603. card.delete()
  3604. return True, ""
  3605. @staticmethod
  3606. def record_dev_card_no(devNo, cardNo):
  3607. serviceCache.set('%s-cardno' % devNo, cardNo, 2 * 60)
  3608. @staticmethod
  3609. def get_dev_card_no(devNo):
  3610. return serviceCache.get('%s-cardno' % devNo, None)
  3611. @staticmethod
  3612. def clear_dev_card_no(devNo):
  3613. serviceCache.delete('%s-cardno' % devNo)
  3614. def to_dict(self):
  3615. data = {
  3616. "cardId": str(self.id),
  3617. "cardNo": self.cardNo,
  3618. "cardName": self.cardName,
  3619. "phone": self.phone,
  3620. "cardType": self.cardType,
  3621. "remarks": self.remarks,
  3622. "frozen": self.frozen,
  3623. "balance": self.balance,
  3624. "lastMaxBalance": self.lastMaxBalance,
  3625. "bindStatus": self.isBinded,
  3626. }
  3627. return data
  3628. def clear_card(self):
  3629. return self.update(
  3630. cardType = '',
  3631. openId = '',
  3632. nickName = '',
  3633. productAgentId = '',
  3634. cardName = '',
  3635. phone = '',
  3636. managerialAppId = '',
  3637. managerialOpenId = '',
  3638. dealerId = '',
  3639. groupId = '',
  3640. remarks = '',
  3641. isHaveBalance = True,
  3642. lastMaxBalance = RMB('0.0'),
  3643. devNo = '',
  3644. devTypeCode = '',
  3645. frozen = False,
  3646. status = 'active',
  3647. boundVirtualCardId = None,
  3648. attachParas = {}
  3649. )
  3650. class RefundMoneyRecord(Searchable):
  3651. """
  3652. 用户退款记录
  3653. """
  3654. class Status(object):
  3655. """ 退款单的状态 正常流程顺序是从上至下 """
  3656. CREATED = 'created' # 创建
  3657. PROCESSING = 'processing' # 申请中
  3658. FAILURE = 'failure' # 申请失败 (彻底失败 需要重新发起)
  3659. SUCCESS = 'success' # 退款成功 (异步结果)
  3660. CLOSED = 'closed' # 退款失败 (异步结果 不需要重新发起)
  3661. # 订单常见的状态
  3662. # created ---> processing ---> success (申请 然后接受成功)
  3663. # created ---> failure (直接申请失败)
  3664. # created ---> processing ---> closed (申请成功 回调失败)
  3665. # created ---> processing ---> failure (申请成功 回调失败 不可以重试)
  3666. rechargeObjId = ObjectIdField(verbose_name = u'对应充值单')
  3667. # refundSeq = IntField(verbose_name=u'多次退款,计算退款序列号', default=1)
  3668. orderNo = StringField(verbose_name = u'商户退款订单号', default = '')
  3669. errorCode = StringField(verbose_name = u"错误代码", default = "")
  3670. errorDesc = StringField(verbose_name = u"错误描述", default = "")
  3671. money = MonetaryField(verbose_name = u"退款的金钱数额", default = RMB('0.00'))
  3672. coins = MonetaryField(verbose_name = u"清理用户金币数", default = VirtualCoin('0.00'))
  3673. status = StringField(verbose_name = u'订单状态', default = Status.CREATED)
  3674. datetimeAdded = DateTimeField(verbose_name = u"退款创建时间", default = datetime.datetime.now)
  3675. datetimeUpdated = DateTimeField(verbose_name = u"退款更新时间")
  3676. finishedTime = DateTimeField(verbose_name = u"退款到账时间")
  3677. tradeRefundNo = StringField(verbose_name = u"交易机构退款单号", default = "")
  3678. meta = {
  3679. "collection": "RefundMoneyRecord",
  3680. "db_alias": "default",
  3681. }
  3682. @classmethod
  3683. def issue(cls, order, refundCash, deductCoins):
  3684. # type:(RechargeRecord, RMB, VirtualCoin)->RefundMoneyRecord
  3685. refund_order_no = OrderNoMaker.make_order_no_32(
  3686. identifier = order.logicalCode,
  3687. main_type = OrderMainType.REFUND,
  3688. sub_type = RefundSubType.REFUND)
  3689. return cls(
  3690. rechargeObjId = order.id,
  3691. # refundSeq=next_seq,
  3692. orderNo = refund_order_no,
  3693. money = refundCash,
  3694. coins = deductCoins,
  3695. status = cls.Status.PROCESSING).save()
  3696. def succeed(self, tradeRefundNo, finishedTime): # type:(str, datetime.datetime) -> bool
  3697. """ 更新为成功状态 已经成功退款 """
  3698. result = self.__class__.get_collection().update_one(
  3699. {
  3700. 'orderNo': self.orderNo,
  3701. 'status': self.Status.PROCESSING
  3702. },
  3703. {
  3704. '$set': {
  3705. 'status': self.Status.SUCCESS,
  3706. 'datetimeUpdated': datetime.datetime.now(),
  3707. 'tradeRefundNo': tradeRefundNo,
  3708. 'finishedTime': finishedTime
  3709. }
  3710. }
  3711. )
  3712. return result.matched_count == 1
  3713. def fail(self, errorCode="", errorDesc=""): # type:(str, unicode) -> bool
  3714. """
  3715. 更新为失败状态 表示退款申请失败 这种情况下 可能就需要售后进行介入
  3716. 1. 订单申请发起时候即失败
  3717. 2. 订单申请通过 但是回调的时候明确失败 并且是不可重试的失败
  3718. 3. 此状态即表示订单没退款 理论上可以重新发起退款
  3719. """
  3720. result = self.__class__.objects.filter(
  3721. orderNo=self.orderNo,
  3722. status__in=[self.Status.PROCESSING, self.Status.CREATED]
  3723. ).update(
  3724. status=self.Status.FAILURE,
  3725. datetimeUpdated=datetime.datetime.now(),
  3726. errorCode=errorCode,
  3727. errorDesc=errorDesc
  3728. )
  3729. # TODO 这种情况需不需要将对于经销商的分账重新执行
  3730. return result
  3731. def closed(self, tradeRefundNo, errorCode="", errorDesc=""): # type:(str, str, str) -> bool
  3732. """
  3733. 退款申请成功了 表示合法的退款申请 但是由于某些原因业务进行不下去
  3734. """
  3735. result = self.__class__.objects.filter(
  3736. orderNo=self.orderNo,
  3737. status=self.Status.PROCESSING
  3738. ).update(
  3739. status=self.Status.CLOSED,
  3740. datetimeUpdated=datetime.datetime.now(),
  3741. tradeRefundNo=tradeRefundNo,
  3742. errorCode=errorCode,
  3743. errorDesc=errorDesc
  3744. )
  3745. return result
  3746. def processing(self): # type:() -> bool
  3747. result = self.__class__.objects.filter(
  3748. orderNo=self.orderNo,
  3749. status__in=[self.Status.CLOSED, self.Status.CREATED, self.Status.FAILURE]
  3750. ).update(
  3751. status=self.Status.PROCESSING,
  3752. datetimeUpdated=datetime.datetime.now()
  3753. )
  3754. return result
  3755. @property
  3756. def is_fail(self):
  3757. """ 是否订单失败 """
  3758. return self.status == self.Status.FAILURE
  3759. @property
  3760. def is_closed(self):
  3761. """ 退款单是否已经关闭 目前这个状态没有用 适用于用户手动取消退款申请 """
  3762. return self.status == self.Status.CLOSED
  3763. @property
  3764. def is_success(self):
  3765. """ 退款已经成功 """
  3766. return self.status == self.Status.SUCCESS
  3767. @property
  3768. def is_created(self):
  3769. return self.status == self.Status.CREATED
  3770. @property
  3771. def is_apply(self):
  3772. """ 是否已经发出退款申请 """
  3773. return self.status in [self.Status.CREATED, self.Status.PROCESSING]
  3774. @property
  3775. def is_processing(self):
  3776. return self.status == self.Status.PROCESSING
  3777. @property
  3778. def is_successful(self):
  3779. """ 保留旧的方法 """
  3780. return self.status in [self.Status.SUCCESS, self.Status.PROCESSING]
  3781. @property
  3782. def pay_sub_order(self):
  3783. # type: ()->RechargeRecord
  3784. if not hasattr(self, '__pay_sub_order__'):
  3785. pay_order = RechargeRecord.objects(id = str(self.rechargeObjId)).first()
  3786. setattr(self, '__pay_sub_order__', pay_order)
  3787. return getattr(self, '__pay_sub_order__')
  3788. @pay_sub_order.setter
  3789. def pay_sub_order(self, order):
  3790. setattr(self, '__pay_sub_order__', order)
  3791. @property
  3792. def refund_order_record(self):
  3793. if not hasattr(self, '__refund_order_record__'):
  3794. order = RechargeRecord.objects(orderNo = self.orderNo).first()
  3795. setattr(self, '__refund_order_record__', order)
  3796. return getattr(self, '__refund_order_record__')
  3797. @refund_order_record.setter
  3798. def refund_order_record(self, refund_order_record):
  3799. setattr(self, '__refund_order_record__', refund_order_record)
  3800. @classmethod
  3801. def get_record(cls, order_no):
  3802. return cls.objects(orderNo = order_no).first()
  3803. @property
  3804. def user(self):
  3805. # type: ()->MyUser
  3806. return self.pay_sub_order.user
  3807. class VCardConsumeRecord(OrderRecordBase):
  3808. orderNo = StringField(verbose_name = "订单号", default = "")
  3809. cardId = StringField(verbose_name = "实体卡ID", default = "")
  3810. dealerId = StringField(verbose_name = "经销商ID", default = "")
  3811. # : 在部分场合,设备会失败启动,但这时应该添加该字段为False方便追溯
  3812. remarks = StringField(verbose_name = "备注", default = "")
  3813. desc = StringField(verbose_name = "描述", default = "")
  3814. finishedTime = DateTimeField(verbose_name = '结束时间', default = datetime.datetime.now)
  3815. servicedInfo = DictField(verbose_name = '服务内容', default = {})
  3816. attachParas = DictField(verbose_name = '服务内容', default = {})
  3817. # 本次消费消耗的额度
  3818. consumeData = DictField(verbose_name = '消耗的额度', default = {})
  3819. # 日限额消耗的额度
  3820. consumeDayData = DictField(verbose_name = '消耗的额度', default = {})
  3821. meta = {
  3822. "collection": "VCardConsumeRecord",
  3823. "db_alias": "default",
  3824. }
  3825. @staticmethod
  3826. def make_no(identifier):
  3827. # time:14;main_type:1;sub_type:1;identifier:15;extra:5
  3828. return '{time}{main_type}{sub_type}{identifier}{reserved}'.format(
  3829. time = datetime.datetime.now().strftime("%Y%m%d%H%M%S"),
  3830. main_type = OrderMainType.CONSUME,
  3831. sub_type = UserConsumeSubType.VIRTUAL_CARD,
  3832. identifier = '{:0>15}'.format(identifier),
  3833. reserved = get_random_str(5, string.digits + string.uppercase))
  3834. @staticmethod
  3835. def paginate(cardId, pageIndex, pageSize):
  3836. records = VCardConsumeRecord.objects.filter(cardId = cardId) \
  3837. .order_by('-dateTimeAdded').skip((pageIndex - 1) * pageSize).limit(pageSize)
  3838. if not records:
  3839. return 0, []
  3840. dataList = []
  3841. for record in records:
  3842. newData = {
  3843. 'id': str(record.id),
  3844. 'createdTime': record.dateTimeAdded.strftime(Const.DATETIME_FMT),
  3845. 'address': record.address,
  3846. 'devNo': record.devNo,
  3847. 'openId': record.openId,
  3848. 'nickname': record.nickname,
  3849. 'logicalCode': record.logicalCode,
  3850. 'groupNumber': record.groupNumber,
  3851. 'groupName': record.groupName,
  3852. 'unit': record.consumeData.get("unit", ""),
  3853. 'devTypeCode': record.devTypeCode,
  3854. 'devTypeName': record.dev_type_name,
  3855. 'isNormal': True,
  3856. 'ownerId': record.dealerId
  3857. }
  3858. amount = record.consumeData.get("count", "")
  3859. if amount:
  3860. amount = Quantity(amount)
  3861. newData.update({'amount': amount})
  3862. if record.servicedInfo:
  3863. newData.update(record.servicedInfo)
  3864. dataList.append(newData)
  3865. return len(dataList), dataList
  3866. @property
  3867. def amount(self):
  3868. _amount = self.consumeData.get("count", None)
  3869. if _amount is not None:
  3870. return str(Quantity(_amount))
  3871. else:
  3872. return ""
  3873. @property
  3874. def unit(self):
  3875. return self.consumeData.get("unit", "")
  3876. @property
  3877. def created_date(self):
  3878. return self.dateTimeAdded
  3879. @property
  3880. def completion_date(self):
  3881. return self.finishedTime
  3882. @property
  3883. def device_start_time(self):
  3884. return self.dateTimeAdded
  3885. @property
  3886. def device_finished_time(self):
  3887. return self.finishedTime
  3888. def to_user_detail(self):
  3889. """
  3890. 消费记录对于用户端显示的详细信息
  3891. :return:
  3892. """
  3893. data = {
  3894. 'id': str(self.id),
  3895. 'orderNo': self.orderNo,
  3896. 'createdTime': self.dateTimeAdded.strftime(Const.DATETIME_FMT),
  3897. 'deviceStatTime': self.device_start_time,
  3898. 'address': self.address,
  3899. 'devNo': self.devNo,
  3900. 'logicalCode': self.logicalCode,
  3901. 'groupNumber': self.groupNumber,
  3902. 'groupName': self.groupName,
  3903. 'devTypeCode': self.devTypeCode,
  3904. 'devTypeName': self.dev_type_name,
  3905. 'isNormal': True,
  3906. 'amount': self.amount,
  3907. 'unit': self.unit,
  3908. 'ownerId': self.dealerId,
  3909. 'consumeType': 'mobile_vcard',
  3910. 'openId': self.openId,
  3911. 'userNickname': self.nickname
  3912. }
  3913. # 兼容之前的订单 凡是没有订单状态机的订单,一律视为 之前的订单
  3914. if hasattr(self, "state") and self.state:
  3915. data.update({"orderStatus": self.state})
  3916. if self.servicedInfo:
  3917. data.update(self.servicedInfo)
  3918. return data
  3919. @property
  3920. def ownerId(self):
  3921. return self.dealerId
  3922. @property
  3923. def owner(self):
  3924. from apps.web.dealer.models import Dealer
  3925. _attr_name = '__my_owner__'
  3926. if hasattr(self, _attr_name):
  3927. return getattr(self, _attr_name)
  3928. else:
  3929. if not self.dealerId:
  3930. return None
  3931. else:
  3932. dealer = Dealer.objects(id = self.dealerId).first()
  3933. setattr(self, _attr_name, dealer)
  3934. return dealer
  3935. @property
  3936. def summary(self):
  3937. return {
  3938. 'id': str(self.id),
  3939. 'consumeType': 'mobile_vcard',
  3940. 'ownerId': self.ownerId,
  3941. 'logicalCode': self.logicalCode,
  3942. 'groupName': self.groupName,
  3943. 'devTypeName': self.dev_type_name,
  3944. 'amount': self.amount,
  3945. 'unit': self.unit,
  3946. 'createdTime': self.dateTimeAdded.strftime(Const.DATETIME_FMT),
  3947. }
  3948. VirtualCardQuotaList = List[Dict[str, Union[float, unicode]]]
  3949. class UserVirtualCard(Searchable):
  3950. cardNo = StringField(verbose_name = "卡编号", default = "")
  3951. cardTypeId = StringField(verbose_name = u'卡的类型ID', default = '') # 如果卡停止销售了,这里就可以不允许续充
  3952. openIds = ListField(verbose_name = "绑定的用户") # 里面放的是MyUser的Id
  3953. cardName = StringField(verbose_name = "卡名称", default = "")
  3954. ownerOpenId = StringField(verbose_name = "卡的所有者", default = "")
  3955. nickname = StringField(verbose_name = "昵称", default = "")
  3956. logicalCode = StringField(verbose_name = u'发卡的时候记录的设备编码', default = '')
  3957. groupId = StringField(verbose_name = u'发卡的时候记录的地址', default = '')
  3958. dealerId = StringField(verbose_name = "卡的发布老板", default = "")
  3959. groupIds = ListField(verbose_name = "可用使用卡的地址", default = []) # 空表示没有地址,*表示所有地址下可用
  3960. devTypeList = ListField(verbose_name = "设备类型清单", default = [])
  3961. price = MonetaryField(verbose_name = "卡的售价", default = RMB('0.00'))
  3962. periodDays = IntField(verbose_name = "卡的可用天数", default = 30)
  3963. expiredTime = DateTimeField(verbose_name = "过期时间",
  3964. default = lambda: datetime.datetime.now() + datetime.timedelta(days = 365))
  3965. userLimit = IntField(verbose_name = "用户数量限制", default = 10)
  3966. userDesc = StringField(verbose_name = "描述", default = "")
  3967. dayQuota = ListField(verbose_name = "日限额度", default = []) # {'unit':u'次','count':2}
  3968. quota = ListField(verbose_name = "总的额度", default = [])
  3969. frozenQuota = ListField(verbose_name = u'冻结额度', default = [])
  3970. startTime = DateTimeField(verbose_name = "买卡的时间", default = datetime.datetime.now)
  3971. dayUsed = DictField(verbose_name = "日消耗情况", default = {}) # {'2018-12-26':[{'unit':'次','count':3}]}
  3972. quotaUsed = ListField(verbose_name = "总的消耗情况", default = []) # [{'unit':'次','count':3}]
  3973. status = StringField(verbose_name = "卡状态", default = 'normal')
  3974. # expired:过期,耗尽:exhausted, 未激活:nonactivated(经销商开虚拟卡),used(经销商开卡解绑了,并未过期)
  3975. remark = StringField(verbose_name = "卡片备注", default = "")
  3976. continueTime = DateTimeField(verbose_name="买卡的时间", default = None) # 接续时间,用于续费后,下一次卡接续开始
  3977. ongoingList = ListField(DictField(), verbose = u'冻结的配额')
  3978. managerialAppId = StringField(verbose_name = "管理公众号AppId", default = None)
  3979. managerialOpenId = StringField(verbose_name = "管理openId", default = None)
  3980. # 坤元虚拟卡特有:功率
  3981. power = IntField(verbose_name="包月卡功率限制", default=0)
  3982. meta = {
  3983. "collection": "UserVirtualCard",
  3984. "db_alias": "default",
  3985. }
  3986. search_fields = ('nickname', 'cardNo', 'cardName')
  3987. REVERSE_DAY = 3
  3988. @property
  3989. def payVia(self):
  3990. return "virtualCard"
  3991. @staticmethod
  3992. def make_no():
  3993. timestamp = generate_timestamp_ex()
  3994. random_int = random.randint(1, 1000)
  3995. return "%d%03d" % (timestamp, random_int)
  3996. # --------------------------------------------------------------- 所有的虚拟卡的新方法 需要兼容以前 一定需要严格测试
  3997. @staticmethod
  3998. def find_match_unit(quotaList, package):
  3999. """
  4000. 从额度列表中找到符合套餐单位的 返回index
  4001. :param quotaList: [{"unit": "次", "count": 15}, {"unit": "分钟", "count": 600}]
  4002. :param package: {"unit": "分钟", "time": 30}
  4003. """
  4004. # 只有一种额度的 走之前的流程 不要影响客户的使用
  4005. if len(quotaList) == 1:
  4006. return UserVirtualCard._find_match_unit(quotaList, package)
  4007. # 多单位的情况下 不再对单位进行转换 如果单位不匹配 就直接跳过
  4008. pCount, pUnit = package["time"], package["unit"]
  4009. for _index, _quota in enumerate(quotaList):
  4010. if _quota["unit"] != pUnit or float(_quota["count"]) < float(pCount):
  4011. continue
  4012. ii = _index
  4013. break
  4014. else:
  4015. ii = -1
  4016. return ii
  4017. @staticmethod
  4018. def calc_left_quota(quotaList, usedList):
  4019. """
  4020. 计算剩余的额度
  4021. :param quotaList: 总的额度列表
  4022. :param usedList: 已经使用额度情况
  4023. """
  4024. # TODO 目前使用双循环解决,其实可以加一个单位排序,一次遍历搞定 时间复杂度将会降低一个量级
  4025. leftQuotaList = list()
  4026. for _quota in quotaList:
  4027. _qCount, _qUnit = float(_quota["count"]), _quota["unit"]
  4028. for _used in usedList:
  4029. # 单位有匹配的 则直接计算额度
  4030. if _used["unit"] == _qUnit:
  4031. _usedQuota = {"count": float(max(_qCount-_used["count"], 0)), "unit": _qUnit}
  4032. break
  4033. else:
  4034. continue
  4035. # 没有找到这种单位 说明这种额度没有使用过 则剩余额度为满值
  4036. else:
  4037. _usedQuota = {"count": float(_qCount), "unit": _qUnit}
  4038. leftQuotaList.append(_usedQuota)
  4039. return leftQuotaList
  4040. def calc_left_total_quota(self):
  4041. """
  4042. 计算剩余的总额度 将所有的冻结额度都算上使用的
  4043. :return:
  4044. """
  4045. usedMap, frozenMap = dict(), dict()
  4046. for _item in self.quotaUsed:
  4047. if _item["unit"] in usedMap:
  4048. usedMap[_item["unit"]] += float(_item["count"])
  4049. else:
  4050. usedMap[_item["unit"]] = float(_item["count"])
  4051. for _on in self.ongoingList:
  4052. if _on["quota"]["unit"] in frozenMap:
  4053. frozenMap[_on["quota"]["unit"]] += float(_on["quota"]["count"])
  4054. else:
  4055. frozenMap[_on["quota"]["unit"]] = float(_on["quota"]["count"])
  4056. quotaList = self.quota
  4057. usedList = [{"count": _v, "unit": _k} for _k, _v in dict(Counter(usedMap)+Counter(frozenMap)).items()]
  4058. return self.calc_left_quota(quotaList, usedList)
  4059. def calc_left_day_quota(self, dayKey):
  4060. """
  4061. 计算当天剩下的总额度 注意key的匹配 将所有的当天的额度都算上使用的
  4062. :param dayKey:
  4063. :return:
  4064. """
  4065. usedMap, frozenMap = dict(), dict()
  4066. for _item in self.dayUsed.get(dayKey, list()):
  4067. if _item["unit"] in usedMap:
  4068. usedMap[_item["unit"]] += _item["count"]
  4069. else:
  4070. usedMap[_item["unit"]] = _item["count"]
  4071. for _on in self.ongoingList:
  4072. if _on["dayKey"] != dayKey:
  4073. continue
  4074. if _on["dayQuota"]["unit"] in frozenMap:
  4075. frozenMap[_on["dayQuota"]["unit"]] += _on["dayQuota"]["count"]
  4076. else:
  4077. frozenMap[_on["dayQuota"]["unit"]] = _on["dayQuota"]["count"]
  4078. quotaList = self.dayQuota
  4079. usedList = [{"count": _v, "unit": _k} for _k, _v in dict(Counter(usedMap) + Counter(frozenMap)).items()]
  4080. return self.calc_left_quota(quotaList, usedList)
  4081. # --------------------------------------------------------------- 新方法结束
  4082. @staticmethod
  4083. def _find_match_unit(quotaList, package):
  4084. # type:(VirtualCardQuotaList, dict)->int
  4085. match = False
  4086. ii = 0
  4087. for t1Dict in quotaList:
  4088. unit = t1Dict['unit']
  4089. count = t1Dict['count']
  4090. if unit == u'次' and count >= 1:
  4091. match = True
  4092. break
  4093. elif (unit in [u'分钟', u'小时', u'天']) and (package['unit'] in [u'分钟', u'小时', u'天']):
  4094. if unit == u'小时':
  4095. t1Count = count * 60
  4096. elif unit == u'天':
  4097. t1Count = count * 60 * 24
  4098. else:
  4099. t1Count = count
  4100. if package['unit'] == u'小时':
  4101. pCount = float(package['time']) * 60
  4102. elif package['unit'] == u'天':
  4103. pCount = float(package['time']) * 60 * 24
  4104. else:
  4105. pCount = float(package['time'])
  4106. if t1Count >= pCount:
  4107. match = True
  4108. break
  4109. elif (unit in [u'分钟', u'小时', u'天']) and package['unit'] == u'度': # 直接按照12小时计算
  4110. if unit == u'小时':
  4111. t1Count = count * 60
  4112. elif unit == u'天':
  4113. t1Count = count * 60 * 24
  4114. else:
  4115. t1Count = count
  4116. if t1Count >= 6.0 * 60: # 6小时打底
  4117. match = True
  4118. break
  4119. elif unit == u'币' and count >= package['coins']:
  4120. match = True
  4121. break
  4122. elif unit == u'度' and package['unit'] == u'度' and count >= package['time']:
  4123. match = True
  4124. break
  4125. elif unit == u'度' and package['unit'] != u'度' and count >= 3.0:
  4126. match = True
  4127. break
  4128. elif package['unit'] == u'次':
  4129. if unit == u'分钟' and count >= 360.0:
  4130. match = True
  4131. break
  4132. elif unit == u'小时' and count >= 6.0:
  4133. match = True
  4134. break
  4135. elif unit == u'天' and count >= 0.25:
  4136. match = True
  4137. break
  4138. elif unit == u'度' and count >= 3.0: # 默认写死3度电,多退少补
  4139. match = True
  4140. break
  4141. elif unit == u'币' and count >= package['coins']:
  4142. match = True
  4143. break
  4144. ii += 1
  4145. return ii if match else -1
  4146. @staticmethod
  4147. def find_match_unit_and_can_use_count(quotaList, package):
  4148. """
  4149. 找出虚拟卡能否使用多少次 某些设备的特殊需求 例如和动
  4150. """
  4151. # 兼容之前的
  4152. if len(quotaList) == 1:
  4153. usage_count = 0
  4154. for t1Dict in quotaList:
  4155. unit = t1Dict['unit']
  4156. count = t1Dict['count']
  4157. if unit == u'次' and count >= 1:
  4158. usage_count = count
  4159. break
  4160. elif (unit in [u'分钟', u'小时', u'天']) and (package['unit'] in [u'分钟', u'小时', u'天']):
  4161. if unit == u'小时':
  4162. t1Count = count * 60
  4163. elif unit == u'天':
  4164. t1Count = count * 60 * 24
  4165. else:
  4166. t1Count = count
  4167. if package['unit'] == u'小时':
  4168. pCount = float(package['time']) * 60
  4169. elif package['unit'] == u'天':
  4170. pCount = float(package['time']) * 60 * 24
  4171. else:
  4172. pCount = float(package['time'])
  4173. if t1Count >= pCount:
  4174. usage_count = t1Count // pCount
  4175. break
  4176. elif (unit in [u'分钟', u'小时', u'天']) and package['unit'] == u'度': # 直接按照12小时计算
  4177. if unit == u'小时':
  4178. t1Count = count * 60
  4179. elif unit == u'天':
  4180. t1Count = count * 60 * 24
  4181. else:
  4182. t1Count = count
  4183. if t1Count >= 6.0 * 60: # 6小时打底
  4184. usage_count = t1Count // 6.0 * 60
  4185. break
  4186. elif unit == u'币' and count >= package['coins']:
  4187. usage_count = count // float(package['coins'])
  4188. break
  4189. elif unit == u'度' and package['unit'] == u'度' and count >= package['time']:
  4190. usage_count = count // float(package['time'])
  4191. break
  4192. elif unit == u'度' and package['unit'] != u'度' and count >= 3.0:
  4193. usage_count = count // 3
  4194. break
  4195. elif package['unit'] == u'次':
  4196. if unit == u'分钟' and count >= 360.0:
  4197. usage_count = count // 360.0
  4198. break
  4199. elif unit == u'小时' and count >= 6.0:
  4200. usage_count = count // 6
  4201. break
  4202. elif unit == u'天' and count >= 0.25:
  4203. usage_count = count // 0.25
  4204. break
  4205. elif unit == u'度' and count >= 3.0: # 默认写死3度电,多退少补
  4206. usage_count = count // 3
  4207. break
  4208. elif unit == u'币' and count >= package['coins']:
  4209. usage_count = count // float(package['coins'])
  4210. break
  4211. return int(usage_count)
  4212. # 两种单位的
  4213. else:
  4214. pCount, pUnit = package["time"], package["unit"]
  4215. for _index, _quota in enumerate(quotaList):
  4216. # 单位不一致 或者数量不够的 直接跳过
  4217. if _quota["unit"] != pUnit or float(_quota["count"]) < float(pCount):
  4218. continue
  4219. # 然后求出count
  4220. if pUnit == u"次":
  4221. return _quota["count"]
  4222. # 时间的 由于单位一样 直接除法
  4223. else:
  4224. return _quota["count"] // pCount
  4225. else:
  4226. return 0
  4227. @staticmethod
  4228. def count_by_unit(unit, package):
  4229. # type:(unicode, dict)->Union[float, int]
  4230. """
  4231. 根据单位,计算消耗量
  4232. 根据套餐计算以及单位,计算需要扣除的指标数量。比如单位是:币,那就直接看套餐多少币,就需要扣除多少币
  4233. :param unit:
  4234. :param package:
  4235. :return:
  4236. """
  4237. if unit == u'次':
  4238. return 1
  4239. elif unit == u'币':
  4240. return package['coins']
  4241. elif unit == u'分钟':
  4242. if package['unit'] == u'分钟':
  4243. return float(package['time'])
  4244. elif package['unit'] == u'小时':
  4245. return float(package['time']) * 60.0
  4246. elif package['unit'] == u'天':
  4247. return float(package['time']) * 60 * 24.0
  4248. else:
  4249. return 6.0 * 60
  4250. elif unit == u'小时':
  4251. if package['unit'] == u'分钟':
  4252. return float(package['time']) / 60.0
  4253. elif package['unit'] == u'小时':
  4254. return float(package['time'])
  4255. elif package['unit'] == u'天':
  4256. return float(package['time']) * 24.0
  4257. else:
  4258. return 6.0
  4259. elif unit == u'天':
  4260. if package['unit'] == u'分钟':
  4261. return float(package['time']) / 360.0
  4262. elif package['unit'] == u'小时':
  4263. return float(package['time']) / 24.0
  4264. elif package['unit'] == u'天':
  4265. return float(package['time'])
  4266. else:
  4267. return 0.25
  4268. elif unit == u'度':
  4269. if package['unit'] == u'度':
  4270. return float(package['time'])
  4271. else:
  4272. return 3.0
  4273. else:
  4274. raise ValueError(u'does not support unit %s' % (unit,))
  4275. def can_use_today(self, package):
  4276. # type:(dict)->bool
  4277. """
  4278. 检查今天是否可以用卡
  4279. :param package:
  4280. :return:
  4281. """
  4282. if package.get("isTempPackage"):
  4283. return False
  4284. # 非正常状态的卡都不让用
  4285. if self.status not in ["normal", "warning"]:
  4286. return False
  4287. # 先计算出总的剩余额度,以及每日的剩余额度
  4288. leftQuota = self.calc_left_total_quota()
  4289. dayKey = datetime.datetime.now().strftime("%Y-%m-%d")
  4290. leftDayQuota = self.calc_left_day_quota(dayKey)
  4291. # 剩余总额度以及剩余日限额度检查没有问题,就直接返回
  4292. if UserVirtualCard.find_match_unit(leftQuota, package) >= 0 and UserVirtualCard.find_match_unit(leftDayQuota, package) >= 0:
  4293. return True
  4294. return False
  4295. def freeze_quota(self, package, transaction_id): # type:(dict, str) -> bool
  4296. """
  4297. 消费一次,包括扣掉总额,增加消费记录
  4298. :param package:
  4299. :param transaction_id: 交易的ID
  4300. :return:
  4301. """
  4302. try:
  4303. # 找出额度单元,然后扣掉
  4304. leftQuota = self.calc_left_total_quota()
  4305. dayKey = today_format_str()
  4306. leftDayQuota = self.calc_left_day_quota(dayKey)
  4307. allIndex = UserVirtualCard.find_match_unit(leftQuota, package)
  4308. dayIndex = UserVirtualCard.find_match_unit(leftDayQuota, package)
  4309. if allIndex < 0 or dayIndex < 0:
  4310. raise InsufficientFundsError(u'虚拟卡配额不足')
  4311. # 先处理总额度
  4312. unit = leftQuota[allIndex]['unit']
  4313. consume = {'unit': unit, 'count': UserVirtualCard.count_by_unit(unit, package)}
  4314. # 处理每日额度
  4315. unit = leftDayQuota[dayIndex]['unit']
  4316. day_consume = {'unit': unit, 'count': UserVirtualCard.count_by_unit(unit, package)}
  4317. query = {'_id': self.id,
  4318. 'ongoingList.transaction_id': {'$ne': transaction_id}}
  4319. update = {
  4320. '$addToSet': {'ongoingList': {
  4321. 'transaction_id': transaction_id,
  4322. 'quota': consume,
  4323. 'dayQuota': day_consume,
  4324. 'dayKey': dayKey
  4325. }}}
  4326. result = self.get_collection().update_one(filter=query, update=update, upsert = False)
  4327. return bool(result.modified_count == 1)
  4328. finally:
  4329. try:
  4330. day3Key = (datetime.datetime.now() - datetime.timedelta(days = self.REVERSE_DAY)).strftime('%Y-%m-%d')
  4331. for dk in self.dayUsed.keys():
  4332. if dk <= day3Key:
  4333. self.dayUsed.pop(dk)
  4334. self.save()
  4335. except Exception as e:
  4336. logger.exception('save user vcard e={}'.format(e))
  4337. def clear_frozen_quota(self, transaction_id, usedTime, spendElec, backCoins):
  4338. """
  4339. 清除冻结的额度,并按照一定的比例退还
  4340. """
  4341. def calc_left_by_unit(count, unit, ut, ue, uc):
  4342. if unit in [u'分钟', u'小时', u'天']:
  4343. if unit == u'分钟':
  4344. ut = int(ut)
  4345. elif unit == u'小时':
  4346. ut = round(ut / 60.0, 2)
  4347. else:
  4348. ut = round(ut / (60.0 * 24.0), 2)
  4349. leftCount = count - ut
  4350. elif unit in [u'度']:
  4351. leftCount = count - ue
  4352. elif unit in [u'次']: # 如果没有用,比如没有插电,一样,不能扣除配额
  4353. if int(ut) == 0 and int(ue) == 0:
  4354. leftCount = count
  4355. else:
  4356. leftCount = 0
  4357. elif unit in [u'币']: # 注意,这个是已经算好的退币数额
  4358. leftCount = uc
  4359. else:
  4360. leftCount = 0
  4361. return max(leftCount, 0)
  4362. frozen_item = None
  4363. for item in self.ongoingList:
  4364. if item['transaction_id'] == transaction_id:
  4365. frozen_item = item
  4366. break
  4367. if not frozen_item:
  4368. logger.debug('not find this frozen item. card = {}, consume id = {}'.format(repr(self), transaction_id))
  4369. return False, None, None
  4370. consumeTotal = frozen_item['quota']
  4371. consumeTotalLeft = calc_left_by_unit(consumeTotal['count'], consumeTotal['unit'], usedTime, spendElec,
  4372. backCoins)
  4373. consumeTotal['count'] -= float(consumeTotalLeft)
  4374. consumeDay = frozen_item['dayQuota']
  4375. consumeDayLeft = calc_left_by_unit(consumeDay['count'], consumeDay['unit'],
  4376. usedTime, spendElec, backCoins)
  4377. consumeDay['count'] -= float(consumeDayLeft)
  4378. day_key = frozen_item['dayKey']
  4379. clean_day_key = (datetime.datetime.now() - datetime.timedelta(days = self.REVERSE_DAY)).strftime('%Y-%m-%d')
  4380. if day_key <= clean_day_key:
  4381. match_quota_used = None
  4382. for item in self.quotaUsed:
  4383. if item['unit'] == consumeTotal['unit']:
  4384. match_quota_used = item
  4385. break
  4386. if not match_quota_used:
  4387. self.quotaUsed.append({'unit': consumeTotal['unit'], 'count': 0})
  4388. self.save()
  4389. query = {
  4390. '_id': self.id,
  4391. 'ongoingList': {'$elemMatch': {'transaction_id': transaction_id}},
  4392. 'quotaUsed.unit': consumeTotal['unit']
  4393. }
  4394. update = {
  4395. '$inc': {
  4396. 'quotaUsed.$.count': consumeTotal['count']
  4397. },
  4398. '$pull': {'ongoingList': {'transaction_id': transaction_id}},
  4399. }
  4400. result = self.get_collection().update_one(
  4401. filter = query,
  4402. update = update,
  4403. upsert = False
  4404. )
  4405. modified = bool(result.modified_count == 1)
  4406. return modified, consumeTotal, consumeDay
  4407. else:
  4408. match_quota_used = None
  4409. for item in self.quotaUsed:
  4410. if item['unit'] == consumeTotal['unit']:
  4411. match_quota_used = item
  4412. break
  4413. if not match_quota_used:
  4414. self.quotaUsed.append({'unit': consumeTotal['unit'], 'count': 0})
  4415. if day_key not in self.dayUsed:
  4416. self.dayUsed[day_key] = []
  4417. match_day_used = None
  4418. for item in self.dayUsed.get(day_key, []):
  4419. if item['unit'] == consumeDay['unit']:
  4420. match_day_used = item
  4421. break
  4422. if not match_day_used:
  4423. self.dayUsed[day_key].append({'unit': consumeDay['unit'], 'count': 0})
  4424. self.save()
  4425. query = {
  4426. '_id': self.id,
  4427. 'ongoingList': {'$elemMatch': {'transaction_id': transaction_id}},
  4428. 'quotaUsed.unit': consumeTotal['unit'],
  4429. 'dayUsed.{}.unit'.format(day_key): consumeDay['unit']
  4430. }
  4431. update = {
  4432. '$inc': {
  4433. 'quotaUsed.$.count': consumeTotal['count'],
  4434. 'dayUsed.{}.$.count'.format(day_key): consumeDay['count']
  4435. },
  4436. '$pull': {'ongoingList': {'transaction_id': transaction_id}},
  4437. }
  4438. result = self.get_collection().update_one(
  4439. filter = query,
  4440. update = update,
  4441. upsert = False)
  4442. modified = bool(result.modified_count == 1)
  4443. self.reload()
  4444. return modified, consumeTotal, consumeDay
  4445. def recover_frozen_quota(self, transaction_id):
  4446. """
  4447. 清除虚拟卡的冻结额度 不退换
  4448. """
  4449. query = {'_id': self.id}
  4450. update = {
  4451. '$pull': {'ongoingList': {'transaction_id': transaction_id}},
  4452. }
  4453. result = self.get_collection().update_one(
  4454. filter = query,
  4455. update = update,
  4456. upsert = False
  4457. )
  4458. return bool(result.modified_count == 1)
  4459. def consume(self, openId, group, dev, package, attachParas, nickname='', orderNo=None):
  4460. """
  4461. 消费一次,包括扣掉总额,增加消费记录
  4462. :param openId:
  4463. :param group:
  4464. :param dev:
  4465. :param package:
  4466. :param attachParas:
  4467. :param nickname:
  4468. :param orderNo: 订单编号
  4469. :return:
  4470. """
  4471. # 找出额度单元,然后扣掉
  4472. leftQuota = self.calc_left_total_quota()
  4473. dayKey = datetime.datetime.now().strftime("%Y-%m-%d")
  4474. leftDayQuota = self.calc_left_day_quota(dayKey)
  4475. allIndex = UserVirtualCard.find_match_unit(leftQuota, package)
  4476. dayIndex = UserVirtualCard.find_match_unit(leftDayQuota, package)
  4477. if allIndex < 0 or dayIndex < 0:
  4478. return None
  4479. # 先处理总额度
  4480. unit = leftQuota[allIndex]['unit']
  4481. consumeData = {'unit': unit, 'count': UserVirtualCard.count_by_unit(unit, package)}
  4482. match = False
  4483. for tDict in self.quotaUsed:
  4484. if tDict['unit'] == unit:
  4485. tDict['count'] = float(tDict['count']) + float(consumeData['count'])
  4486. match = True
  4487. break
  4488. if not match:
  4489. self.quotaUsed.append(consumeData)
  4490. # 处理每日额度
  4491. unit = leftDayQuota[dayIndex]['unit']
  4492. consumeDayData = {'unit': unit, 'count': UserVirtualCard.count_by_unit(unit, package)}
  4493. match = False
  4494. for tDict in self.dayUsed.get(dayKey, []):
  4495. if tDict['unit'] == unit:
  4496. consumeDayData = {'unit': unit, 'count': UserVirtualCard.count_by_unit(unit, package)}
  4497. tDict['count'] = float(tDict['count']) + float(consumeDayData['count'])
  4498. match = True
  4499. break
  4500. if not match:
  4501. self.dayUsed[dayKey] = [consumeDayData]
  4502. # 清理掉3天以前的每日消费情况
  4503. day3Key = (datetime.datetime.now() - datetime.timedelta(days=3)).strftime('%Y-%m-%d')
  4504. for dk in self.dayUsed.keys():
  4505. if dk <= day3Key:
  4506. self.dayUsed.pop(dk)
  4507. try:
  4508. self.save()
  4509. except Exception as e:
  4510. logger.exception('[consume] save user vcard e={}'.format(e))
  4511. return None
  4512. # 增加一条消费记录
  4513. newRcd = VCardConsumeRecord(
  4514. orderNo = orderNo if orderNo is not None else VCardConsumeRecord.make_no(dev.logicalCode),
  4515. openId = openId,
  4516. nickname = nickname,
  4517. cardId = str(self.id),
  4518. dealerId = self.dealerId,
  4519. devNo = dev['devNo'],
  4520. devTypeCode = dev.devTypeCode,
  4521. devTypeName = dev.devTypeName,
  4522. logicalCode = dev['logicalCode'],
  4523. groupId = group['groupId'],
  4524. address = group['address'],
  4525. groupNumber = dev['groupNumber'],
  4526. groupName = group['groupName'],
  4527. attachParas = attachParas,
  4528. consumeData = consumeData,
  4529. consumeDayData = consumeDayData
  4530. )
  4531. try:
  4532. newRcd.save()
  4533. except Exception as e:
  4534. logger.error('[consume] save vcard consume e=%s' % e)
  4535. return None
  4536. return newRcd
  4537. def new_consume_record(self, dev, user, consumeTotal, attachParas=None, orderNo=None):
  4538. newRcd = VCardConsumeRecord(
  4539. orderNo=orderNo if orderNo is not None else VCardConsumeRecord.make_no(dev.logicalCode),
  4540. openId=user.openId,
  4541. nickname=user.nickname,
  4542. cardId=str(self.id),
  4543. dealerId=dev.ownerId,
  4544. devNo=dev.devNo,
  4545. devTypeCode = dev.devTypeCode,
  4546. devTypeName = dev.devTypeName,
  4547. logicalCode=dev.logicalCode,
  4548. groupId=dev.group.groupId,
  4549. address=dev.group.address,
  4550. groupNumber=dev.groupNumber,
  4551. groupName=dev.group.groupName,
  4552. attachParas=attachParas or dict(),
  4553. consumeData=consumeTotal,
  4554. consumeDayData=consumeTotal
  4555. )
  4556. try:
  4557. newRcd.save()
  4558. except Exception as e:
  4559. logger.error('[consume] save vcard consume e=%s' % e)
  4560. return None
  4561. return newRcd
  4562. def refund_quota(self, consumeRcd, usedTime, spendElec, backCoins):
  4563. """
  4564. 将虚拟卡的额度返回 对应的使用方法是consume
  4565. :param consumeRcd:
  4566. :param usedTime: 单位一定是分钟
  4567. :param spendElec: 单位一定是度
  4568. :param backCoins: 单位一定是金币
  4569. :return:
  4570. """
  4571. def calc_left_by_unit(count, unit, ut, ue, uc):
  4572. if unit in [u'分钟', u'小时', u'天']:
  4573. if unit == u'分钟':
  4574. ut = int(ut)
  4575. elif unit == u'小时':
  4576. ut = round(ut / 60.0, 2)
  4577. else:
  4578. ut = round(ut / (60.0 * 24.0), 2)
  4579. leftCount = count - ut
  4580. elif unit in [u'度']:
  4581. leftCount = count - ue
  4582. elif unit in [u'次']: # 如果没有用,比如没有插电,一样,不能扣除配额
  4583. if int(ut) == 0 and int(ue) == 0:
  4584. leftCount = count
  4585. else:
  4586. leftCount = 0
  4587. elif unit in [u'币']: # 注意,这个是已经算好的退币数额
  4588. leftCount = float(VirtualCoin(uc))
  4589. else:
  4590. leftCount = 0
  4591. return leftCount
  4592. # 当消费已经定下来的时候 则本次退换的单位实际上已经固定了
  4593. consumeLeftTemp = calc_left_by_unit(
  4594. consumeRcd.consumeData['count'],
  4595. consumeRcd.consumeData['unit'],
  4596. usedTime,
  4597. spendElec,
  4598. backCoins
  4599. )
  4600. consumeLeft = max(consumeLeftTemp,0)
  4601. consumeRcd.consumeData['count'] -= float(consumeLeft)
  4602. consumeDayLeft = calc_left_by_unit(
  4603. consumeRcd.consumeDayData['count'],
  4604. consumeRcd.consumeDayData['unit'],
  4605. usedTime,
  4606. spendElec,
  4607. backCoins
  4608. )
  4609. consumeRcd.consumeDayData['count'] -= float(consumeDayLeft)
  4610. # 刷新消费记录数据
  4611. try:
  4612. consumeRcd.save()
  4613. except Exception as e:
  4614. logger.exception('[refund_quota] save consume rcd error = %s' % e)
  4615. return
  4616. # 更新额度
  4617. for tDict in self.quotaUsed:
  4618. if tDict['unit'] != consumeRcd.consumeData['unit']:
  4619. continue
  4620. tDict['count'] -= consumeLeft
  4621. dayKey = datetime.datetime.now().strftime("%Y-%m-%d")
  4622. for tDict in self.dayUsed.get(dayKey, []):
  4623. if tDict['unit'] != consumeRcd.consumeDayData['unit']:
  4624. continue
  4625. tDict['count'] -= float(consumeDayLeft)
  4626. try:
  4627. self.save()
  4628. except Exception as e:
  4629. logger.exception('[refund_quota] save quota error=%s' % e)
  4630. # 返回本条记录余下的配套
  4631. returnUsedTime,returnSpendElec,returnBackCoins = usedTime,spendElec,backCoins
  4632. if consumeRcd.consumeData['unit'] in [u'分钟',u'小时',u'秒']:
  4633. return consumeLeftTemp,returnSpendElec,returnBackCoins
  4634. if consumeRcd.consumeData['unit'] in [u'度']:
  4635. return returnUsedTime,consumeLeftTemp,returnBackCoins
  4636. return returnUsedTime,returnSpendElec,returnBackCoins
  4637. # 卡续费
  4638. def continue_card(self):
  4639. if not self.continueTime:
  4640. return
  4641. vCard = VirtualCard.objects.get(id = self.cardTypeId)
  4642. # 以最新的属性作为 购买
  4643. self.price = vCard.price
  4644. self.dayQuota = vCard.dayQuota
  4645. self.userLimit = vCard.userLimit
  4646. self.cardName = vCard.cardName
  4647. self.quota = vCard.quota
  4648. self.userDesc = vCard.userDesc
  4649. self.dayUsed = {}
  4650. self.quotaUsed = []
  4651. self.status = 'normal'
  4652. self.expiredTime = self.continueTime + datetime.timedelta(seconds=vCard.periodDays * 24 * 3600)
  4653. self.continueTime = None
  4654. self.save()
  4655. return self
  4656. @staticmethod
  4657. def get_user_cards(openId, groupId, ownerId):
  4658. """
  4659. 获取用户所有的虚拟卡
  4660. 修改: 去掉continueTime的check判断 没意义
  4661. """
  4662. cardList = []
  4663. vCards = UserVirtualCard.objects.filter(
  4664. dealerId = ownerId, openIds = openId,
  4665. groupIds__in = ['*', groupId]
  4666. )
  4667. nowTime = datetime.datetime.now()
  4668. for obj in vCards:
  4669. if obj.startTime <= nowTime <= obj.expiredTime:
  4670. cardList.append(obj)
  4671. return cardList
  4672. def add_expired_time(self, days):
  4673. self.expiredTime += datetime.timedelta(days = int(days))
  4674. self.save()
  4675. def support_dev_type(self, devTypeCode):
  4676. devTypesCount = DeviceType.objects.filter(code = devTypeCode, id__in = self.devTypeList).count()
  4677. if devTypesCount <= 0:
  4678. return False
  4679. return True
  4680. @property
  4681. def groupInfo(self):
  4682. """
  4683. 获取虚拟卡的相应的 地址信息
  4684. :return:
  4685. """
  4686. if '*' in self.groupIds:
  4687. groups = [{'groupName': grp['groupName'], 'address': grp['address']} for grp in
  4688. Group.get_groups_by_group_ids(Group.get_group_ids_of_dealer(self.dealerId)).values()]
  4689. else:
  4690. groups = [{'groupName': grp['groupName'], 'address': grp['address']} for grp in
  4691. Group.get_groups_by_group_ids(self.groupIds).values()]
  4692. return groups
  4693. @property
  4694. def membersInfo(self):
  4695. """
  4696. 虚拟卡成员信息
  4697. :return:
  4698. """
  4699. users = MyUser.objects.filter(openId__in = self.openIds).only("openId", "nickname")
  4700. bindUsersList = list()
  4701. repeatUsers = set()
  4702. for user in users:
  4703. if user.openId in repeatUsers:
  4704. continue
  4705. # openId 相同去重
  4706. repeatUsers.add(user.openId)
  4707. tempDict = {
  4708. 'userName': user.nickname,
  4709. 'userId': user.openId
  4710. }
  4711. user.openId == self.ownerOpenId and tempDict.update({"role": "owner"})
  4712. bindUsersList.append(tempDict)
  4713. return bindUsersList
  4714. @property
  4715. def quotaInfo(self):
  4716. """
  4717. 获取和额度相关的信息
  4718. :return:
  4719. """
  4720. dayUsedList = self.dayUsed.get(datetime.datetime.now().strftime("%Y-%m-%d"), list())
  4721. dayUsed = 0
  4722. if len(dayUsedList) > 0:
  4723. dayUsed = dayUsedList[0]['count']
  4724. return {
  4725. 'quota': self.quota[0]['count'],
  4726. 'quotaUnit': self.quota[0]['unit'],
  4727. 'quotaUsed': round(self.quotaUsed[0]['count'], 2) if len(self.quotaUsed) > 0 else 0,
  4728. 'dayQuota': self.dayQuota[0]['count'],
  4729. 'dayQuotaUnit': self.dayQuota[0]['unit'],
  4730. 'dayUsed': round(dayUsed, 2)
  4731. }
  4732. @property
  4733. def devTypes(self):
  4734. devs = DeviceType.objects.filter(id__in=self.devTypeList).only("majorDeviceType", "name")
  4735. devTypes = list()
  4736. for _dev in devs: # type: DeviceType
  4737. devTypes.append({"devTypeName": _dev.name, "majorDeviceType": _dev.majorDeviceType})
  4738. return devTypes
  4739. def isOwner(self, openId):
  4740. """是不是卡的所有者"""
  4741. return openId == self.ownerOpenId
  4742. def to_dict(self):
  4743. """
  4744. 序列化
  4745. :return:
  4746. """
  4747. data = {
  4748. 'cardId': str(self.id),
  4749. 'cardNo': self.cardNo,
  4750. 'cardTypeId': self.cardTypeId,
  4751. 'cardName': self.cardName,
  4752. 'periodDays': self.periodDays,
  4753. 'createTime': self.startTime.strftime('%Y-%m-%d'),
  4754. 'expiredTime': self.expiredTime.strftime('%Y-%m-%d'),
  4755. 'userDesc': self.userDesc,
  4756. 'userLimit': self.userLimit,
  4757. 'userName': self.nickname,
  4758. 'logicalCode': self.logicalCode,
  4759. 'groupId': self.groupId,
  4760. 'remark': self.remark,
  4761. 'power': self.power
  4762. }
  4763. return data
  4764. def to_detail(self):
  4765. data = {
  4766. 'cardId': str(self.id),
  4767. 'cardNo': self.cardNo,
  4768. 'cardTypeId': self.cardTypeId,
  4769. 'cardName': self.cardName,
  4770. 'periodDays': self.periodDays,
  4771. 'createTime': self.startTime.strftime('%Y-%m-%d'),
  4772. 'expiredTime': self.expiredTime.strftime('%Y-%m-%d'),
  4773. 'userDesc': self.userDesc,
  4774. 'userLimit': self.userLimit,
  4775. 'userName': self.nickname,
  4776. 'logicalCode': self.logicalCode,
  4777. 'groupId': self.groupId,
  4778. 'remark': self.remark,
  4779. "groups": self.groupInfo,
  4780. "sharedMembers": self.membersInfo,
  4781. "devTypes": self.devTypes,
  4782. # 功率
  4783. 'power': self.power,
  4784. # 额度
  4785. 'quota': {
  4786. "day": self.dayQuota,
  4787. "total": self.quota
  4788. },
  4789. # 剩余的额度
  4790. 'leftQuota': {
  4791. "day": self.calc_left_day_quota(today_format_str()),
  4792. "total": self.calc_left_total_quota()
  4793. }
  4794. }
  4795. return data
  4796. # 以下是霍坡的特殊业务 之前是不支持按次 现在支持了 考虑业务合并
  4797. def can_use_hp_gate(self):
  4798. """霍珀虚拟卡 对于道闸计费 只要卡没过期就可以使用"""
  4799. # 尚未过期的虚拟卡可用于霍珀道闸
  4800. # 要求变更 虚拟卡过期后还能在使用24小时
  4801. if self.status not in ["normal", "warning"]:
  4802. return False
  4803. return self.expiredTime > datetime.datetime.now() - datetime.timedelta(hours = 24)
  4804. def consume_hp_gate(self, openId, group, dev, package = None, attachParas={}, nickname=''):
  4805. # type:(str, GroupDict, DeviceDict, dict, dict, str)->Optional[VCardConsumeRecord]
  4806. """
  4807. 消费一次
  4808. :param openId:
  4809. :param group:
  4810. :param dev:
  4811. :param package:
  4812. :param attachParas:
  4813. :param nickname:
  4814. :return:
  4815. """
  4816. # 增加一条消费记录
  4817. newRcd = VCardConsumeRecord(
  4818. orderNo = VCardConsumeRecord.make_no(dev.logicalCode),
  4819. openId = openId,
  4820. nickname = nickname,
  4821. cardId = str(self.id),
  4822. dealerId = self.dealerId,
  4823. devNo = dev['devNo'],
  4824. devTypeCode = dev.devTypeCode,
  4825. devTypeName = dev.devTypeName,
  4826. logicalCode = dev['logicalCode'],
  4827. groupId = group['groupId'],
  4828. address = group['address'],
  4829. groupNumber = dev['groupNumber'],
  4830. groupName = group['groupName'],
  4831. attachParas = attachParas,
  4832. remarks = u"道闸刷卡服务"
  4833. )
  4834. try:
  4835. newRcd.save()
  4836. self.record_gate_used_times()
  4837. except Exception as e:
  4838. logger.error('save vcard consume e=%s' % e)
  4839. return None
  4840. return newRcd
  4841. def record_gate_used_times(self):
  4842. """
  4843. 霍珀要求的 对于道闸次数进行监控,但是不能限制次数
  4844. :return:
  4845. """
  4846. dayKey = datetime.datetime.now().strftime("%Y-%m-%d")
  4847. cacheKey = "{}-{}".format(dayKey, self.cardNo)
  4848. times = serviceCache.get(cacheKey, 0)
  4849. times += 1
  4850. serviceCache.set(cacheKey, times, timeout = 60 * 60 * 24)
  4851. if times < 8:
  4852. status = "normal"
  4853. else:
  4854. status = "warning"
  4855. try:
  4856. self.update(status = status)
  4857. except Exception as e:
  4858. logger.exception(e)
  4859. return
  4860. class VirtualCardRechargeRecord(Searchable):
  4861. orderNo = StringField(verbose_name = u'使用RechargeRecord的orderNo关联两张表', required = True)
  4862. cardId = StringField(verbose_name = "虚拟卡ID", default = "")
  4863. cardNo = StringField(verbose_name = "虚拟卡卡号", default = "")
  4864. openId = StringField(verbose_name = "openId", default = "")
  4865. gateway = StringField(verbose_name = 'gateway', default = "")
  4866. groupId = StringField(verbose_name = "设备地址编号", default = "")
  4867. ownerId = ObjectIdField(verbose_name = "dealerId")
  4868. dateTimeAdded = DateTimeField(default = datetime.datetime.now, verbose_name = '生成时间')
  4869. # 以下字段后续废弃, 只做兼容使用
  4870. devNo = StringField(verbose_name = "设备ID", default = None)
  4871. logicalCode = StringField(verbose_name = "设备逻辑编码", default = None)
  4872. wxOrderNo = StringField(verbose_name = "orderNo", default = None)
  4873. money = MonetaryField(verbose_name = "付款金额", default = None)
  4874. coins = MonetaryField(verbose_name = "充值金额", default = None)
  4875. devTypeCode = StringField(verbose_name = "设备类型编码", default = None)
  4876. time = StringField(verbose_name = 'time', default = None)
  4877. devType = StringField(verbose_name = "设备类型", default = None)
  4878. address = StringField(verbose_name = "设备地址", default = None)
  4879. groupNumber = StringField(verbose_name = "设备", default = None)
  4880. groupName = StringField(verbose_name = "交易场地", default = None)
  4881. remarks = StringField(verbose_name = "备注", default = None)
  4882. status = StringField(verbose_name = "状态", default = None)
  4883. meta = {
  4884. "collection": "virtualCardRechargeRecord",
  4885. "db_alias": "default",
  4886. }
  4887. @classmethod
  4888. def get_link_orderNo_list(cls, cardId, startTime, endTime, openId = None):
  4889. records = cls.objects(
  4890. cardId = cardId, dateTimeAdded__gte = startTime, dateTimeAdded__lt = get_tomorrow_zero_time(endTime))
  4891. if openId:
  4892. records = records.filter(openId = openId)
  4893. return [record.orderNo for record in records]
  4894. @property
  4895. def created_date(self):
  4896. return self.dateTimeAdded
  4897. def is_refund_available(self, customer): # type:(CapitalUser) -> bool
  4898. return False
  4899. @property
  4900. def has_refund_order(self):
  4901. return False
  4902. @property
  4903. def isQuickPay(self):
  4904. return False
  4905. @classmethod
  4906. def issue(cls, order):
  4907. # type:(RechargeRecord)->VirtualCardRechargeRecord
  4908. return cls(
  4909. orderNo = order.orderNo,
  4910. cardId = order.attachParas['cardId'],
  4911. cardNo = order.attachParas['cardNo'],
  4912. openId = order.openId,
  4913. ownerId = order.ownerId,
  4914. groupId = order.groupId,
  4915. gateway = order.gateway,
  4916. dateTimeAdded = order.dateTimeAdded).save()
  4917. # 监督号的用户,用于和实际系统的用户进行关联
  4918. class MoniUser(Searchable):
  4919. moniAppId = StringField(verbose_name = 'moniAppId', default = '')
  4920. moniOpenId = StringField(verbose_name = 'moniOpenId', default = '')
  4921. openId = StringField(verbose_name = u'用户的主APPID', default = '')
  4922. isSubscribe = BooleanField(verbose_name = u'是否订阅', default = False)
  4923. checkTime = DateTimeField(verbose_name = u'检查时间')
  4924. subTime = DateTimeField(verbose_name = u'关注时间')
  4925. unsubTime = DateTimeField(verbose_name = u'取关时间')
  4926. subLogicalCode = StringField(verbose_name = u'关注来源logicalCode', default = '')
  4927. subDealerId = StringField(verbose_name = u'关注来源dealerId', default = '')
  4928. subAgentId = StringField(verbose_name = u'关注来源agentId', default = '')
  4929. subPoint = StringField(verbose_name = u'关注来源point', default = '')
  4930. meta = {
  4931. "collection": "moni_user",
  4932. "db_alias": "logdata",
  4933. }
  4934. @classmethod
  4935. def get_or_create(cls, moniOpenId, moniAppId, **kwargs):
  4936. """
  4937. 创建一条用户的记录
  4938. """
  4939. obj = cls.objects.filter(moniOpenId=moniOpenId, moniAppId=moniAppId).order_by("-id").first() or cls(moniOpenId=moniOpenId, moniAppId=moniAppId)
  4940. needUpdate = False
  4941. for _field in cls._fields.keys():
  4942. if _field in kwargs and kwargs[_field] != getattr(obj, _field, kwargs[_field]):
  4943. setattr(obj, _field, kwargs[_field])
  4944. needUpdate = True
  4945. if needUpdate:
  4946. obj = obj.save()
  4947. return obj
  4948. @classmethod
  4949. def get_or_delete(cls, moniOpenId, moniAppId):
  4950. cls.objects.filter(moniOpenId=moniOpenId, moniAppId=moniAppId).update(isSubscribe=False)
  4951. class AskStep(Searchable):
  4952. operKey = StringField(verbose_name = 'oper key', default = '')
  4953. answerText = StringField(verbose_name = 'answer text', default = '')
  4954. answerPics = StringField(verbose_name = 'answer picture', default = '') # '文件名称用,分割'
  4955. answerFunc = StringField(verbose_name = 'answer func', default = '')
  4956. @staticmethod
  4957. def replyUseDevice(moniOpenId, appId, **kwargs):
  4958. moniUser = MoniUser.objects.filter(moniOpenId = moniOpenId, moniAppId = appId).first()
  4959. if moniUser:
  4960. devLink = concat_user_login_entry_url(l=moniUser.subLogicalCode)
  4961. return [{'msgType': 'text', 'value': u'\ue231<a href="%s">戳我使用设备</a>\ue230' % devLink}]
  4962. return [{'msgType': 'text', 'value': u'您直接点击菜单栏的扫一扫,或者关闭当前页面重新扫码,即可使用'}]
  4963. @staticmethod
  4964. def replayQueryBalance(moniOpenId, appId, **kwargs):
  4965. openId = kwargs.get('openId')
  4966. agentId = kwargs.get('agentId')
  4967. users = MyUser.objects.filter(openId = openId)
  4968. balance = VirtualCoin(0.0)
  4969. for user in users:
  4970. balance += user.balance
  4971. url = concat_user_center_entry_url(agentId=agentId,
  4972. redirect=concat_front_end_url(uri='/user/index.html#/user/help'))
  4973. return [{'msgType': 'text',
  4974. 'value': u'您在系统总的余额是%s,如果界面上显示不对,请您先刷新页面,还是不行的话,<a href="%s">联系客服</a>' % (balance, url)}]
  4975. @staticmethod
  4976. def get_service_phone_from_rcd(consumeRcd):
  4977. dealer = Dealer.objects(id=consumeRcd.ownerId).first() # type: Dealer
  4978. return dealer.service_phone
  4979. @staticmethod
  4980. def replyGetToushuUrl(moniOpenId, appId, **kwargs):
  4981. agentId = kwargs.get('agentId')
  4982. url = concat_user_center_entry_url(agentId=agentId,
  4983. redirect=concat_front_end_url(uri='/user/index.html#/complaint/list'))
  4984. return [{'msgType': 'text', 'value': u'<a href="%s">投诉点这里</a>' % url}]
  4985. # 先检查用户的订单是否扣掉了钱。然后检查设备的连通性。
  4986. # 调用此函数的时候,一般是用户付款了,设备没有响应。
  4987. @staticmethod
  4988. def replyQueryConsumeRcd(moniOpenId, appId, **kwargs):
  4989. try:
  4990. endTime = datetime.datetime.now()
  4991. startTime = endTime - datetime.timedelta(days = 7)
  4992. consumeRcd = ConsumeRecord.objects.filter(openId = kwargs['openId'], dateTimeAdded__gte = startTime,
  4993. dateTimeAdded__lte = endTime).order_by('-dateTimeAdded').first()
  4994. agentId = kwargs.get('agentId')
  4995. url = concat_user_center_entry_url(agentId=agentId,
  4996. redirect=concat_front_end_url(uri='/user/index.html#/user/feedbackList'))
  4997. if consumeRcd is None:
  4998. return [
  4999. {'msgType': 'text',
  5000. 'value': u'系统中没有查到您最近7天的使用记录/::-|,这种情况比较少见,建议您先创建一张<a href="%s">报告单</a>,然后拨打客服电话,让客服尽快处理' % url},
  5001. {'msgType': 'image', 'value': '1.png'},
  5002. {'msgType': 'image', 'value': '2.png'},
  5003. {'msgType': 'image', 'value': '3.png'},
  5004. {'msgType': 'image', 'value': '4.png'},
  5005. ]
  5006. rcdDesc = u'地址:%s,编号:%s,付款:%s,时间:%s' % (consumeRcd.address, consumeRcd.logicalCode, consumeRcd.coin,
  5007. consumeRcd.dateTimeAdded.strftime(Const.DATETIME_FMT))
  5008. if consumeRcd.isNormal:
  5009. return [
  5010. {'msgType': 'text', 'value': u'查到您最新的订单,信息如下:%s' % rcdDesc},
  5011. {'msgType': 'text',
  5012. 'value': u'钱确实用出去了/:--b,设备如果没有正常响应或者故障,您先拍照留下证据。然后您在服务&投诉中,<a href="%s">报告老板</a>,申请上分或者退币。然后在常见问题菜单中,拨打客服电话,让客服尽快处理/::)。如果没有及时响应,您也可以先行自己补款,然后记得向设备老板申请退币/:rose' % url}
  5013. ]
  5014. else:
  5015. return [
  5016. {'msgType': 'text', 'value': u'查到您最新的订单,信息如下:%s' % rcdDesc},
  5017. {'msgType': 'text', 'value': u'亲,刚查询到您确实有个消费订单,没有成功/:--b。系统已经自动给您退币了,您可以重新扫码尝试启动设备,或者换台别的设备试试吧/::)'}
  5018. ]
  5019. except Exception, e:
  5020. return [{'msgType': 'text',
  5021. 'value': u'我在系统中没有查到您最近的使用记录/::<,这种情况,建议您在服务&投诉中<a href="%s">报告老板</a>,申请上分或者退币。然后在常见问题中,拨打客服电话,让客服尽快处理/::P' % url}]
  5022. @staticmethod
  5023. def replyQueryServicedPhone(moniOpenId, appId, **kwargs):
  5024. agentId = kwargs.get('agentId')
  5025. endTime = datetime.datetime.now()
  5026. startTime = endTime - datetime.timedelta(days = 7)
  5027. consumeRcd = ConsumeRecord.objects.filter(openId = kwargs['openId'], dateTimeAdded__gte = startTime,
  5028. dateTimeAdded__lte = endTime).order_by('-dateTimeAdded').first()
  5029. if consumeRcd:
  5030. phone = AskStep.get_service_phone_from_rcd(consumeRcd)
  5031. return [{'msgType': 'text', 'value': u'/::P这个是客服电话吧:%s' % phone}]
  5032. else:
  5033. url = concat_user_center_entry_url(agentId=agentId,
  5034. redirect=concat_front_end_url(uri='/user/index.html#/user/help'))
  5035. return [{'msgType': 'text', 'value': u'这个给您,<a href="%s">客服电话</a>' % url}]
  5036. def reply(self, moniOpenId, appId, **kwargs):
  5037. if self.answerFunc:
  5038. return eval('AskStep.%s(moniOpenId,appId,**kwargs)' % self.answerFunc)
  5039. elif self.answerText:
  5040. return [{'msgType': 'text', 'value': self.answerText}]
  5041. elif self.answerPics:
  5042. return [{'msgType': 'image', 'value': pic} for pic in self.answerPics]
  5043. else:
  5044. return [{'msgType': 'text', 'value': ''}]
  5045. # 聊天模式匹配库。用于批量智能客服(不需要那么复杂的聊天,而且也贵;可以针对性进行数据库查询,更灵活)
  5046. class AskRobot(Searchable):
  5047. ask = StringField(verbose_name = 'ask', default = '') # 问题模式,是可以匹配的: 机器/洗衣机/设备/,不启动/没反应/不动/不转/没动静
  5048. steps = ListField(verbose_name = 'answer steps', default = []) # 支持
  5049. priority = IntField(verbose_name = 'priority', default = 1)
  5050. askList = []
  5051. @staticmethod
  5052. def load_in_mem_if_not_exist():
  5053. # if AskRobot.askList:#调试比较麻烦,直接查询所有吧,以后做了界面,再打开
  5054. # return
  5055. objs = AskRobot.objects.all().order_by('-priority')
  5056. for obj in objs:
  5057. AskRobot.askList.append(obj)
  5058. @staticmethod
  5059. def match_sentence(ask, sentence, score):
  5060. wordList = sentence.split('/')
  5061. for word in wordList:
  5062. if word in ask:
  5063. logger.info('it is match,sentence=%s,ask=%s' % (sentence, ask))
  5064. return score
  5065. return 0.0
  5066. # 根据图片文件名称找ID,文章标题找ID
  5067. @staticmethod
  5068. def find_material_id(proxy, mediaType, value):
  5069. if mediaType == 'image':
  5070. results = proxy.client.material.batchget(media_type = 'image', offset = 0,
  5071. count = 1000) # 简单粗暴,目前看素材不会超过20个
  5072. for item in results.get('item', []):
  5073. if item['name'] == value:
  5074. return item['media_id']
  5075. return None
  5076. return None
  5077. @staticmethod
  5078. def reply(event, moniOpenId, ask, appId, secret, **kwargs):
  5079. AskRobot.load_in_mem_if_not_exist()
  5080. # 产生一个线程,让线程后台处理这个客服消息
  5081. class replyer(threading.Thread):
  5082. def __init__(self, event, moniOpenId, ask, appId, secret, **kwargs):
  5083. super(replyer, self).__init__()
  5084. self._event = event
  5085. self._moniOpenId = moniOpenId
  5086. self._ask = ask
  5087. self._appId = appId
  5088. self._secret = secret
  5089. self._kwargs = kwargs
  5090. def run(self):
  5091. try:
  5092. replys = []
  5093. # 如果是关注事件,直接推送文章
  5094. if self._event == 'subscribe':
  5095. materialId = MoniApp.objects.get(appId = self._appId).welcomeArticleMaterialId
  5096. replys.append({'msgType': 'news', 'value': materialId})
  5097. moniUser = MoniUser.objects.filter(moniOpenId = self._moniOpenId,
  5098. moniAppId = self._appId).first()
  5099. if moniUser:
  5100. devLink = concat_user_login_entry_url(l=moniUser.subLogicalCode)
  5101. replys.append({'msgType': 'text',
  5102. 'value': u'/:rose亲,欢迎您关注自助设备服务平台!乐乐可以为您解决售后问题以及投诉事宜/:hug,——————— \ue231<a href="%s">戳我使用设备</a>\ue230' % devLink})
  5103. else:
  5104. # 找到所有问题的答案
  5105. for askConfig in AskRobot.askList:
  5106. sentenceList = askConfig.ask.split(',')
  5107. sentenceScore = 100.0 / len(sentenceList)
  5108. allScore = 0
  5109. for sentence in sentenceList:
  5110. allScore += AskRobot.match_sentence(self._ask, sentence, sentenceScore)
  5111. if allScore >= 90:
  5112. logger.info('find match answer key = %s' % ','.join(askConfig.steps))
  5113. replys = askConfig.answer(self._moniOpenId, self._appId, **self._kwargs)
  5114. break
  5115. logger.info('reply len %s' % len(replys))
  5116. # 把答案的文章、图片发给用户
  5117. app = WechatAuthApp(appid = self._appId, secret = self._secret)
  5118. proxy = WechatClientProxy(app)
  5119. for reply in replys:
  5120. if not reply:
  5121. continue
  5122. try:
  5123. if reply['msgType'] == 'text' and reply['value']:
  5124. proxy.client.message.send_text(moniOpenId, reply['value'])
  5125. elif reply['msgType'] == 'image' and reply['value']:
  5126. materialId = AskRobot.find_material_id(proxy, 'image', reply['value'])
  5127. proxy.client.message.send_image(moniOpenId, materialId)
  5128. elif reply['msgType'] == 'news' and reply['value']:
  5129. proxy.client.message.send_articles(moniOpenId, reply['value'])
  5130. time.sleep(1)
  5131. except Exception, e:
  5132. logger.exception('send msg to user failed = %s' % (e))
  5133. continue
  5134. except Exception as e:
  5135. logger.info('replyer error,e=%s' % e)
  5136. sender = replyer(event, moniOpenId, ask, appId, secret, **kwargs)
  5137. sender.start()
  5138. def answer(self, moniOpenId, appId, **kwargs):
  5139. # 首先把当前回话的历史信息拉出来
  5140. historyReplys = serviceCache.get(moniOpenId, '')
  5141. replyList = []
  5142. for stepKey in self.steps:
  5143. step = AskStep.objects.get(operKey = stepKey)
  5144. sameReplyCount = historyReplys.count(stepKey)
  5145. if sameReplyCount == 0:
  5146. reply = step.reply(moniOpenId, appId, **kwargs)
  5147. elif sameReplyCount == 1:
  5148. if 'xiangyin' not in stepKey:
  5149. reply = [{'msgType': 'text', 'value': u'建议您试试我给的建议哦/::P'}].extend(
  5150. step.reply(moniOpenId, appId, **kwargs))
  5151. else:
  5152. reply = [{'msgType': 'text', 'value': u'嗯,您的问题是什么呢?/::P'}]
  5153. elif sameReplyCount == 2:
  5154. step = AskStep.objects.get(operKey = 'kefudianhua')
  5155. reply = step.reply(moniOpenId, appId, **kwargs)
  5156. else:
  5157. reply = [{'msgType': 'text', 'value': ''}]
  5158. if reply:
  5159. historyReplys += '%s,' % stepKey
  5160. serviceCache.set(moniOpenId, historyReplys, 600)
  5161. replyList.extend(reply)
  5162. return replyList
  5163. class WxAuthTransfer(Searchable):
  5164. agentId = StringField(verbose_name = u'代理商ID', default = '')
  5165. oldAppId = StringField(verbose_name = u'老的appid', default = '')
  5166. oldOpenId = StringField(verbose_name = u'老的openid', default = '')
  5167. newAppId = StringField(verbose_name = u'新的appid', default = '')
  5168. newOpenId = StringField(verbose_name = u'新的openid', default = '')
  5169. # subAgentList = ListField(verbose_name = '子agent列表', default = '')
  5170. firstPhase = BooleanField(verbose_name = u'第1阶段任务完成', default = False)
  5171. secondPhase = BooleanField(verbose_name = u'第2阶段任务完成', default = False)
  5172. meta = {
  5173. "collection": "wx_auth_transfer",
  5174. "db_alias": "logdata"
  5175. }
  5176. def __repr__(self):
  5177. return '{}<agent={}, old=({},{}), new=({},{})>'.format(
  5178. self.__class__.__name__,
  5179. str(self.agentId),
  5180. self.oldAppId, self.oldOpenId,
  5181. self.newAppId, self.newOpenId)
  5182. class ICChargeOperating(Searchable):
  5183. """
  5184. deprecated 该功能废弃, 保证脚本能正常进行
  5185. """
  5186. cardId = StringField(verbose_name = 'cardId', default = '')
  5187. balance = MonetaryField(verbose_name = u"卡余额", default = RMB('0.00'))
  5188. needChargeMoney = MonetaryField(verbose_name = u"需要充值的钱", default = RMB('0.00'))
  5189. dateTimeAdded = DateTimeField(verbose_name = u'操作时间', default = datetime.datetime.now)
  5190. meta = {
  5191. "collection": "ic_charge_operating",
  5192. "db_alias": "logdata"
  5193. }
  5194. class BlackListUsers(Searchable):
  5195. class Status(IterConstant):
  5196. BLACK = "black"
  5197. WHITE = 'white'
  5198. openId = StringField(verbose_name = "被拉黑用户OpenId")
  5199. dealerId = StringField(verbose_name = "经销商ID")
  5200. agentId = StringField(verbose_name = "代理商ID")
  5201. reason = StringField(verbose_name = "被拉黑的理由", default = "")
  5202. operator = StringField(verbose_name = "操作人")
  5203. status = StringField(verbose_name = "拉黑状态", choices = Status.choices(), default = Status.BLACK)
  5204. dateTimeAdded = DateTimeField(verbose_name = "拉黑进去的时间", default = datetime.datetime.now)
  5205. meta = {
  5206. "db_alias": "default",
  5207. 'indexes': [
  5208. 'dealerId',
  5209. ],
  5210. }
  5211. @classmethod
  5212. def add_black_user(cls, openId, dealerId, operator, reason = ""):
  5213. """
  5214. 拉黑用户
  5215. :param openId:
  5216. :param dealerId:
  5217. :param operator:
  5218. :param reason:
  5219. :return:
  5220. """
  5221. try:
  5222. record = cls.objects.get(dealerId = dealerId, openId = openId)
  5223. except DoesNotExist:
  5224. dealer = Dealer.get_dealer(dealerId) or dict()
  5225. record = cls(
  5226. openId = openId,
  5227. dealerId = dealerId,
  5228. agentId = dealer.get("agentId", ""),
  5229. operator = operator,
  5230. reason = reason
  5231. ).save()
  5232. record.update(status = cls.Status.BLACK)
  5233. @classmethod
  5234. def freed_black_user(cls, openId, dealerId):
  5235. """
  5236. 解除用户的拉黑状态 这个地方不删除记录 防止之后要有需求对拉黑用户进行天数限制
  5237. :param openId:
  5238. :param dealerId:
  5239. :return:
  5240. """
  5241. try:
  5242. record = cls.objects.get(dealerId = dealerId, openId = openId)
  5243. except DoesNotExist:
  5244. return
  5245. record.update(status = cls.Status.WHITE)
  5246. @classmethod
  5247. def check_black(cls, openId, dealerId):
  5248. """
  5249. 检查该用户是否被拉黑
  5250. :param openId:
  5251. :param dealerId:
  5252. :return:
  5253. """
  5254. record = cls.objects(openId=openId, dealerId__in=[dealerId, '*']).first()
  5255. if not record:
  5256. return False
  5257. if record.status == cls.Status.WHITE:
  5258. return False
  5259. else:
  5260. return True
  5261. # 此表用于解决复制卡的问题。每张卡都有一个变化的随机码,每次扣费后会新生成一个随机码。每次扣钱后,会上报上次的随机码
  5262. # 如果随机码出现一次重复,说明出现一次复制卡。随机码是刷卡器生成的时间标签,恰好重复的可能性微乎其微。
  5263. class WeifuleCardStamp(Searchable):
  5264. cardNo = StringField(verbose_name = "卡号", default = "")
  5265. dealerId = StringField(verbose_name = "经销商ID", default = "")
  5266. dateTimeAdded = DateTimeField(verbose_name = '添加时间', default = datetime.datetime.now())
  5267. stamp = StringField(verbose_name = 'stamp', default = "")
  5268. meta = {
  5269. "collection": "weifule_card_stamp",
  5270. "db_alias": "default"
  5271. }
  5272. @staticmethod
  5273. def is_copy_card(cardNo, dealerId, stamp):
  5274. harfYear = datetime.datetime.now() - datetime.timedelta(days = 6 * 30)
  5275. copyCount = WeifuleCardStamp.objects.filter(cardNo = cardNo, dealerId = dealerId, dateTimeAdded__gte = harfYear,
  5276. stamp = stamp).count()
  5277. if copyCount:
  5278. return True
  5279. return False
  5280. class SwapCardRecord(Searchable):
  5281. oldCardNo = StringField(verbose_name = u"旧卡卡号")
  5282. newCardNo = StringField(verbose_name = u"新卡卡号")
  5283. agentId = StringField(verbose_name = u"补卡的代理商ID")
  5284. operator = StringField(verbose_name = u"操作人")
  5285. dateTimeAdded = DateTimeField(verbose_name = u"补卡时间", default = datetime.datetime.now)
  5286. meta = {'collection': 'SwapCardRecord', 'db_alias': 'logdata'}
  5287. @classmethod
  5288. def add_record(cls, oldCardNo, newCardNo, agentId, operator):
  5289. """
  5290. 添加一条补卡记录
  5291. :param oldCardNo: 旧卡卡号
  5292. :param newCardNo: 新卡卡号
  5293. :param agentId: 补卡代理商ID
  5294. :param operator: 操作者ID
  5295. :return:
  5296. """
  5297. record = cls(
  5298. newCardNo = newCardNo,
  5299. oldCardNo = oldCardNo,
  5300. agentId = agentId,
  5301. operator = operator
  5302. )
  5303. record.save()
  5304. return record
  5305. class DuibijiOrderMap(Searchable):
  5306. devOrderId = IntField(verbose_name = 'device order id', unique = True)
  5307. consumeRcdId = StringField(verbose_name = 'consume rcd id', unique = True)
  5308. rechargeRcdId = StringField(verbose_name = 'rechargeRcdId id', default = '')
  5309. status = StringField(verbose_name = 'status', default = '')
  5310. openId = StringField(verbose_name = 'openId', default = '')
  5311. dateTimeAdded = DateTimeField(verbose_name = 'dateTimeAdded', default = datetime.datetime.now())
  5312. class MonthlyPackageUseInfo(EmbeddedDocument):
  5313. orderNo = StringField(verbose_name = u"使用订单编号", required = True)
  5314. orderTime = DateTimeField(verbose_name = u"使用的时间", required = True)
  5315. def to_dict(self):
  5316. return {"orderNo": self.orderNo, "orderTime": self.orderTime}
  5317. class MonthlyPackage(Searchable):
  5318. # 用户相关信息
  5319. name = StringField(verbose_name = u"包月套餐的名称", default = "")
  5320. groupId = StringField(verbose_name = u"包月绑定的地址组", required = True)
  5321. openId = StringField(verbose_name = u"用户的唯一ID", required = True)
  5322. cardNo = StringField(verbose_name = u"刷卡能够使用的卡号", default = "")
  5323. bothCardAndMobile = BooleanIntField(verbose_name = u"是否是通用的(扫码和刷卡通用)", default = False)
  5324. # 包月相关信息
  5325. startDateTime = DateTimeField(verbose_name = u"有效期起始时间", required = True)
  5326. expireDateTime = DateTimeField(verbose_name = u"过期的时间", required = True)
  5327. maxCountOfDay = IntField(verbose_name = u"每日的最大的次数", default = 0, min_value = 0)
  5328. maxCount = IntField(verbose_name = u"可供使用的最大次数", default = 0, min_value = 0)
  5329. maxTimeOfCount = IntField(verbose_name = u"每次最大的充电时间 单位分钟", default = 0, min_value = 0)
  5330. maxElecOfCount = IntField(verbose_name = u"每次最大充电量 单位度", default = 0, min_value = 0)
  5331. # 基本信息
  5332. isDisable = BooleanIntField(verbose_name = u"是否已经失效", default = 0)
  5333. usedTotal = IntField(verbose_name = u"使用的总次数", default = 0)
  5334. usedDetail = MapField(verbose_name = u"用户使用情况",
  5335. field = EmbeddedDocumentListField(document_type = MonthlyPackageUseInfo))
  5336. dateTimeAdded = DateTimeField(verbose_name = u"购买的时间 一般与起始时间开始", default = datetime.datetime.now)
  5337. @property
  5338. def payVia(self):
  5339. return "monthlyPackage"
  5340. @staticmethod
  5341. def format_date(date): # type: (datetime.date) -> str
  5342. """
  5343. :param date:
  5344. :return:
  5345. """
  5346. return "T{}".format(date.strftime("%Y_%m_%d"))
  5347. @classmethod
  5348. def get_enable_one(cls, openId, groupId, cardNo = ""):
  5349. """
  5350. 找包月的 openId groupId 一定需要对应上 并且 允许通用 或者不允许通用但是卡号不为空
  5351. :return:
  5352. """
  5353. group = Group.get_group(groupId) # type: GroupDict
  5354. dealer = Dealer.objects.get(id = group.ownerId)
  5355. groupIds = Dealer.get_currency_group_ids(dealer, group)
  5356. groupIds.append(groupId)
  5357. objs = cls.objects.filter(
  5358. openId = openId, groupId__in = groupIds, isDisable = 0
  5359. ).filter(
  5360. Q(cardNo = cardNo) | Q(bothCardAndMobile = 1)
  5361. ).order_by('dateTimeAdded')
  5362. result = filter(
  5363. lambda obj: obj.useable,
  5364. map(lambda obj: obj.disable(), objs)
  5365. )
  5366. return result[0] if result else None
  5367. @classmethod
  5368. def get_user_all(cls, openId):
  5369. """
  5370. 获取用户所有的包月套餐规则
  5371. :param openId:
  5372. :return:
  5373. """
  5374. return cls.objects.filter(openId = openId, isDisable = 0)
  5375. @classmethod
  5376. def create_by_template(cls, template, rechargeOrder): # type:(MonthlyPackageTemp, RechargeRecord) -> MonthlyPackage
  5377. subIndex = rechargeOrder.attachParas.get("subTempIndex")
  5378. purchasedMonth = template.subTemplate[subIndex].numOfMonth
  5379. data = {
  5380. "name": template.subTemplate[subIndex].displayName,
  5381. "groupId": rechargeOrder.groupId,
  5382. "openId": rechargeOrder.openId,
  5383. "cardNo": rechargeOrder.attachParas.get("cardNo", ""),
  5384. "bothCardAndMobile": template.bothCardAndMobile,
  5385. "startDateTime": rechargeOrder.dateTimeAdded,
  5386. "expireDateTime": rechargeOrder.dateTimeAdded + relativedelta.relativedelta(months = purchasedMonth),
  5387. "maxCountOfDay": template.maxCountOfDay,
  5388. "maxCount": template.maxCountOfMonth * purchasedMonth,
  5389. "maxTimeOfCount": template.maxTimeOfCount,
  5390. "maxElecOfCount": template.maxElecOfCount,
  5391. }
  5392. obj = cls(**data)
  5393. return obj.save()
  5394. @property
  5395. def useable(self):
  5396. """
  5397. :return:
  5398. """
  5399. if self.isDisable:
  5400. return False
  5401. # 总次数
  5402. if len(self.usedDetail) >= self.maxCount:
  5403. return False
  5404. # 0 表示不限制
  5405. if self.maxCountOfDay == 0:
  5406. return True
  5407. else:
  5408. # 当日的次数超限
  5409. dateStr = self.format_date(datetime.date.today())
  5410. dateUseInfo = self.usedDetail.get(dateStr, list())
  5411. if len(dateUseInfo) >= self.maxCountOfDay:
  5412. return False
  5413. return True
  5414. def deduct(self, order):
  5415. # type:(ConsumeRecord)->None
  5416. """
  5417. 使用一次, 扣除一次额度, 根据订单进行扣除
  5418. :return:
  5419. """
  5420. orderTime = order.dateTimeAdded # type: datetime.datetime
  5421. orderInfo = MonthlyPackageUseInfo(orderNo = order.orderNo, orderTime = orderTime)
  5422. updateData = {
  5423. "inc__usedTotal": 1,
  5424. "add_to_set__usedDetail__{}".format(self.format_date(orderTime)): orderInfo
  5425. }
  5426. return self.update(**updateData)
  5427. def rollback(self, order):
  5428. """
  5429. 用于抵扣回退 目前存在与5分钟以内 需要全退的情况
  5430. """
  5431. orderTime = order.dateTimeAdded # type: datetime.datetime
  5432. orderInfo = MonthlyPackageUseInfo(orderNo=order.orderNo, orderTime=orderTime)
  5433. updateData = {
  5434. "dec__usedTotal": 1,
  5435. "pull__usedDetail__{}".format(self.format_date(orderTime)): orderInfo
  5436. }
  5437. return self.update(**updateData)
  5438. def disable(self):
  5439. """
  5440. 检查将 过期的包月套餐直接清理掉
  5441. :return:
  5442. """
  5443. if self.expireDateTime <= datetime.datetime.now():
  5444. self.update(isDisable = 1)
  5445. return self.reload()
  5446. return self
  5447. def is_suit_package(self, package):
  5448. """
  5449. 判断当前的包月套餐是否试用用户选择的套餐
  5450. 时间和电量的 检验是否足够抵扣 单次的直接抵扣一次 其余的暂不支持
  5451. :param package:
  5452. :return:
  5453. """
  5454. if package.get("unit") == u"分钟":
  5455. unit = "Time"
  5456. elif package.get("unit") == u"度":
  5457. unit = "Elec"
  5458. elif package.get("unit") == u"次":
  5459. return True
  5460. else:
  5461. return False
  5462. # 这个地方默认的规则时 如果没有适配到合适的套餐 就直接判定为不能够使用
  5463. count = getattr(self, "max{}OfCount".format(unit), 0)
  5464. # 添加一种情况(当count为0时候) 不做限制,只用次数
  5465. if count == 0:
  5466. return True
  5467. if count >= int(package.get("time", 0xFFFF)):
  5468. return True
  5469. return False
  5470. def to_dict(self):
  5471. dayUse = len(self.usedDetail.get(self.format_date(date = datetime.date.today()), list()))
  5472. return {
  5473. "id": self.id,
  5474. "name": self.name,
  5475. "expireDateTime": self.expireDateTime.strftime("%Y-%m-%d %H:%M:%S"),
  5476. "dayLeft": self.maxCountOfDay - dayUse,
  5477. "totalLeft": self.maxCount - self.usedTotal,
  5478. "maxElecOfCount": self.maxElecOfCount,
  5479. "maxTimeOfCount": self.maxTimeOfCount,
  5480. }
  5481. def get_used_detail(self):
  5482. usedDetail = list()
  5483. for _k, _v in self.usedDetail.items():
  5484. usedDetail.extend([_m.to_dict() for _m in _v])
  5485. usedDetail.sort(key = lambda x: x["orderTime"])
  5486. return usedDetail
  5487. @classmethod
  5488. def get_user_ticket(cls, openId, groupId, cardNo = ""):
  5489. group = Group.get_group(groupId) # type: GroupDict
  5490. dealer = Dealer.objects.get(id = group.ownerId)
  5491. groupIds = Dealer.get_currency_group_ids(dealer, group)
  5492. groupIds.append(groupId)
  5493. objs = cls.objects.filter(
  5494. openId = openId, groupId__in = groupIds, isDisable = 0
  5495. ).filter(
  5496. Q(cardNo = cardNo) | Q(bothCardAndMobile = 1)
  5497. ).order_by('dateTimeAdded')
  5498. return map(lambda obj: obj.disable(), objs)
  5499. @staticmethod
  5500. def get_can_use_one(all_tickets, package):
  5501. # type: (List[MonthlyPackage], dict) -> (MonthlyPackage)
  5502. for ticket in all_tickets:
  5503. if ticket.useable and ticket.is_suit_package(package):
  5504. return ticket
  5505. return None
  5506. class Redpack(Searchable):
  5507. # 红包信息
  5508. # TODO 建立索引 factoryCode(索引), openId(索引)
  5509. # 红包活动类型
  5510. class RedpackType():
  5511. LAXIN = 'laxin'
  5512. RUHUI = 'ruhui'
  5513. class Result(IterConstant):
  5514. FINISHED = 'finished'
  5515. PROCESSING = 'processing'
  5516. openId = StringField(verbose_name='openId', default='')
  5517. title = StringField(verbose_name='红包标题', default='')
  5518. factoryCode = StringField(verbose_name='TaskId')
  5519. redpackType = StringField(verbose_name='红包类型', default='')
  5520. # 红包自身流程
  5521. money = MonetaryField(verbose_name='红包金额', default=RMB('0.00'))
  5522. leastPayMoney = MonetaryField(verbose_name='最小使用金额', default=RMB('0.00'))
  5523. effectTime = DateTimeField(verbose_name='红包生效时间')
  5524. expiredTime = DateTimeField(verbose_name='红包过期时间')
  5525. usedStatus = BooleanField(verbose_name='是否使用', default=False)
  5526. usedTime = DateTimeField(verbose_name='使用时间')
  5527. consumeRecordId = StringField(verbose_name='关联消费订单')
  5528. package = DictField(verbose_name='选中的套餐')
  5529. # 来源信息
  5530. gateway = StringField(verbose_name='平台', default='')
  5531. logicalCode = StringField(verbose_name='设备号', default='')
  5532. devNo = StringField(verbose_name='IMEI', default='')
  5533. dateTimeAdded = DateTimeField(verbose_name='记录添加时间', default=datetime.datetime.now)
  5534. extra = DictField(verbose_name='其他', default={})
  5535. taskStatus = StringField(verbose_name="任务当前状态", default=Result.PROCESSING)
  5536. showType = StringField(verbose_name="红包展示方式")
  5537. meta = {
  5538. "db_alias": "default"
  5539. }
  5540. # 红包业务流程
  5541. @classmethod
  5542. def pre_deducted_coins(cls, redpackId, package):
  5543. # 预计抵扣的金额计算, 此函数目前只用于蓝牙流程
  5544. redpack = cls.objects.filter(id=redpackId, usedStatus=False).first()
  5545. if not redpack:
  5546. return 0.0
  5547. if redpack.money > RMB(package['price']):
  5548. return round(RMB(package['coins']), 2)
  5549. else:
  5550. ratio = Ratio(float(package['coins']) / float(package['price']))
  5551. return round(RMB(redpack.money) * ratio, 2)
  5552. @property
  5553. def to_deducted_coins(self):
  5554. if not self.package:
  5555. return 0.0
  5556. if self.money > RMB(self.package['price']):
  5557. return round(RMB(self.package['coins']), 2)
  5558. else:
  5559. ratio = Ratio(float(self.package['coins']) / float(self.package['price']))
  5560. return round(RMB(self.money) * ratio, 2)
  5561. @property
  5562. def to_deducted_money(self):
  5563. if not self.package:
  5564. return self.money
  5565. else:
  5566. if self.money > RMB(self.package['price']):
  5567. return round(RMB(self.package['price']), 2)
  5568. else:
  5569. return round(RMB(self.money), 2)
  5570. @staticmethod
  5571. def use_redpack(redPacketId, consumeRecordId, package):
  5572. # type: (str, str, dict) -> int
  5573. # 1 校验红包(过期, 状态)
  5574. nowTime = datetime.datetime.now()
  5575. return Redpack.objects.filter(id=redPacketId, usedStatus=False).update(usedStatus=True, usedTime=nowTime,
  5576. consumeRecordId=consumeRecordId, package=package)
  5577. @classmethod
  5578. def get_redpack_list(cls, filtetr, rmb=None):
  5579. """
  5580. rmb 当前使用金额, 如果没有的话
  5581. 例: 红包 满2 - 0.2 最小使用金额必须
  5582. """
  5583. nowTime = datetime.datetime.now()
  5584. if not rmb:
  5585. rmb = RMB(0).mongo_amount
  5586. exp_time = nowTime + datetime.timedelta(minutes=30) # 给30分钟时间给设备启动留用
  5587. redPackets = cls.objects.filter(
  5588. effectTime__lte=nowTime,
  5589. expiredTime__gte=exp_time,
  5590. leastPayMoney__lte=rmb,
  5591. usedStatus=False,
  5592. taskStatus=cls.Result.FINISHED,
  5593. **filtetr
  5594. ).order_by('expiredTime')
  5595. if not redPackets:
  5596. return []
  5597. return list(map(lambda x: x.to_dict(), redPackets))
  5598. def to_dict(self):
  5599. return {
  5600. 'openId': self.openId,
  5601. 'title': self.title,
  5602. 'factoryCode': self.factoryCode,
  5603. 'money': RMB(self.money).mongo_amount,
  5604. 'leastPayMoney': RMB(self.leastPayMoney).mongo_amount,
  5605. 'effectTime': self.effectTime,
  5606. 'expiredTime': self.expiredTime,
  5607. 'usedStatus': self.usedStatus,
  5608. 'logicalCode': self.logicalCode,
  5609. 'consumeRecordId': self.consumeRecordId,
  5610. 'redpackCoins': self.to_deducted_coins,
  5611. 'redpackMoney': self.to_deducted_money,
  5612. 'id': str(self.id),
  5613. 'taskStatus': self.taskStatus,
  5614. 'showType': self.showType,
  5615. }
  5616. def rollback_redpack(self, redPacketId):
  5617. redPacket = self.objects.filter(id=redPacketId).first()
  5618. if not Redpack:
  5619. return False
  5620. return redPacket.update(usedStatus=False)
  5621. @classmethod
  5622. def can_use(cls, dealer, devTypeCode):
  5623. # type: (Dealer, str)->bool
  5624. dealer_features = map(lambda x: x['key'], filter(lambda x: x['value'] == True, dealer.feature_list))
  5625. if 'disable_redpack' in dealer_features:
  5626. return False
  5627. else:
  5628. if devTypeCode and devTypeCode in Redpack.list_of_devices_that_support_redpack():
  5629. return True
  5630. return False
  5631. @classmethod
  5632. def auto_suit_with_money(cls, query, rmb):
  5633. """
  5634. 自动挑选合适的红包抵扣支付金额
  5635. # 有两种判断:
  5636. 1 红包金额大于支付金额, 优先选择与红包金额接近金额支付的红包使用
  5637. 2 红包金额小于支付金额, 优先选择大的红包金额
  5638. """
  5639. try:
  5640. redpacks = cls.get_redpack_list(query)
  5641. if redpacks:
  5642. # 第一种情况做筛选 红包金额大于支付金额, 优先选择与红包金额接近金额支付的红包使用
  5643. _enough_list = filter(lambda _: RMB(_.get('money')) >= rmb, redpacks)
  5644. if _enough_list:
  5645. return min(_enough_list, key=lambda _: RMB(_.get('money', 0)))
  5646. # 第二种情况 没有足够支付的金额 选取最大的红包进行抵扣
  5647. return max(redpacks, key=lambda _: RMB(_.get('money', 0)))
  5648. except:
  5649. import traceback
  5650. logger.error(traceback.format_exc())
  5651. return
  5652. @classmethod
  5653. def auto_suit_with_coins(cls, query, userBalance, package):
  5654. """
  5655. 自动挑选合适的红包抵扣金币
  5656. # 有三种判断:
  5657. 1 红包足以 抵扣 挑选最小的红包
  5658. 2 红包 + 金币 抵扣 挑选最小的红包
  5659. 2 返回 都不足以抵扣 返回空
  5660. """
  5661. try:
  5662. package_money = RMB(package.get('price', 0))
  5663. if package_money == RMB(0):
  5664. return
  5665. package_coins = RMB(package.get('coins', 0))
  5666. ratio = Ratio(float(package.get('coins', 0.0)) / float(package.get('price', 0.0)))
  5667. redpacks = cls.get_redpack_list(query)
  5668. if redpacks:
  5669. # 第一种情况 红包足以 抵扣 挑选最小的红包
  5670. _enough_list = filter(lambda _: RMB(_.get('money')) >= package_money, redpacks)
  5671. if _enough_list:
  5672. return min(_enough_list, key=lambda _: RMB(_.get('money', 0)))
  5673. # 第二种情况 红包 + 金币 抵扣 挑选最小的红包
  5674. _mix_enough_list = filter(lambda _: RMB(_.get('money', 0)) * ratio + RMB(userBalance) >= package_coins, redpacks)
  5675. if _mix_enough_list:
  5676. return min(_mix_enough_list, key=lambda _: RMB(_.get('money', 0)))
  5677. except:
  5678. import traceback
  5679. logger.error(traceback.format_exc())
  5680. return
  5681. @classmethod
  5682. def get_one(self, redpackId):
  5683. redpack = Redpack.objects.filter(id=redpackId).first() # type: Redpack
  5684. if not redpack:
  5685. return {}
  5686. else:
  5687. return redpack.to_dict()
  5688. @classmethod
  5689. def show_redpack(cls, openId):
  5690. nowTime = datetime.datetime.now()
  5691. exp_time = nowTime - datetime.timedelta(days=3)
  5692. # 筛选出过期时间超过三天或者 使用后时间超过三天的红包
  5693. redPackets = cls.objects.filter(Q(expiredTime__gte=exp_time) | Q(usedTime__gte=exp_time),
  5694. openId=openId, taskStatus=cls.Result.FINISHED).order_by('-expiredTime')
  5695. if not redPackets:
  5696. return []
  5697. return list(map(lambda x: x.to_dict(), redPackets))
  5698. @staticmethod
  5699. def list_of_devices_that_support_redpack():
  5700. from apps.web.core.models import SystemSettings
  5701. support_redpack_list = SystemSettings.get_support_redpack_list()
  5702. return support_redpack_list
  5703. @staticmethod
  5704. def add_device_to_support_redpack_list(devTypeCode):
  5705. from apps.web.core.models import SystemSettings
  5706. support_redpack_list = SystemSettings.get_support_redpack_list()
  5707. support_redpack_list.append(devTypeCode)
  5708. support_redpack_list = list(set(support_redpack_list))
  5709. SystemSettings.set_support_redpack_list(support_redpack_list)
  5710. logger.info('Current support redpck list:{}'.format(support_redpack_list))
  5711. @staticmethod
  5712. def remove_device_to_support_redpack_list(devTypeCode):
  5713. from apps.web.core.models import SystemSettings
  5714. support_redpack_list = SystemSettings.get_support_redpack_list()
  5715. if devTypeCode in support_redpack_list:
  5716. support_redpack_list.remove(devTypeCode)
  5717. support_redpack_list = list(set(support_redpack_list))
  5718. SystemSettings.set_support_redpack_list(support_redpack_list)
  5719. logger.info('Current support redpck list:{}'.format(support_redpack_list))
  5720. @staticmethod
  5721. def renew_support_repack_list(devTypeCodeList):
  5722. from apps.web.core.models import SystemSettings
  5723. SystemSettings.set_support_redpack_list(devTypeCodeList)
  5724. # 拉新创建红包*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-
  5725. @staticmethod
  5726. def create_redpack_by_laxin(factoryCode, openId, money, leastPayMoney, effectTime, expiredTime, gateway,
  5727. logicalCode, devNo, extra):
  5728. if Redpack.objects.filter(factoryCode=factoryCode, openId=openId).first():
  5729. return
  5730. else:
  5731. red_packet = {
  5732. 'factoryCode': factoryCode,
  5733. 'title': '支付宝每日任务红包',
  5734. 'openId': openId,
  5735. 'money': money,
  5736. 'leastPayMoney': leastPayMoney,
  5737. 'effectTime': effectTime,
  5738. 'expiredTime': expiredTime,
  5739. 'gateway': gateway,
  5740. 'logicalCode': logicalCode,
  5741. 'devNo': devNo,
  5742. 'extra': extra,
  5743. 'taskStatus': Redpack.Result.FINISHED,
  5744. 'redpackType': Redpack.RedpackType.LAXIN
  5745. }
  5746. try:
  5747. redpack = Redpack.objects.create(**red_packet)
  5748. except:
  5749. return None
  5750. return redpack
  5751. # 入会创建红包*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-
  5752. @staticmethod
  5753. def create_redpack_by_ruhui(taskId, openId, urlId, money, leastPayMoney, gateway, logicalCode, devNo, extra, showType=''):
  5754. redpack = Redpack.objects.filter(factoryCode=taskId, openId=openId, redpackType=Redpack.RedpackType.RUHUI,
  5755. taskStatus__ne=Redpack.Result.FINISHED).first()
  5756. if redpack:
  5757. redpack.update(**{
  5758. 'logicalCode': logicalCode,
  5759. 'devNo': devNo,
  5760. 'extra': extra,
  5761. 'taskStatus': Redpack.Result.PROCESSING,
  5762. 'expiredTime': datetime.datetime.now() + datetime.timedelta(days=7),
  5763. 'urlId': urlId,
  5764. 'dateTimeAdded': datetime.datetime.now(),
  5765. 'showType':showType,
  5766. })
  5767. else:
  5768. red_packet = {
  5769. 'factoryCode': taskId,
  5770. 'title': '支付宝每日任务红包.',
  5771. 'openId': openId,
  5772. 'money': money,
  5773. 'leastPayMoney': leastPayMoney,
  5774. 'gateway': gateway,
  5775. 'logicalCode': logicalCode,
  5776. 'devNo': devNo,
  5777. 'extra': extra,
  5778. 'taskStatus': Redpack.Result.PROCESSING,
  5779. 'expiredTime': datetime.datetime.now() + datetime.timedelta(days=7),
  5780. 'redpackType': Redpack.RedpackType.RUHUI,
  5781. 'urlId': urlId,
  5782. 'showType': showType,
  5783. }
  5784. try:
  5785. redpack = Redpack.objects.create(**red_packet)
  5786. except:
  5787. return
  5788. return redpack
  5789. @staticmethod
  5790. def take_effect(openId, urlId, **kw):
  5791. try:
  5792. obj = Redpack.objects.filter(openId=openId, urlId=urlId, redpackType=Redpack.RedpackType.RUHUI,
  5793. taskStatus__ne=Redpack.Result.FINISHED).first()
  5794. if not obj:
  5795. return
  5796. extra = obj.extra
  5797. extra.update(kw)
  5798. obj.update(
  5799. extra=extra,
  5800. taskStatus=Redpack.Result.FINISHED,
  5801. effectTime=datetime.datetime.now(),
  5802. expiredTime=datetime.datetime.now() + datetime.timedelta(days=15)
  5803. )
  5804. return obj.reload()
  5805. except:
  5806. import traceback
  5807. logger.error(traceback.format_exc())
  5808. class OneCardGateLog(Searchable):
  5809. devNo = StringField(verbose_name=u"设备编号")
  5810. logicalCode = StringField(verbose_name=u"逻辑编号")
  5811. openId = StringField(verbose_name=u"用户信息")
  5812. control = IntField(verbose_name=u"进出标记")
  5813. result = BooleanField(verbose_name=u"操纵结果")
  5814. dateTimeAdded = DateTimeField(verbose_name=u"添加时间")
  5815. meta = {
  5816. 'collection': 'OneCardGateLog',
  5817. 'db_alias': 'logdata'
  5818. }
  5819. # ------------------------------------ 以下为新增 ------------------------
  5820. class OrderPackage(dict):
  5821. def belong_category(self, category): # type: (str) -> bool
  5822. return self.category == category
  5823. @property
  5824. def policyType(self): # type: () -> Optional[str, None]
  5825. return self.get("policyType") or self.get("category")
  5826. @property
  5827. def category(self): # type: () -> str
  5828. """
  5829. 套餐的种类 实际上也是相应的启动方式
  5830. """
  5831. if self.policyType:
  5832. return self.policyType
  5833. unit = self.get("unit")
  5834. if unit in [u"度"]:
  5835. return PackageCategory.ELEC
  5836. if unit in [u"元"]:
  5837. return PackageCategory.COIN
  5838. if unit in [u"小时", u"分钟"]:
  5839. return PackageCategory.TIME
  5840. return "unknown"
  5841. @property
  5842. def isPostpaid(self):
  5843. return self.get("isPostpaid", False)
  5844. @property
  5845. def autoRefund(self):
  5846. return self.get("autoRefund", False)
  5847. @property
  5848. def autoStop(self):
  5849. return self.get("autoStop", False)
  5850. @property
  5851. def minFee(self):
  5852. return RMB(self.get("minFee", 0))
  5853. @property
  5854. def minAfterStartCoins(self):
  5855. return RMB(self.get("minAfterStartCoins", 0))
  5856. @property
  5857. def isFree(self):
  5858. return self.get("isFree", False)
  5859. @property
  5860. def price(self):
  5861. """
  5862. 套餐的价格 后付费 即先用后付的情况 显示价格为0
  5863. """
  5864. if self.isFree or self.isPostpaid:
  5865. return RMB(0)
  5866. return RMB(self.get("price", 0))
  5867. @property
  5868. def time(self): # type:()->int
  5869. """
  5870. 套餐的时间单位 最终为 分钟
  5871. 注意区分是否已经初始化过了
  5872. """
  5873. value = self.get("m_time")
  5874. if value:
  5875. return value
  5876. # 初始化过程
  5877. value = self.get('time', 0) if self.belong_category(PackageCategory.TIME) else 0
  5878. if self.get("unit") == u"小时":
  5879. return int(float(value) * 60)
  5880. if self.get("unit") == u"分钟":
  5881. return int(value)
  5882. return 0
  5883. @property
  5884. def elec(self):
  5885. """
  5886. 套餐的电量单位 最终为 度
  5887. """
  5888. value = self.get("m_elec")
  5889. if value:
  5890. return value
  5891. return float(self.get('time', 0) if self.belong_category(PackageCategory.ELEC) else 0)
  5892. @property
  5893. def coin(self):
  5894. """
  5895. 注意 这个coin不是价格的意思 指的是直接是设备运行的硬币数 例如投币 或者某些直接以金额启动设备的套餐
  5896. """
  5897. value = self.get("m_coin")
  5898. if value:
  5899. return value
  5900. return int(self.get('time', 0) if self.belong_category(PackageCategory.COIN) else 0)
  5901. @property
  5902. def name(self):
  5903. return self.get("name", "")
  5904. @property
  5905. def desc(self):
  5906. return self.get("desc", "")
  5907. @property
  5908. def rules(self):
  5909. """
  5910. 即计费规则
  5911. """
  5912. return self.get("rules") or None
  5913. @property
  5914. def refundProtectTime(self):
  5915. return self.get("refundProtectTime", 0)
  5916. @property
  5917. def dumpDict(self):
  5918. """
  5919. 需要被订单固化的数据
  5920. """
  5921. return {
  5922. "category": self.category,
  5923. "name": self.name,
  5924. "price": self.price.mongo_amount,
  5925. "m_time": self.time,
  5926. "m_elec": self.elec,
  5927. "m_coin": self.coin,
  5928. "minAfterStartCoins": self.minAfterStartCoins.mongo_amount,
  5929. "minFee": self.minFee.mongo_amount,
  5930. "autoRefund": self.autoRefund,
  5931. "autoStop": self.autoStop,
  5932. "rules": self.rules,
  5933. "isFree": self.isFree,
  5934. "isPostpaid": self.isPostpaid,
  5935. "desc": self.desc,
  5936. "refundProtectTime": self.refundProtectTime
  5937. }
  5938. @property
  5939. def showDict(self):
  5940. return {
  5941. "category": self.category,
  5942. "name": self.name,
  5943. "time": self.time,
  5944. "elec": self.elec,
  5945. "coin": self.coin,
  5946. "minAfterStartCoins": self.minAfterStartCoins.mongo_amount,
  5947. "minFee": self.minFee.mongo_amount,
  5948. "autoRefund": self.autoRefund,
  5949. "autoStop": self.autoStop,
  5950. "rules": self.rules,
  5951. "isFree": self.isFree,
  5952. "isPostpaid": self.isPostpaid,
  5953. "desc": self.desc,
  5954. "refundProtectTime": self.refundProtectTime
  5955. }
  5956. class PaymentInfo(dict):
  5957. @property
  5958. def deduct_list(self):
  5959. return self.get("deduct_list") or list()
  5960. @property
  5961. def time(self):
  5962. return self.get("time")
  5963. @time.setter
  5964. def time(self, value):
  5965. self.update({"time": value})
  5966. @property
  5967. def isPaid(self):
  5968. return self.time is not None
  5969. @property
  5970. def via(self):
  5971. return self.get("via")
  5972. @property
  5973. def totalAmount(self):
  5974. s = RMB(0)
  5975. for _item in self.deduct_list:
  5976. s += RMB(_item.get("chargeBalance"))
  5977. s += RMB(_item.get("bestowBalance"))
  5978. return s
  5979. @property
  5980. def actualAmount(self):
  5981. """
  5982. 排除赠送余额之后的支付金额
  5983. """
  5984. s = RMB(0)
  5985. for _item in self.deduct_list:
  5986. s += RMB(_item.get("chargeBalance"))
  5987. return s
  5988. @property
  5989. def bestowAmount(self):
  5990. s = RMB(0)
  5991. for _item in self.deduct_list:
  5992. s += RMB(_item.get("bestowBalance"))
  5993. return s
  5994. class ServiceInfo(dict):
  5995. """
  5996. 和设备运行的一切相关信息
  5997. """
  5998. @property
  5999. def deviceStartTime(self):
  6000. return self.get(ConsumeOrderServiceItem.START_TIME)
  6001. @deviceStartTime.setter
  6002. def deviceStartTime(self, value):
  6003. if isinstance(value, datetime.datetime):
  6004. value = value.strftime("%Y-%m-%d %H:%M:%S")
  6005. self.update({ConsumeOrderServiceItem.START_TIME: value})
  6006. @property
  6007. def elec(self):
  6008. return self.get(ConsumeOrderServiceItem.ELEC)
  6009. @elec.setter
  6010. def elec(self, value):
  6011. self.update({ConsumeOrderServiceItem.ELEC: float(value)})
  6012. @property
  6013. def duration(self):
  6014. return self.get(ConsumeOrderServiceItem.DURATION)
  6015. @duration.setter
  6016. def duration(self, value):
  6017. self.update({ConsumeOrderServiceItem.DURATION: int(value)})
  6018. @property
  6019. def deviceEndTime(self):
  6020. return self.get(ConsumeOrderServiceItem.END_TIME)
  6021. @deviceEndTime.setter
  6022. def deviceEndTime(self, value):
  6023. if isinstance(value, datetime.datetime):
  6024. value = value.strftime("%Y-%m-%d %H:%M:%S")
  6025. self.update({ConsumeOrderServiceItem.END_TIME: value})
  6026. @property
  6027. def maxPower(self):
  6028. return self.get(ConsumeOrderServiceItem.MAX_POWER)
  6029. @maxPower.setter
  6030. def maxPower(self, value):
  6031. self.update({ConsumeOrderServiceItem.MAX_POWER: value})
  6032. @property
  6033. def reason(self):
  6034. return self.get(ConsumeOrderServiceItem.REASON)
  6035. @reason.setter
  6036. def reason(self, value):
  6037. self.update({ConsumeOrderServiceItem.REASON: value})
  6038. @property
  6039. def spendMoney(self):
  6040. return RMB(self.get(ConsumeOrderServiceItem.SPEND, 0))
  6041. @spendMoney.setter
  6042. def spendMoney(self, value):
  6043. self.update({ConsumeOrderServiceItem.SPEND: RMB(value)})
  6044. class ConsumeRecord(Searchable):
  6045. """
  6046. 用户的消费订单记录
  6047. """
  6048. class Status(IterConstant):
  6049. CREATED = 'created' # 订单创建初始状态 刚刚下单
  6050. WAIT_CONF = 'waitConf' # 用户下完单之后【等待】用户进一步的动作确认
  6051. FINISHED = 'finished' # 订单的结束状态 表示相应信息已经被记录
  6052. WAITING = 'waiting' # 订单的执行等待状态 设备尚未执行该订单
  6053. RUNNING = 'running' # 订单执行运行状态 设备正在执行该订单
  6054. END = "end" # 订单运行结束状态 设备已经执行订单完毕 订单完结
  6055. TIMEOUT = 'timeout' # 订单启动超时状态 设备启动超时 有可能已经启动了
  6056. FAILURE = 'failure' # 订单执行失败状态 设备明确启动失败 订单已经完结
  6057. UNKNOWN = 'unknown' # 订单的未知状态 未知的订单状态
  6058. WAIT_PAY = 'waitPay' # 订单支付的中间状态
  6059. orderNo = StringField(verbose_name=u"订单号", required=True, unique=True)
  6060. sequenceNo = StringField(verbose_name=u"流水号", unique=True)
  6061. logicalCode = StringField(verbose_name=u"设备编号", required=True)
  6062. groupId = StringField(verbose_name=u"设备地址编号", default=None)
  6063. price = MonetaryField(verbose_name=u"订单价格", default=RMB('0'))
  6064. startType = IntField(verbose_name=u"启动方式", choices=StartDeviceType.choices())
  6065. status = StringField(verbose_name=u"状态", default=Status.CREATED)
  6066. ownerId = StringField(verbose_name=u"经销商", required=True)
  6067. remarks = StringField(verbose_name=u"备注(系统、用户)", default='')
  6068. description = StringField(verbose_name=u"订单描述(错误信息)")
  6069. association = DictField(verbose_name=u'关联单', default={})
  6070. serviceInfo = DictField(verbose_name=u"订单的服务信息", default={})
  6071. isFree = BooleanField(verbose_name=u"是否订单免费", default=False)
  6072. dateTimeAdded = DateTimeField(verbose_name=u"创建时间", default=datetime.datetime.now)
  6073. finishedTime = DateTimeField(verbose_name=u"结束时间")
  6074. cardId = StringField(verbose_name=u"卡ID")
  6075. openId = StringField(verbose_name=u"用户", required=True)
  6076. nickname = StringField(verbose_name=u'用户昵称', default='')
  6077. # 以下信息属于订单快照 防止引用被修改
  6078. devNo = StringField(verbose_name=u"设备ID", required=True)
  6079. port = IntField(verbose_name=u"启动端口")
  6080. devTypeName = StringField(verbose_name=u"设备类型名称", default=None)
  6081. devTypeCode = StringField(verbose_name=u"设备类型编码", default=None)
  6082. address = StringField(verbose_name=u"设备地址", default=None)
  6083. groupNumber = StringField(verbose_name=u"设备", default=None)
  6084. groupName = StringField(verbose_name=u"交易场地", default=None)
  6085. # 退款状态应该是用户下单的时候即被锁定 和设备、支付、套餐有关系
  6086. startPackage = DictField(verbose_name=u'启动套餐', required=True)
  6087. paymentInfo = DictField(verbose_name=u'支付的相关信息', default={})
  6088. refundInfo = DictField(verbose_name=u"退还的相关信息", default={})
  6089. dailyStats = LazyReferenceField(verbose_name="统计关联单", document_type='DealerGroupStats')
  6090. feedbackId = ObjectIdField(verbose_name=u'用户反馈ID')
  6091. search_fields = ('openId', 'devNo', 'orderNo', 'remarks', 'logicalCode')
  6092. _shard_key = ('ownerId', 'dateTimeAdded')
  6093. _origin_meta = {
  6094. "collection": "ConsumeRecord",
  6095. "db_alias": "default"
  6096. }
  6097. meta = _origin_meta
  6098. def __str__(self):
  6099. return '{}<id={} orderNo={}>'.format(self.__class__.__name__, str(self.id), self.orderNo)
  6100. @cached_property
  6101. def owner(self):
  6102. # 防止id报错
  6103. if not self.ownerId:
  6104. return None
  6105. from apps.web.dealer.models import Dealer
  6106. dealer = Dealer.objects(id=self.ownerId).first()
  6107. return dealer
  6108. @cached_property
  6109. def user(self): # type:() -> MyUser
  6110. return MyUser.objects.filter(openId=self.openId, groupId=self.groupId).first()
  6111. @cached_property
  6112. def card(self):
  6113. return Card.objects.filter(id=self.cardId).first()
  6114. @property
  6115. def coin(self):
  6116. return self.price
  6117. @property
  6118. def package(self): # type: () -> OrderPackage
  6119. return OrderPackage(self.startPackage)
  6120. @property
  6121. def device(self): # type:()-> DeviceDict
  6122. return Device.get_dev(self.devNo)
  6123. @property
  6124. def group(self): # type:() -> GroupDict
  6125. return Group.get_group(self.groupId)
  6126. @property
  6127. def startLockKey(self):
  6128. return "{}-{}-start-device-lock".format(self.openId, self.orderNo)
  6129. @property
  6130. def isPayTimeOut(self):
  6131. """
  6132. 订单到支付下单的一个过程
  6133. """
  6134. now = datetime.datetime.now()
  6135. return (now - self.dateTimeAdded).total_seconds() > CONSUME_ORDER_PAY_TIMEOUT
  6136. @property
  6137. def isPostPaid(self):
  6138. """
  6139. True 后付费订单 使用完给钱
  6140. False 预付费订单 先给钱再使用然后再决定是否退款
  6141. """
  6142. return self.package.isPostpaid
  6143. @property
  6144. def isStartNetPay(self):
  6145. return self.startType == StartDeviceType.ON_LIEN
  6146. @property
  6147. def isStartCardPay(self):
  6148. return self.startType == StartDeviceType.CARD
  6149. @property
  6150. def payer(self):
  6151. if self.startType == StartDeviceType.ON_LIEN:
  6152. return self.user
  6153. else:
  6154. return self.card
  6155. @property
  6156. def payment(self): # type: () -> PaymentInfo
  6157. return PaymentInfo(self.paymentInfo)
  6158. @property
  6159. def refund(self): # type: () -> PaymentInfo
  6160. return PaymentInfo(self.refundInfo)
  6161. @property
  6162. def service(self): # type:() -> ServiceInfo
  6163. return ServiceInfo(self.serviceInfo)
  6164. @property
  6165. def device_start_time(self):
  6166. return self.service.deviceStartTime
  6167. @property
  6168. def device_end_time(self):
  6169. return self.service.deviceEndTime
  6170. @property
  6171. def isPaid(self):
  6172. if self.isFree:
  6173. return True
  6174. return self.payment.isPaid
  6175. @property
  6176. def actualAmount(self):
  6177. """
  6178. 用户订单实际消费的金额
  6179. """
  6180. return self.payment.actualAmount - self.refund.actualAmount
  6181. @property
  6182. def detail_link(self):
  6183. if self.device.deviceAdapter.support_count_down(self.openId, self.port):
  6184. return concat_count_down_page_url(devNo=self.devNo, port=self.port)
  6185. else:
  6186. return concat_front_end_url(uri='/user/index.html#/user/consumeDetail?id={}'.format(str(self.id)))
  6187. @property
  6188. def receiptDesc(self):
  6189. rv = {
  6190. 'orderNo': self.orderNo,
  6191. 'createdTime': self.dateTimeAdded,
  6192. 'payment': u'{} 元'.format(self.price)
  6193. }
  6194. if self.isFree:
  6195. rv.update({"payment": u"本次免费"})
  6196. if self.isPostPaid:
  6197. rv.update({"payment": u"先用后付"})
  6198. return rv
  6199. @property
  6200. def startKey(self):
  6201. return self.orderNo
  6202. @property
  6203. def actualAmount(self):
  6204. return self.payment.actualAmount - self.refund.actualAmount
  6205. @property
  6206. def bestowAmount(self):
  6207. return self.payment.bestowAmount - self.refund.bestowAmount
  6208. @property
  6209. def isNormal(self):
  6210. """
  6211. 是否为正常订单
  6212. """
  6213. if self.status in [self.Status.FAILURE, self.Status.UNKNOWN, self.Status.TIMEOUT]:
  6214. return False
  6215. if self.status in [self.Status.WAIT_CONF, self.Status.CREATED]:
  6216. return False
  6217. return True
  6218. @property
  6219. def device_identity_info(self):
  6220. return {
  6221. 'logicalCode': self.logicalCode,
  6222. 'devTypeCode': self.devTypeCode,
  6223. 'devTypeName': self.devTypeName,
  6224. 'groupName': self.groupName,
  6225. 'groupNumber': self.groupNumber,
  6226. 'address': self.address,
  6227. 'groupId': self.groupId
  6228. }
  6229. @classmethod
  6230. def make_no(cls, *args, **kwargs):
  6231. """
  6232. 有可能会重复 需要保证唯一性
  6233. """
  6234. timestamp = generate_timestamp_ex()
  6235. random_int = random.randint(1, 1000)
  6236. return "%d%03d" % (timestamp, random_int)
  6237. @classmethod
  6238. def new_one(cls, orderNo, user, device, context): # type:(str, MyUser, DeviceDict, StartParamContext) -> ConsumeRecord
  6239. order = cls(
  6240. orderNo=orderNo,
  6241. sequenceNo=context.sequence,
  6242. logicalCode=device.logicalCode,
  6243. groupId=device.groupId,
  6244. price=context.package.price,
  6245. startType=context.startType,
  6246. ownerId=device.ownerId,
  6247. openId=user.openId,
  6248. cardId=context.cardId,
  6249. nickname=user.nickname,
  6250. devNo=device.devNo,
  6251. port=context.port,
  6252. devTypeName=device.devTypeName,
  6253. devTypeCode=device.devTypeCode,
  6254. address=device.group.address,
  6255. groupNumber=device.groupNumber,
  6256. groupName=device.group.groupName,
  6257. startPackage=context.package.dumpDict,
  6258. isFree=context.package.isFree,
  6259. autoRefund=context.package.autoRefund
  6260. )
  6261. return order.save()
  6262. def update_payment(self, payment): # type:(dict) -> bool
  6263. """支付信息添加"""
  6264. if self.isPaid:
  6265. return False
  6266. self.paymentInfo = payment
  6267. return self.save()
  6268. def frozen_payer_balance(self):
  6269. """
  6270. 执行订单的支付过程 实际上是将用户的金额冻结
  6271. """
  6272. if not self.payment:
  6273. return
  6274. if self.isPaid:
  6275. return
  6276. payer, payment = self.payer, self.payment
  6277. result = payer.__class__.freeze_balance(str(self.id), self.payment)
  6278. if not result:
  6279. return
  6280. # 更新一次付款信息
  6281. payment.time = datetime.datetime.now().strftime("%Y-%m-%d %H:%M:%S")
  6282. self.update_payment(payment)
  6283. payer.account_consume(self)
  6284. def clear_payer_frozen(self, refundMoney=RMB(0)):
  6285. from apps.web.user.utils2 import generate_refund
  6286. refundInfo = PaymentInfo(generate_refund(self, refundMoney))
  6287. result = self.payer.__class__.clear_frozen_balance(
  6288. str(self.id),
  6289. self.payment,
  6290. refundInfo
  6291. )
  6292. if not result:
  6293. return
  6294. if refundMoney == RMB(0):
  6295. return
  6296. refundInfo.time = datetime.datetime.now().strftime("%Y-%m-%d %H:%M:%S")
  6297. self.refundInfo = refundInfo
  6298. self.save()
  6299. self.payer.account_refund(self)
  6300. def link_state(self, state):
  6301. self.dailyStats = state
  6302. self.finishedTime = datetime.datetime.now().strftime("%Y-%m-%d %H:%M:%S")
  6303. self.save()
  6304. def to_user_detail(self):
  6305. data = {
  6306. "id": str(self.id),
  6307. "orderNo": self.orderNo,
  6308. "ownerId": self.ownerId,
  6309. "money": self.price,
  6310. "createdTime": self.dateTimeAdded,
  6311. "completionTime": self.finishedTime,
  6312. "startResult": True if self.device_start_time else False,
  6313. "errorDesc": self.description,
  6314. "deviceStatTime": self.device_start_time,
  6315. "deviceFinishedTime": self.device_end_time,
  6316. "isRefund": True if self.refund.time else False,
  6317. "orderStatus": self.status
  6318. }
  6319. if self.refund.time:
  6320. data.update({
  6321. "refundedMoney": self.refund.totalAmount
  6322. })
  6323. data.update(self.device_identity_info)
  6324. return data
  6325. def to_detail(self):
  6326. data = {
  6327. "id": str(self.id),
  6328. 'orderNo': self.orderNo,
  6329. 'createdTime': self.dateTimeAdded,
  6330. 'completionTime': self.finishedTime,
  6331. 'deviceStatTime': self.device_start_time,
  6332. 'deviceFinishedTime': self.device_end_time,
  6333. 'amount': self.price,
  6334. 'openId': self.openId,
  6335. 'groupId': self.groupId,
  6336. 'userNickname': u'用户' if not self.nickname else self.nickname,
  6337. 'ownerId': self.ownerId,
  6338. 'devNo': self.devNo,
  6339. 'logicalCode': self.logicalCode,
  6340. 'groupName': self.groupName,
  6341. 'address': self.address,
  6342. 'groupNumber': self.groupNumber,
  6343. 'devType': self.devTypeName, # 兼容
  6344. 'devTypeName': self.devTypeName,
  6345. 'remarks': self.remarks,
  6346. 'startResult': 'success' if self.isNormal else 'failed',
  6347. 'errorDesc': self.description,
  6348. 'servicedInfo': [
  6349. u'%s: %s'.encode('utf-8') % (GLOSSARY_TRANSLATION.get(k, k), v) for k, v in self.service.iteritems()
  6350. ],
  6351. # 以下字段和以前做兼容
  6352. 'finishedTime': self.device_end_time,
  6353. 'port': self.port
  6354. }
  6355. return data
  6356. class BalanceLog(Searchable):
  6357. meta = {
  6358. 'abstract': True,
  6359. }
  6360. bAmount = MonetaryField(verbose_name=u"变动前充值余额", required=True)
  6361. bBestowAmount = MonetaryField(verbose_name=u"变动前赠送余额", required=True)
  6362. aAmount = MonetaryField(verbose_name=u"变动后充值余额", required=True)
  6363. aBestowAmount = MonetaryField(verbose_name=u"变动后赠送金额", required=True)
  6364. # 充值订单 1 退费订单 2 消费订单 3
  6365. category = IntField(verbose_name=u"变动类型", choices=UserBalanceChangeCategory.choices())
  6366. sourceObj = GenericLazyReferenceField(verbose_name=u'资金变动来源', required=True)
  6367. dateTimeAdded = DateTimeField(verbose_name=u"变动时间", default=datetime.datetime.now)
  6368. @cached_property
  6369. def source(self):
  6370. return self.sourceObj.fetch()
  6371. @property
  6372. def beforeBalance(self):
  6373. return self.bAmount + self.bBestowAmount
  6374. @property
  6375. def afterBalance(self):
  6376. return self.aBestowAmount + self.aAmount
  6377. def to_dict(self):
  6378. data = {
  6379. "id": str(self.id),
  6380. "category": self.category,
  6381. "beforeBalance": self.beforeBalance,
  6382. "afterBalance": self.afterBalance,
  6383. "dateTimeAdded": self.dateTimeAdded.strftime("%Y-%m-%d %H:%M:%S"),
  6384. "sourceId": str(self.sourceObj.id)
  6385. }
  6386. return data
  6387. class UserBalanceLog(BalanceLog):
  6388. """
  6389. 用户账户余额变化记录
  6390. """
  6391. openId = StringField(verbose_name=u"资金主体标识", required=True)
  6392. productAgentId = StringField(verbose_name=u"平台标识", required=True)
  6393. @classmethod
  6394. def consume(cls, user, afterAmount, afterBestowAmount, consumeAmount, consumeBestowAmount, order): # type: (MyUser, RMB, RMB, RMB, RMB, ConsumeRecord) -> UserBalanceLog
  6395. return cls(
  6396. openId=user.openId,
  6397. productAgentId=user.productAgentId,
  6398. bAmount=afterAmount+consumeAmount,
  6399. bBestowAmount=afterBestowAmount+consumeBestowAmount,
  6400. aAmount=afterAmount,
  6401. aBestowAmount=afterBestowAmount,
  6402. sourceObj=order,
  6403. category=UserBalanceChangeCategory.CONSUME
  6404. ).save()
  6405. @classmethod
  6406. def recharge(cls, user, afterAmount, afterBestowAmount, chargeAmount, chargeBestowAmount, order):
  6407. return cls(
  6408. openId=user.openId,
  6409. productAgentId=user.productAgentId,
  6410. bAmount=afterAmount-chargeAmount,
  6411. bBestowAmount=afterBestowAmount-chargeBestowAmount,
  6412. aAmount=afterAmount,
  6413. aBestowAmount=afterBestowAmount,
  6414. sourceObj=order,
  6415. category=UserBalanceChangeCategory.RECHARGE
  6416. ).save()
  6417. @classmethod
  6418. def refund(cls, user, afterAmount, afterBestowAmount, refundAmount, refundBestowAmount, order):
  6419. return cls(
  6420. openId=user.openId,
  6421. productAgentId=user.productAgentId,
  6422. bAmount=afterAmount-refundAmount,
  6423. bBestowAmount=afterBestowAmount-refundBestowAmount,
  6424. aAmount=afterAmount,
  6425. aBestowAmount=afterBestowAmount,
  6426. sourceObj=order,
  6427. category=UserBalanceChangeCategory.REFUND
  6428. ).save()
  6429. @classmethod
  6430. def get_logs(cls, user, pageIndex, pageSize): # type:(MyUser, int, int) -> QuerySet
  6431. return cls.objects.filter(
  6432. openId=user.openId,
  6433. productAgentId=user.productAgentId
  6434. ).skip((pageIndex-1)*pageSize).limit(pageSize)
  6435. class CardBalanceLog(BalanceLog):
  6436. cardId = StringField(verbose_name=u"资金主体标识", required=True)
  6437. openId = StringField(verbose_name=u"卡的持有人", required=True)
  6438. @classmethod
  6439. def consume(cls, card, afterAmount, afterBestowAmount, consumeAmount, consumeBestowAmount, order): # type: (Card, RMB, RMB, RMB, RMB, ConsumeRecord) -> UserBalanceLog
  6440. return cls(
  6441. cardId=str(card.id),
  6442. openId=card.openId,
  6443. bAmount=afterAmount + consumeAmount,
  6444. bBestowAmount=afterBestowAmount + consumeBestowAmount,
  6445. aAmount=afterAmount,
  6446. aBestowAmount=afterBestowAmount,
  6447. sourceObj=order,
  6448. category=UserBalanceChangeCategory.CONSUME
  6449. ).save()
  6450. @classmethod
  6451. def recharge(cls, card, afterAmount, afterBestowAmount, chargeAmount, chargeBestowAmount, order):
  6452. return cls(
  6453. cardId=str(card.id),
  6454. openId=card.openId,
  6455. bAmount=afterAmount - chargeAmount,
  6456. bBestowAmount=afterBestowAmount - chargeBestowAmount,
  6457. aAmount=afterAmount,
  6458. aBestowAmount=afterBestowAmount,
  6459. sourceObj=order,
  6460. category=UserBalanceChangeCategory.RECHARGE
  6461. ).save()
  6462. @classmethod
  6463. def refund(cls, card, afterAmount, afterBestowAmount, refundAmount, refundBestowAmount, order):
  6464. return cls(
  6465. cardId=str(card.id),
  6466. openId=card.openId,
  6467. bAmount=afterAmount - refundAmount,
  6468. bBestowAmount=afterBestowAmount - refundBestowAmount,
  6469. aAmount=afterAmount,
  6470. aBestowAmount=afterBestowAmount,
  6471. sourceObj=order,
  6472. category=UserBalanceChangeCategory.REFUND
  6473. ).save()
  6474. @classmethod
  6475. def get_logs(cls, card, pageIndex, pageSize):
  6476. return cls.objects.filter(
  6477. cardId=str(card.id)
  6478. ).skip((pageIndex - 1) * pageSize).limit(pageSize)