models.py 240 KB

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