models.py 273 KB

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