controls.py 55 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219122012211222122312241225122612271228122912301231123212331234123512361237123812391240124112421243124412451246124712481249125012511252125312541255125612571258125912601261126212631264126512661267126812691270127112721273127412751276127712781279128012811282128312841285128612871288128912901291129212931294129512961297129812991300130113021303130413051306130713081309131013111312131313141315131613171318131913201321132213231324132513261327132813291330133113321333133413351336133713381339134013411342134313441345134613471348134913501351135213531354135513561357135813591360136113621363136413651366136713681369137013711372137313741375137613771378137913801381138213831384138513861387138813891390139113921393139413951396139713981399140014011402140314041405140614071408140914101411141214131414141514161417141814191420142114221423142414251426142714281429143014311432143314341435143614371438143914401441144214431444144514461447144814491450145114521453
  1. #coding=utf-8
  2. '''
  3. '''
  4. import os,re,random
  5. import json,time,datetime
  6. import shutil
  7. import tempfile
  8. import zipfile,requests
  9. from docxtpl import DocxTemplate,InlineImage
  10. from docx.shared import Mm
  11. from django.db import transaction
  12. from django.conf import settings
  13. import common.models as cm
  14. import common.error_info as ce
  15. import common.common_functions as ccf
  16. import common.common_control as ccc
  17. import account.password_handle as ph
  18. from utils.exceltool import ExcelTool
  19. from utils.jgpush import send_notification_by_registration_ids
  20. import wzhifuSDK as wxpay
  21. from utils.exceltool import ExcelTool
  22. from utils.qrcodetool import gen_general_qrcode
  23. from django.db.models import Q,Sum,Count
  24. from PIL import Image
  25. from PIL import Image,ImageDraw,ImageFont
  26. from threading import Thread
  27. #from django.core.cache import cache
  28. from django.db import connection
  29. from utils.wxSubscribeMessage import send_consult_reply_message
  30. from utils.aliyunpush import aliyunpush
  31. from tools.article_spider import startSpider
  32. from utils.plottool import plot_records,get_zipfile
  33. import xlrd
  34. import xlwt
  35. from xlutils.copy import copy
  36. from xltpl.writer import BookWriter
  37. from weixin.wzhifuSDK_V3 import refund_order
  38. @ccc.cache_data()
  39. def get_user_info(uid):
  40. user = cm.UserInfo.objects.filter(id=uid).values().first()
  41. if user:
  42. user["style"] = []
  43. if user["zq"]:
  44. user["style"].append(user["zq"])
  45. if user["cw"]:
  46. user["style"].append(user["cw"])
  47. if user["df"]:
  48. user["style"].append(user["df"])
  49. return user
  50. @ccc.cache_data()
  51. def get_match_info(match_id):
  52. match = cm.Match.objects.filter(id=match_id).values().first()
  53. return match
  54. def async(f):
  55. def wrapper(*args, **kwargs):
  56. thr = Thread(target=f, args=args, kwargs=kwargs)
  57. thr.start()
  58. return wrapper
  59. def get_match_validdates(match_id):
  60. """
  61. """
  62. #qdata = request.json
  63. #match_id = qdata.get("match_id")
  64. match = cm.Match.objects.filter(id=match_id).first()
  65. if match:
  66. validdates = json.loads(match.valid_dates) if match.valid_dates else []
  67. validdates.sort()
  68. return validdates
  69. return []
  70. def add_model(cls,**kwargs):
  71. """
  72. """
  73. model_name = re.search(r'.*\.(\w+)View',str(cls.__class__)).groups()[0]
  74. model = getattr(cm,model_name)
  75. if model_name == "UserInfo":
  76. username = kwargs.get("username")
  77. usercode = kwargs.get("usercode")
  78. phone = kwargs.get("phone")
  79. if cm.UserInfo.objects.filter(username=username).exists():
  80. raise ce.TipException(u"用户名重复!")
  81. if cm.UserInfo.objects.filter(usercode=usercode).exists():
  82. raise ce.TipException(u"用户代码重复!")
  83. if cm.UserInfo.objects.filter(phone=phone).exists():
  84. raise ce.TipException(u"手机号重复!")
  85. if model_name == "Match":
  86. name = kwargs.get("name")
  87. if cm.Match.objects.filter(name=name).exists():
  88. raise ce.TipException(u"比赛名称重复!")
  89. if model_name == "MatchGoup":
  90. name = kwargs.get("name")
  91. match_id = kwargs.get("match_id")
  92. if cm.MatchGroup.objects.filter(name=name,match_id=match_id).exists():
  93. raise ce.TipException(u"分组名称重复!")
  94. if model_name == "Player":
  95. user_id = kwargs.get("user_id")
  96. match_id = kwargs.get("match_id")
  97. if cm.Player.objects.filter(user_id=user_id,match_id=match_id).exists():
  98. raise ce.TipException(u"重复添加选手!")
  99. user =cm.UserInfo.objects.filter(id=user_id).first()
  100. username = user.username if user else ""
  101. usercode = user.usercode if user else ""
  102. kwargs["username"] = username
  103. kwargs["usercode"] = usercode
  104. if model_name == "PlayerRecord":
  105. #return None
  106. match_id = kwargs.get("match_id")
  107. stock_date = kwargs.get("stock_date")
  108. today_stock = json.dumps(kwargs.get("today_stock"))
  109. today_stock_img = json.dumps(kwargs.get("today_stock_img"))
  110. player_id = kwargs.get("player_id")
  111. today_fund = float(kwargs.get("today_fund"))
  112. is_markt = int(kwargs.get("is_markt",0))
  113. player = cm.Player.objects.filter(id=player_id).first()
  114. user_id = player.user_id
  115. init_fund = player.fund
  116. user = cm.UserInfo.objects.filter(id=user_id).first()
  117. username = user.username
  118. usercode = user.usercode
  119. match_group = player.match_group
  120. match = cm.Match.objects.filter(id=match_id).first()
  121. if player.match_status < 1:
  122. raise ce.TipException(u"该账号已暂停/退出比赛,如有疑问请联系管理员获取详情信息!")
  123. now_date = datetime.datetime.now().strftime("%Y-%m-%d")
  124. if match.end_time < now_date:
  125. raise ce.TipException(u"该比赛已结束!")
  126. #yesterday_date = (datetime.datetime.now()-datetime.timedelta(days=1)).strftime("%Y-%m-%d")
  127. #yesterday = cm.PlayerRecord.get_db_model(match_id).objects.filter(
  128. # match_id=match_id,player_id=player_id).order_by("-stock_date").first()
  129. validdates = get_match_validdates(match_id)
  130. today_index = validdates.index(stock_date)
  131. yesterday_index = today_index -1 if today_index >=1 else 0
  132. yesterday_date = validdates[yesterday_index]
  133. yesterday = cm.PlayerRecord.get_db_model(match_id).objects.filter(match_id=match_id,player_id=player_id,
  134. stock_date=yesterday_date).exclude(stock_date=stock_date).first()
  135. if yesterday:
  136. yesterday_fund = yesterday.today_fund
  137. yesterday_stock = yesterday.today_stock
  138. yesterday_stock_img = yesterday.today_stock_img
  139. yesterday_is_markt = yesterday.is_markt
  140. else:
  141. yesterday_fund = today_fund
  142. yesterday_stock = ""
  143. yesterday_stock_img = ""
  144. yesterday_is_markt = 0
  145. if kwargs.get("id"):
  146. if cm.PlayerRecord.get_db_model(match_id).objects.filter(match_id=match_id,user_id=user_id).count()==1:
  147. init_fund = today_fund
  148. cm.Player.objects.filter(user_id=user_id,match_id=match_id).update(fund=today_fund)
  149. else:
  150. if not cm.PlayerRecord.get_db_model(match_id).objects.filter(match_id=match_id,user_id=user_id).exists():
  151. init_fund = today_fund
  152. cm.Player.objects.filter(user_id=user_id,match_id=match_id).update(fund=today_fund)
  153. obj,flag = cm.PlayerRecord.get_db_model(match_id).objects.get_or_create(
  154. player_id=player_id,
  155. match_id=match_id,
  156. stock_date=stock_date)
  157. obj.init_fund = init_fund
  158. obj.yesterday_fund = yesterday_fund
  159. obj.yesterday_stock = yesterday_stock
  160. obj.yesterday_stock_img = yesterday_stock_img
  161. obj.today_fund = today_fund
  162. obj.today_stock = today_stock
  163. obj.today_stock_img = today_stock_img
  164. obj.user_id = user_id
  165. obj.username = username
  166. obj.usercode = usercode
  167. obj.match_group = match_group
  168. obj.is_markt = is_markt
  169. obj.yesterday_is_markt = yesterday_is_markt
  170. #计算今日和昨日盈亏
  171. today_income = (today_fund - yesterday_fund)/float(yesterday_fund) if yesterday_fund else 0.0
  172. total_income = (today_fund - init_fund)/float(init_fund) if init_fund else 0.0
  173. obj.today_income = round(today_income,4)
  174. obj.total_income = round(total_income,4)
  175. if not flag:
  176. obj.ctime = datetime.datetime.now()
  177. obj.save()
  178. #更新group_rank
  179. return obj.id
  180. if model_name == "PlayerRecord":
  181. match_id = kwargs.get("match_id")
  182. obj = model.get_db_model(match_id).objects.create(**kwargs)
  183. else:
  184. obj = model.objects.create(**kwargs)
  185. #
  186. if model_name == "Match":
  187. cm.MatchGroup.objects.create(
  188. match_id = obj.id,
  189. name = obj.name+u"游客组",
  190. is_active = 0,
  191. charge = 0
  192. )
  193. return obj.id
  194. def update_model(cls,**kwargs):
  195. """
  196. """
  197. model_name = re.search(r'.*\.(\w+)View',str(cls.__class__)).groups()[0]
  198. model = getattr(cm,model_name)
  199. id = kwargs.pop("id")
  200. if model_name == "UserInfo":
  201. username = kwargs.get("username")
  202. usercode = kwargs.get("usercode")
  203. phone = kwargs.get("phone")
  204. if cm.UserInfo.objects.exclude(id=id).filter(username=username).exists():
  205. raise ce.TipException(u"用户名重复!")
  206. if cm.UserInfo.objects.exclude(id=id).filter(usercode=usercode).exists():
  207. raise ce.TipException(u"用户代码重复!")
  208. if phone and cm.UserInfo.objects.exclude(id=id).filter(phone=phone).exists():
  209. raise ce.TipException(u"手机号重复!")
  210. if model_name == "Player":
  211. user_id = kwargs.get("user_id")
  212. match_id = kwargs.get("match_id")
  213. match_group = kwargs.get("match_group")
  214. if cm.Player.objects.exclude(id=id).filter(user_id=user_id,match_id=match_id).exists():
  215. raise ce.TipException(u"重复添加选手!")
  216. user =cm.UserInfo.objects.filter(id=user_id).first()
  217. username = user.username if user else ""
  218. usercode = user.usercode if user else ""
  219. try:
  220. match_group_name = cm.MatchGroup.objects.filter(id=match_group).first().name
  221. except Exception as e:
  222. match_group_name = ""
  223. kwargs["match_group_name"] = match_group_name
  224. if model_name == "PlayerRecord":
  225. is_markt = int(kwargs.get("is_markt",0))
  226. kwargs["today_stock"] = json.dumps(kwargs.get("today_stock"))
  227. kwargs["today_stock_img"] = json.dumps(kwargs.get("today_stock_img"))
  228. if model_name == "PlayerRecord":
  229. match_id = kwargs.get("match_id")
  230. rst = model.get_db_model(match_id).objects.filter(id=id).update(**kwargs)
  231. else:
  232. rst = model.objects.filter(id=id).update(**kwargs)
  233. if model_name == "PlayerRecord":
  234. match_id = kwargs.get("match_id")
  235. user_id = kwargs.get("user_id")
  236. obj = cm.PlayerRecord.get_db_model(match_id).objects.filter(id=id).first()
  237. today_fund = obj.today_fund
  238. yesterday_fund = obj.yesterday_fund
  239. init_fund = obj.init_fund
  240. today_income = (today_fund - yesterday_fund)/float(yesterday_fund)
  241. total_income = (today_fund - init_fund)/float(init_fund)
  242. obj.today_income = round(today_income,4)
  243. obj.total_income = round(total_income,4)
  244. #
  245. player = cm.Player.objects.filter(user_id=user_id,match_id=match_id).first()
  246. if player:
  247. obj.match_group = player.match_group
  248. obj.username = player.username
  249. obj.usercode = player.usercode
  250. #记录持股情况
  251. stock_date = obj.stock_date
  252. today_stock = obj.today_stock
  253. new_stock_list = []
  254. today_stock_list = json.loads(today_stock)
  255. for ts in today_stock_list:
  256. if ts["name"]:
  257. try:
  258. stock,flag = cm.Stock.objects.get_or_create(
  259. name = ts["name"]
  260. )
  261. stock_id = stock.id
  262. usobj,flag = cm.UserStock.objects.get_or_create(
  263. user_id = user_id,
  264. stock_id = stock_id,
  265. stock_name = ts["name"],
  266. player_id = player.id,
  267. stock_date = stock_date
  268. )
  269. if ts.get("fund"):
  270. usobj.fund = ts["fund"]
  271. usobj.save()
  272. ts["stock_id"] = stock_id
  273. new_stock_list.append(ts)
  274. except Exception as e:
  275. print(e)
  276. today_stock = json.dumps(new_stock_list)
  277. obj.today_stock = today_stock
  278. obj.save()
  279. #更新player的init_fund
  280. player_id = player.id
  281. cm.Player.objects.filter(id=player_id).update(fund=init_fund)
  282. ##更新group_rank
  283. update_group_rank(obj.stock_date,obj.match_group,obj.match_id)
  284. if model_name == "UserInfo":
  285. ccc.del_cache("cdata_get_user_info_(%s,)"%id)
  286. return rst
  287. def delete_model(cls,**kwargs):
  288. """
  289. """
  290. model_name = re.search(r'.*\.(\w+)View',str(cls.__class__)).groups()[0]
  291. model = getattr(cm,model_name)
  292. ids = str(kwargs.get("id")).split(",")
  293. if model_name == "WanzhuConsult":
  294. cm.WanzhuConsult.objects.filter(Q(user_id__in=ids)|Q(reply_user_id__in=ids)).delete()
  295. else:
  296. if model_name == "PlayerRecord":
  297. match_id = kwargs.get("match_id")
  298. obj = cm.PlayerRecord.get_db_model(match_id).objects.filter(id__in=ids).first()
  299. ##更新group_rank
  300. update_group_rank(obj.stock_date,obj.match_group,obj.match_id)
  301. rst = model.get_db_model(match_id).objects.filter(id__in=ids).delete()
  302. else:
  303. rst = model.objects.filter(id__in=ids).delete()
  304. if model_name == "UserInfo":
  305. cm.Player.objects.filter(user_id__in=ids).delete()
  306. cm.PlayerRecord.objects.filter(user_id__in=ids).delete()
  307. if model_name == "Player":
  308. cm.PlayerRecord.objects.filter(player_id__in=ids).delete()
  309. if model_name == "Match":
  310. cm.PlayerRecord.objects.filter(match_id__in=ids).delete()
  311. cm.Player.objects.filter(match_id__in=ids).delete()
  312. cm.MatchGroup.objects.filter(match_id__in=ids).delete()
  313. if model_name == "MatchGroup":
  314. cm.PlayerRecord.objects.filter(match_group__in=ids).delete()
  315. cm.Player.objects.filter(match_group__in=ids).delete()
  316. if model_name == "BaikeCategory":
  317. cm.BaikeDetail.objects.filter(category_id__in=ids).delete()
  318. return ids
  319. def get_search_list(cls,**kwargs):
  320. """
  321. """
  322. model_name = re.search(r'.*\.(\w+)SearchView',str(cls.__class__)).groups()[0]
  323. model = getattr(cm,model_name)
  324. qset = model.objects.all().order_by("-id")
  325. if model_name == "Player":
  326. if kwargs.get("usercode"):
  327. qset = qset.filter(Q(usercode__icontains=kwargs.get("usercode"))|Q(username__icontains=kwargs.get("usercode"))|Q(phone__icontains=kwargs.get("usercode")))
  328. data = list(qset.values("id","user_id","username","usercode"))
  329. for item in data:
  330. item["label"] = "%s(%s)" % (item["username"],item["usercode"])
  331. else:
  332. data = []
  333. elif model_name == "Stock":
  334. if kwargs.get("name"):
  335. qset = qset.filter(Q(name__icontains=kwargs.get("name"))|Q(code__icontains=kwargs.get("name")))
  336. data = list(qset.values("id","name","code"))
  337. for item in data:
  338. item["label"] = "%s(%s)" % (item["name"],item["code"])
  339. else:
  340. data = []
  341. elif model_name == "UserInfo":
  342. if kwargs.get("name"):
  343. qset = qset.filter(username__icontains=kwargs.get("name"))
  344. data = list(qset.values("id","username","usercode","avatar"))
  345. for item in data:
  346. item["label"] = "%s(%s)" % (item["username"],item["usercode"])
  347. else:
  348. data = []
  349. elif model_name == "MatchGroup":
  350. if kwargs.get("match_id"):
  351. qset = qset.filter(match_id__icontains=kwargs.get("match_id"))
  352. if kwargs.get("is_active"):
  353. qset = qset.filter(is_active__icontains=kwargs.get("is_active"))
  354. data = list(qset.values("id","name"))
  355. else:
  356. if kwargs.get("name"):
  357. qset = qset.filter(name__icontains=kwargs.get("name"))
  358. data = list(qset.values("id","name"))
  359. return data
  360. def get_detail_info(cls,**kwargs):
  361. """
  362. """
  363. model_name = re.search(r'.*\.(\w+)View',str(cls.__class__)).groups()[0]
  364. model = getattr(cm,model_name)
  365. id = kwargs.get("id")
  366. if model_name == "PlayerRecord":
  367. match_id = kwargs.get("match_id")
  368. rst = list(model.get_db_model(match_id).objects.filter(id=id).values())
  369. else:
  370. rst = list(model.objects.filter(id=id).values())
  371. rst = rst[0] if rst else {}
  372. if model_name == "Player":
  373. user = get_user_info(rst["user_id"])
  374. match = get_match_info(rst["match_id"])
  375. rst["username"] = user["username"]
  376. rst["usercode"] = user["usercode"]
  377. rst["match_name"] = match["name"] if match else ""
  378. rst["match_group"] = int(rst["match_group"]) if rst["match_group"] else 0
  379. if model_name == "PlayerRecord":
  380. rst["today_stock"] = json.loads(rst["today_stock"]) if rst["today_stock"] else []
  381. rst["today_stock_img"] = json.loads(rst["today_stock_img"]) if rst["today_stock_img"] else []
  382. rst["yesterday_stock"] = json.loads(rst["yesterday_stock"]) if rst["yesterday_stock"] else []
  383. rst["yesterday_stock_img"] = json.loads(rst["yesterday_stock_img"]) if rst["yesterday_stock_img"] else []
  384. rst["is_markt"] = bool(rst["is_markt"])
  385. return rst
  386. def get_group_income(match_id,match_group,stock_date=None):
  387. """
  388. """
  389. data = []
  390. if stock_date:
  391. today = stock_date
  392. else:
  393. today = datetime.datetime.now().strftime("%Y-%m-%d")
  394. player_ids = list(cm.Player.objects.filter(match_id=match_id,match_group=match_group).values_list("id",flat=True))
  395. for player_id in player_ids:
  396. today_record = cm.PlayerRecord.objects.filter(match_id=match_id,match_group=match_group,player_id=player_id,stock_date=today).order_by("-stock_date").first()
  397. if today_record:
  398. data.append(today_record.total_income)
  399. data = list(set(data))
  400. data = sorted(data,key=lambda x:x,reverse=True)
  401. return data
  402. def get_list_info(cls,**kwargs):
  403. """
  404. """
  405. model_name = re.search(r'.*\.(\w+)ListView',str(cls.__class__)).groups()[0]
  406. model = getattr(cm,model_name)
  407. if model_name == "PlayerRecord":
  408. match_id = kwargs.get("match_id",0)
  409. qset = model.get_db_model(match_id).objects.all()
  410. else:
  411. qset = model.objects.all()
  412. if kwargs.get("name"):
  413. qset = qset.filter(name__icontains=kwargs.get("name"))
  414. if model_name == "UserInfo":
  415. if kwargs.get("username"):
  416. qset = qset.filter(Q(username__icontains=kwargs.get("username"))|Q(usercode__icontains=kwargs.get("username"))|Q(phone__icontains=kwargs.get("username")))
  417. if model_name == "Player":
  418. if kwargs.get("username"):
  419. qset = qset.filter(Q(username__icontains=kwargs.get("username"))|Q(usercode__icontains=kwargs.get("username")))
  420. if kwargs.get("match_id"):
  421. qset = qset.filter(match_id=kwargs.get("match_id"))
  422. if kwargs.get("groupId"):
  423. qset = qset.filter(match_group=kwargs.get("groupId"))
  424. if model_name == "MatchGroup":
  425. if kwargs.get("match_id"):
  426. qset = qset.filter(match_id=kwargs.get("match_id"))
  427. if model_name == "PlayerRecord":
  428. if kwargs.get("match_id"):
  429. qset = qset.filter(match_id=kwargs.get("match_id"))
  430. if kwargs.get("groupId"):
  431. qset = qset.filter(match_group=kwargs.get("groupId"))
  432. if kwargs.get("username"):
  433. qset = qset.filter(Q(username__icontains=kwargs.get("username"))|Q(usercode__icontains=kwargs.get("username")))
  434. if kwargs.get("stock_date"):
  435. qset = qset.filter(stock_date=kwargs.get("stock_date"))
  436. if kwargs.get("auto_complete"):
  437. if int(kwargs.get("auto_complete"))==1:
  438. qset = qset.filter(auto_complete__gte=kwargs.get("auto_complete"))
  439. else:
  440. qset = qset.filter(auto_complete=kwargs.get("auto_complete"))
  441. if model_name == "SignupOrder":
  442. if kwargs.get("match_id"):
  443. qset = qset.filter(match_id=kwargs.get("match_id"))
  444. if kwargs.get("username"):
  445. qset = qset.filter(Q(user_name__icontains=kwargs.get("username"))|Q(phone__icontains=kwargs.get("username")))
  446. if kwargs.get("signup_type"):
  447. qset = qset.filter(signup_type=kwargs.get("signup_type"))
  448. if kwargs.get("order_status") or kwargs.get("order_status")==0:
  449. qset = qset.filter(order_status=kwargs.get("order_status"))
  450. if kwargs.get("pay_status") or kwargs.get("pay_status")==0:
  451. qset = qset.filter(pay_status=kwargs.get("pay_status"))
  452. if model_name == "Article":
  453. if kwargs.get("type"):
  454. qset = qset.filter(type=kwargs.get("type"))
  455. if kwargs.get("category"):
  456. qset = qset.filter(category=kwargs.get("category"))
  457. if kwargs.get("isbanner"):
  458. qset = qset.filter(isbanner=kwargs.get("isbanner"))
  459. if kwargs.get("ishot"):
  460. qset = qset.filter(ishot=kwargs.get("ishot"))
  461. if model_name == "Stock":
  462. if kwargs.get("username"):
  463. qset = qset.filter(name__icontains=kwargs.get("username"))
  464. if model_name == "BaikeDetail":
  465. if kwargs.get("name"):
  466. qset = qset.filter(name__icontains=kwargs.get("name"))
  467. if kwargs.get("status"):
  468. qset = qset.filter(status=kwargs.get("status"))
  469. if kwargs.get("category_id"):
  470. qset = qset.filter(category_id=kwargs.get("category_id"))
  471. if model_name == "PlayerRecord":
  472. if kwargs.get("username"):
  473. data = list(qset.order_by("-stock_date").values())
  474. else:
  475. data = list(qset.order_by("-total_income").values())
  476. elif model_name == "Player":
  477. data = list(qset.order_by("-fund").values())
  478. else:
  479. data = list(qset.order_by("-id").values())
  480. page = int(kwargs.get("page",1))
  481. page_size = int(kwargs.get("page_size",20))
  482. if page and page_size:
  483. total,data = ccf.get_page_list(data,page,page_size)
  484. if model_name == "Player":
  485. for item in data:
  486. user = get_user_info(item["user_id"])
  487. if user:
  488. match = get_match_info(item["match_id"])
  489. item["nickname"] = user["nickname"]
  490. item["match_name"] = match["name"] if match else ""
  491. item["openid"] = user["openid"]
  492. item["player_type"] = user["player_type"]
  493. item["phone"] = user.get('phone', '')
  494. if model_name == "PlayerRecord":
  495. for item in data:
  496. user = cm.UserInfo.objects.filter(id=item["user_id"]).first()
  497. item["username"] = user.username if user else ""
  498. item["usercode"] = user.usercode if user else ""
  499. today_stock = json.loads(item["today_stock"]) if item["today_stock"] else []
  500. today_stock = filter(lambda x:x["name"],today_stock if today_stock else [])
  501. today_stock_img = json.loads(item["today_stock_img"]) if item["today_stock_img"] else []
  502. item["today_stock"] = today_stock
  503. ilist = []
  504. for img in today_stock_img:
  505. nlist = img.split('?')
  506. if len(nlist) > 0:
  507. ilist.append(nlist[0])
  508. item["today_stock_img"] = ilist
  509. yesterday_stock = json.loads(item["yesterday_stock"]) if item["yesterday_stock"] else []
  510. yesterday_stock = filter(lambda x:x["name"],yesterday_stock if yesterday_stock else [])
  511. item["yesterday_stock"] = yesterday_stock
  512. item["yesterday_stock_img"] = json.loads(item["yesterday_stock_img"]) if item["yesterday_stock_img"] else []
  513. if item["today_income"]:
  514. item["today_income"] = "{}%".format(item["today_income"]*100)
  515. if item["total_income"]:
  516. item["total_income"] = "{}%".format(item["total_income"]*100)
  517. item["is_markt"] = bool(item["is_markt"])
  518. #资金出入
  519. if cm.FundInOut.objects.filter(user_id=item["user_id"],stock_date=item["stock_date"]).exists():
  520. item["fund_inout"] = cm.FundInOut.objects.filter(user_id=item["user_id"],stock_date=item["stock_date"]).values().first()
  521. if kwargs.get("groupId"):
  522. data = sorted(data,key=lambda x:x["group_rank"])
  523. if model_name == "Match":
  524. for item in data:
  525. groups = ",".join(list(cm.MatchGroup.objects.filter(match_id=item["id"]).values_list("name",flat=True)))
  526. item["join_count"] = cm.Player.objects.filter(match_id=item["id"]).count()
  527. item["out_count"] = cm.Player.objects.filter(match_id=item["id"],match_status=-1).count()
  528. item["groups"] = groups
  529. if model_name == "MatchGroup":
  530. for item in data:
  531. match = cm.Match.objects.filter(id=item["match_id"]).first()
  532. match_name = match.name if match else ""
  533. item["join_count"] = cm.Player.objects.filter(match_group=item["id"]).count()
  534. item["out_count"] = cm.Player.objects.filter(match_group=item["id"],match_status=-1).count()
  535. item["match_name"] = match_name
  536. if model_name == "SignupOrder":
  537. for item in data:
  538. user = cm.UserInfo.objects.filter(id=item["user_id"]).first()
  539. try:
  540. item["match_group_name"] = cm.MatchGroup.objects.filter(id=item["match_group"]).first().name
  541. except Exception as e:
  542. print(e)
  543. item["match_group_name"] = ""
  544. if user:
  545. item["usercode"] = user.usercode
  546. item["nickname"] = user.nickname
  547. item["openid"] = user.openid
  548. if model_name == "FundInOut":
  549. for item in data:
  550. user = cm.UserInfo.objects.filter(id=item["user_id"]).first()
  551. item["username"] = user.username
  552. item["usercode"] = user.usercode
  553. if kwargs.get("username"):
  554. data = filter(lambda x:x["username"] == str(kwargs.get("username")) or kwargs.get("username") ==x["usercode"],data)
  555. return (total,data)
  556. else:
  557. return len(data),data
  558. return rst
  559. def download_records(request):
  560. """
  561. """
  562. kwargs = request.json
  563. header = [u"NO.",u"选手",u"初始资产",u"今日资产",u"今日盈亏",u"总盈亏",u"昨日持股\n(万元)",u"今日持股\n(万元)"]
  564. match_id = kwargs.get("match_id")
  565. qset = cm.PlayerRecord.get_db_model(match_id).objects.all()
  566. if kwargs.get("match_id"):
  567. qset = qset.filter(match_id=kwargs.get("match_id"))
  568. if kwargs.get("groupId"):
  569. qset = qset.filter(match_group=kwargs.get("groupId"))
  570. if kwargs.get("username"):
  571. qset = qset.filter(Q(username__icontains=kwargs.get("username"))|Q(usercode__icontains=kwargs.get("username")))
  572. if kwargs.get("stock_date"):
  573. qset = qset.filter(stock_date=kwargs.get("stock_date"))
  574. qdata = list(qset.order_by("-total_income").values())
  575. if kwargs.get("groupId"):
  576. qdata = sorted(qdata,key=lambda x:x["group_rank"])
  577. data = []
  578. for item in qdata:
  579. try:
  580. user = cm.UserInfo.objects.filter(id=item["user_id"]).first()
  581. item["username"] = user.username
  582. item["usercode"] = user.usercode
  583. item["today_income"] = "{}%".format(item["today_income"]*100)
  584. item["total_income"] = "{}%".format(item["total_income"]*100)
  585. try:
  586. match_group_name = cm.MatchGroup.objects.filter(id=item["match_group"]).first().name
  587. except:
  588. match_group_name = ""
  589. item["match_group_name"] = match_group_name
  590. item["today_stock_imgs"] = json.loads(item["today_stock_img"]) if item["today_stock_img"] else []
  591. today_stock = json.loads(item["today_stock"]) if item["today_stock"] else []
  592. today_stock = list(filter(lambda x:x["name"],today_stock if today_stock else []))
  593. today_stock = ",".join([x["name"]+str(x["fund"]) for x in today_stock]) if today_stock else ""
  594. yesterday_stock = json.loads(item["yesterday_stock"]) if item["yesterday_stock"] else []
  595. yesterday_stock = list(filter(lambda x:x["name"],yesterday_stock if yesterday_stock else []))
  596. yesterday_stock = ",".join([x["name"]+str(x["fund"]) for x in yesterday_stock]) if yesterday_stock else ""
  597. #空仓、开超市、请假判断
  598. if item["auto_complete"] > 0:
  599. today_stock = u"请假"
  600. else:
  601. if item["is_markt"] == 1:
  602. today_stock = u"开超市"
  603. else:
  604. if today_stock == "":
  605. today_stock = u"空仓"
  606. #昨日
  607. if item["yesterday_auto_complete"] > 0:
  608. yesterday_stock = u"请假"
  609. else:
  610. if item["yesterday_is_markt"] == 1:
  611. yesterday_stock = u"开超市"
  612. else:
  613. if yesterday_stock == "":
  614. yesterday_stock = u"空仓"
  615. data.append(
  616. [
  617. item["group_rank"],
  618. item["username"],
  619. item["init_fund"],
  620. item["today_fund"],
  621. item["today_income"],
  622. item["total_income"],
  623. yesterday_stock,
  624. today_stock,
  625. ''
  626. ])
  627. except Exception as e:
  628. print(e)
  629. xlsfile = os.path.join(settings.BASE_DIR,"demo.xls")
  630. exceltool = ExcelTool(xlsfile)
  631. groupName = kwargs.get("groupName")
  632. if not groupName:
  633. groupName = u"全部数据"
  634. xlsdata = exceltool.save_data(groupName,header,data)
  635. return xlsdata
  636. def download_records_image(request):
  637. """
  638. return: filename: str; binary data of the image: bytes.
  639. """
  640. kwargs = request.json
  641. match_id = kwargs.get("match_id")
  642. if not match_id:
  643. return None
  644. qset = cm.PlayerRecord.get_db_model(match_id).objects.all()
  645. match = cm.Match.objects.filter(id=match_id).values('name', 'start_time', 'end_time').first()
  646. if match is None:
  647. return None
  648. match_name = match['name'][:7]
  649. match_start_day_str = datetime.datetime.strptime(match['start_time'], '%Y-%m-%d').strftime('%-m.%-d')
  650. match_stop_day_str = datetime.datetime.strptime(match['end_time'], '%Y-%m-%d').strftime('%-m.%-d')
  651. if kwargs.get("match_id"):
  652. qset = qset.filter(match_id=kwargs.get("match_id"))
  653. if kwargs.get("groupId"):
  654. qset = qset.filter(match_group=kwargs.get("groupId"))
  655. match_group = cm.MatchGroup.objects.filter(id=kwargs.get('groupId')).values('name').first()
  656. if match_group is None:
  657. match_group_name = ''
  658. else:
  659. match_group_name = match_group['name']
  660. else:
  661. match_group_name = ''
  662. if kwargs.get("username"):
  663. qset = qset.filter(Q(username__icontains=kwargs.get("username"))|Q(usercode__icontains=kwargs.get("username")))
  664. if kwargs.get("stock_date"):
  665. qset = qset.filter(stock_date=kwargs.get("stock_date"))
  666. # this formatting method only works on linux
  667. stock_day_str = datetime.datetime.strptime(kwargs.get("stock_date"), '%Y-%m-%d').strftime('%-m.%-d')
  668. if match_group_name:
  669. title = u'{}-{}{}({}-{})'.format(match_name, match_group_name, stock_day_str, match_start_day_str, match_stop_day_str)
  670. else:
  671. title = u'{}-{}'.format(match_name, stock_day_str)
  672. else:
  673. if match_group_name:
  674. title = u'{}-{}'.format(match_name, match_group_name)
  675. else:
  676. title = match_name
  677. qdata = list(qset.order_by("-total_income").values())
  678. if kwargs.get("groupId"):
  679. qdata = sorted(qdata,key=lambda x:x["group_rank"])
  680. data = []
  681. for item in qdata:
  682. try:
  683. user = cm.UserInfo.objects.filter(id=item["user_id"]).first()
  684. item["username"] = user.username
  685. item["usercode"] = user.usercode
  686. item["today_income"] = "{}%".format(item["today_income"]*100)
  687. item["total_income"] = "{}%".format(item["total_income"]*100)
  688. try:
  689. match_group_name = cm.MatchGroup.objects.filter(id=item["match_group"]).first().name
  690. except:
  691. match_group_name = ""
  692. item["match_group_name"] = match_group_name
  693. item["today_stock_imgs"] = json.loads(item["today_stock_img"]) if item["today_stock_img"] else []
  694. today_stock = json.loads(item["today_stock"]) if item["today_stock"] else []
  695. today_stock = list(filter(lambda x:x["name"],today_stock if today_stock else []))
  696. today_stock = ",".join([x["name"]+str(x["fund"]) for x in today_stock if x['fund'] != '0.0']) if today_stock else ""
  697. yesterday_stock = json.loads(item["yesterday_stock"]) if item["yesterday_stock"] else []
  698. yesterday_stock = list(filter(lambda x:x["name"],yesterday_stock if yesterday_stock else []))
  699. yesterday_stock = ",".join([x["name"]+str(x["fund"]) for x in yesterday_stock if x['fund'] != '0.0']) if yesterday_stock else ""
  700. #空仓、开超市、请假判断
  701. if item["auto_complete"] > 0:
  702. today_stock = u"请假"
  703. else:
  704. if item["is_markt"] == 1:
  705. today_stock = u"开超市"
  706. else:
  707. if today_stock == "":
  708. today_stock = u"空仓"
  709. #昨日
  710. if item["yesterday_auto_complete"] > 0:
  711. yesterday_stock = u"请假"
  712. else:
  713. if item["yesterday_is_markt"] == 1:
  714. yesterday_stock = u"开超市"
  715. else:
  716. if yesterday_stock == "":
  717. yesterday_stock = u"空仓"
  718. data.append([item["group_rank"],
  719. item["username"],
  720. item["init_fund"],
  721. item["today_fund"],
  722. item["today_income"],
  723. item["total_income"],
  724. yesterday_stock,
  725. today_stock])
  726. except Exception as e:
  727. print(e)
  728. col_headers = [u"NO.", u"选手", u"初始", u"今日",
  729. u"今日成绩", u"总成绩", u"昨日", u"今日"]
  730. ret = plot_records(data, col_headers, title, 'player_records')
  731. return ret
  732. def fast_save_player(**kwargs):
  733. """
  734. """
  735. username = kwargs.get("username")
  736. usercode = kwargs.get("usercode")
  737. if cm.UserInfo.objects.filter(username=username).exists():
  738. raise ce.TipException(u"用户名重复!")
  739. if cm.UserInfo.objects.filter(usercode=usercode).exists():
  740. raise ce.TipException(u"用户代码重复!")
  741. user = cm.UserInfo.objects.create(username=username,usercode=usercode)
  742. #新建选手
  743. user_id = user.id
  744. kwargs["user_id"] = user_id
  745. cm.Player.objects.create(**kwargs)
  746. def update_player_latest(record):
  747. """更新选手最后一次数据
  748. """
  749. user_id = record["user_id"]
  750. match_id = record["match_id"]
  751. stock_date = record["stock_date"]
  752. match_group = record["match_group"]
  753. wincnt = record["wincnt"]
  754. total = record["totalcnt"]
  755. key = "PLAYER_LATEST_{}".format(user_id)
  756. if record:
  757. ccc.cache.hset(key,"stock_date",stock_date)
  758. ccc.cache.hset(key,"match_id",match_id)
  759. ccc.cache.hset(key,"match_group",match_group)
  760. #更新胜率
  761. qset = cm.PlayerRecord.get_db_model(match_id).objects.filter(match_id=match_id,user_id=user_id)
  762. #win_rate = qset.filter(today_income__gte=0).count()/float(qset.count()) if qset else 0.0
  763. win_rate = wincnt/float(total) if total else 0.0
  764. win_rate = round(win_rate,3)
  765. ccc.cache.hset(key,"win_rate",win_rate)
  766. #badest = cm.PlayerRecord.get_db_model(match_id).objects.filter(user_id=user_id,match_id=match_id).order_by("today_income").first()
  767. #if badest:
  768. # ccc.cache.hset(key,"badest_income",badest.today_income)
  769. def update_group_rank(stock_date=None,match_id=None,group_id=None):
  770. """
  771. """
  772. if stock_date:
  773. stock_date = stock_date
  774. else:
  775. stock_date = datetime.datetime.now().strftime("%Y-%m-%d")
  776. del_keys = ccc.cache.keys("*_%s_%s_%s" % (match_id,group_id,stock_date))
  777. if del_keys:
  778. ccc.cache.delete(*del_keys)
  779. groups = cm.MatchGroup.objects.filter(id=group_id)
  780. for gp in groups:
  781. match_id = gp.match_id
  782. match_group = gp.id
  783. prset = cm.PlayerRecord.get_db_model(match_id).objects.filter(match_id=match_id,match_group=match_group,stock_date=stock_date).order_by("-total_income")
  784. if prset:
  785. records = prset.values()
  786. case_id = " case id "
  787. cases = []
  788. where = []
  789. for index,pr in enumerate(records):
  790. case = "WHEN %s THEN %s" % (pr["id"],index+1)
  791. cases.append(case)
  792. where.append(str(pr["id"]))
  793. #
  794. key = "%s_%s_%s_%s" % (pr["user_id"],match_id,match_group,stock_date)
  795. pr.update({"group_rank":index+1})
  796. ccc.pl.set(key,json.dumps(pr,cls=ccc.CusJSONEncoder))
  797. ccc.pl.execute()
  798. winset = cm.PlayerRecord.get_db_model(match_id).objects.filter(
  799. match_id=match_id,match_group=match_group,today_income__gte=0)\
  800. .values("user_id").annotate(wincnt=Count("user_id")).values("user_id","match_id","match_group","wincnt")
  801. totalset = cm.PlayerRecord.get_db_model(match_id).objects.filter(
  802. match_id=match_id,match_group=match_group)\
  803. .values("user_id").annotate(totalcnt=Count("user_id")).values("user_id","totalcnt")
  804. totallist = list(totalset)
  805. for item in winset:
  806. item["stock_date"] = stock_date
  807. user_id = item["user_id"]
  808. total = filter(lambda x:x["user_id"]==user_id,totallist)
  809. if total:
  810. item.update(total[0])
  811. else:
  812. item.update({"totalcnt":0})
  813. if total:
  814. #更新最后一次数据
  815. update_player_latest(item)
  816. if cases and where:
  817. case = case_id + " ".join(cases)
  818. where = ",".join(where)
  819. sql = "update player_record_%s set group_rank = %s ELSE 0 END where id in (%s)" % (match_id,case,where)
  820. cursor = connection.cursor()
  821. cursor.execute(sql)
  822. cursor.close()
  823. ##更新最后一次数据
  824. #update_player_latest(pr)
  825. def update_comment(**kwargs):
  826. """
  827. """
  828. id = kwargs.get("id")
  829. match_id = kwargs.get("match_id")
  830. wanzhu_comment = kwargs.get("wanzhu_comment")
  831. cm.PlayerRecord.get_db_model(match_id).objects.filter(id=id).update(wanzhu_comment=wanzhu_comment)
  832. def reset_initfund(**kwargs):
  833. """
  834. """
  835. match_id = kwargs.get("match_id")
  836. cm.Player.objects.filter(match_id=match_id).update(fund=1)
  837. #cm.Player.objects.filter(match_id=match_id,id=3011).update(fund=None)
  838. def get_stock_comments_list(**kwargs):
  839. """
  840. """
  841. qset = cm.Comments.objects.filter(stock_id__isnull=False).order_by(*["-istop","-id"])
  842. if kwargs.get("name"):
  843. qset = qset.filter(stock_name__icontains=kwargs.get("name"))
  844. data = list(qset.values())
  845. for item in data:
  846. item["stock_name"] = cm.Stock.objects.filter(id=item["stock_id"]).first().name
  847. #分页
  848. page = int(kwargs.get("page",1))
  849. page_size = int(kwargs.get("page_size",20))
  850. if page and page_size:
  851. total,data = ccf.get_page_list(data,page,page_size)
  852. else:
  853. total = len(data)
  854. return total,data
  855. def get_record_comments_list(**kwargs):
  856. """
  857. """
  858. match_id = kwargs.get("match_id")
  859. qset = cm.Comments.objects.filter(record_id__isnull=False).order_by(*["-istop","-id"])
  860. if kwargs.get("user_name"):
  861. qset = qset.filter(user_name__icontains=kwargs.get("name"))
  862. data = list(qset.values())
  863. for item in data:
  864. item["record_info"] = cm.PlayerRecord.get_db_model(match_id).objects.filter(id=item["record_id"]).values().first()
  865. #分页
  866. page = int(kwargs.get("page",1))
  867. page_size = int(kwargs.get("page_size",20))
  868. if page and page_size:
  869. total,data = ccf.get_page_list(data,page,page_size)
  870. else:
  871. total = len(data)
  872. return total,data
  873. def get_wanzhu_consult_list(**kwargs):
  874. """
  875. """
  876. qset = cm.Consult.objects.all().order_by("-ctime")
  877. if kwargs.get("username"):
  878. qset = qset.filter(user_name__icontains=kwargs.get("username"))
  879. if kwargs.get("reply_status"):
  880. qset = qset.filter(reply_status=kwargs.get("reply_status"))
  881. if kwargs.get("content"):
  882. qset = qset.filter(Q(content__icontains=kwargs.get("content"))|Q(reply_content__icontains=kwargs.get("content")))
  883. data = list(qset.values())
  884. for item in data:
  885. user_id = item["user_id"]
  886. userinfo = get_user_info(user_id)
  887. item["user_code"] = userinfo.get("usercode")
  888. item["phone"] = userinfo.get("phone")
  889. #分页
  890. page = int(kwargs.get("page",1))
  891. page_size = int(kwargs.get("page_size",20))
  892. if page and page_size:
  893. total,data = ccf.get_page_list(data,page,page_size)
  894. else:
  895. total = len(data)
  896. return total,data
  897. def calc_reply(v):
  898. """计算未回复数量
  899. """
  900. user_ids = [str(x["user_id"])[0] for x in v]
  901. user_id_str = "".join(user_ids)
  902. if v[-1]["user_id"]<=0:
  903. return 0
  904. else:
  905. index = user_id_str.rfind("0")
  906. return len(v) - index-1
  907. def reply_wanzhu_consult(**kwargs):
  908. """
  909. """
  910. pid = kwargs.get("pid")
  911. content = kwargs.get("reply_content")
  912. user_id = kwargs.get("user_id")
  913. action = kwargs.get("action")
  914. if action == "send":
  915. user = cm.UserInfo.objects.filter(id=user_id).first()
  916. user_name = user.username
  917. user_avatar = user.avatar
  918. obj,flag = cm.Consult.objects.get_or_create(
  919. user_id = user_id,
  920. )
  921. obj.user_name = user_name
  922. obj.user_avatar = user_avatar
  923. #obj.reply_content = content
  924. obj.view_status = 1
  925. obj.reply_status = 0
  926. obj.ctime = datetime.datetime.now()
  927. obj.save()
  928. wanzhu_obj = cm.WanzhuConsult.objects.create(
  929. user_id = user_id,
  930. user_name = user_name,
  931. user_avatar = user_avatar
  932. )
  933. wanzhu_obj_1 = cm.WanzhuConsult.objects.create(
  934. reply_content = content,
  935. user_id = 0,
  936. player_id = 0,
  937. reply_user_id = user_id,
  938. pid = wanzhu_obj.id
  939. )
  940. cm.Consult.objects.filter(user_id=user_id).update(reply_status=1,reply_content=content,ctime=datetime.datetime.now())
  941. else:
  942. cm.WanzhuConsult.objects.create(
  943. pid = pid,
  944. reply_content = content,
  945. user_id = 0,
  946. player_id = 0,
  947. reply_user_id = user_id
  948. )
  949. cm.Consult.objects.filter(user_id=user_id).update(reply_status=1,reply_content=content,ctime=datetime.datetime.now())
  950. #发送消息提醒
  951. if cm.UserInfo.objects.filter(id=user_id).exists():
  952. user = cm.UserInfo.objects.filter(id=user_id).first()
  953. openid = user.openid
  954. username = user.username
  955. try:
  956. send_consult_reply_message(openid,username)
  957. except Exception as e:
  958. print(e)
  959. #发送app推送消息
  960. title = u"顽主杯"
  961. body = u"你收到一条顽主的消息"
  962. params = {"type":"message","id":str(user_id)}
  963. aliyunpush.push_notice_by_userid(user_id,title,body,params,"ios")
  964. aliyunpush.push_notice_by_userid(user_id,title,body,params,"android")
  965. def update_player_type(**kwargs):
  966. """
  967. """
  968. player_id = kwargs.get("player_id")
  969. user_id = kwargs.get("user_id")
  970. player_type = kwargs.get("player_type")
  971. phone = kwargs.get("phone")
  972. cm.UserInfo.objects.filter(id=user_id).update(player_type=player_type,phone=phone)
  973. ccc.del_cache("cdata_get_user_info_(%sL,)"%user_id)
  974. def update_player_role(**kwargs):
  975. """
  976. """
  977. player_id = kwargs.get("player_id")
  978. user_id = kwargs.get("user_id")
  979. role = kwargs.get("role")
  980. cm.Player.objects.filter(id=player_id).update(role=role)
  981. def comments_top(**kwargs):
  982. """
  983. """
  984. _id = kwargs.get("id")
  985. istop = 1-cm.Comments.objects.filter(id=_id).first().istop
  986. cm.Comments.objects.filter(id=_id).update(istop=istop)
  987. def comments_black(**kwargs):
  988. """
  989. """
  990. _id = kwargs.get("id")
  991. comment_status = 1-cm.UserInfo.objects.filter(id=_id).first().comment_status
  992. cm.UserInfo.objects.filter(id=_id).update(comment_status=comment_status)
  993. def relation_old_user(**kwargs):
  994. """
  995. """
  996. old_user_id = kwargs.get("old_user_id")
  997. user_id = kwargs.get("user_id")
  998. print(kwargs,1111111111111)
  999. user = cm.UserInfo.objects.filter(id=user_id).first()
  1000. if user:
  1001. if user.openid:
  1002. cm.UserInfo.objects.filter(id=old_user_id).update(openid=user.openid)
  1003. cm.UserInfo.objects.filter(id=user_id).delete()
  1004. def get_openid_miss(**kwargs):
  1005. """
  1006. """
  1007. qset = cm.UserInfo.objects.filter(openid__isnull=True)
  1008. data = list(qset.values())
  1009. return data
  1010. def signup_old_user(**kwargs):
  1011. """
  1012. """
  1013. order_id = kwargs.get("id")
  1014. signup_type = kwargs.get("signup_type")
  1015. match_group = kwargs.get("match_group")
  1016. player_type = kwargs.get("player_type")
  1017. role_type = kwargs.get("role_type")
  1018. order = cm.SignupOrder.objects.filter(id=order_id).first()
  1019. if order:
  1020. user_id = order.user_id
  1021. match_id = order.match_id
  1022. signup_name = order.signup_name
  1023. user = cm.UserInfo.objects.filter(id=user_id).first()
  1024. match_group_name = cm.MatchGroup.objects.filter(id=match_group).first().name
  1025. match = cm.Match.objects.filter(id=match_id).first()
  1026. user.player_type = player_type
  1027. user.phone = order.phone
  1028. user.username = signup_name
  1029. user.save()
  1030. order.signup_type = signup_type
  1031. order.match_group = match_group
  1032. order.match_group_name = match_group_name
  1033. order.order_status = 1
  1034. order.save()
  1035. #生产选手信息
  1036. player,flag = cm.Player.objects.get_or_create(
  1037. user_id = user_id,
  1038. match_id = match_id
  1039. )
  1040. player.match_name = match.name
  1041. player.match_group = match_group
  1042. player.match_group_name = match_group_name
  1043. player.username = signup_name
  1044. player.usercode = user.usercode
  1045. player.role = role_type
  1046. player.match_status = 1
  1047. player.save()
  1048. def get_consult_detail(**kwargs):
  1049. """
  1050. """
  1051. rst = {}
  1052. user_id = kwargs.get("user_id")
  1053. qset = cm.WanzhuConsult.objects.filter(Q(user_id=user_id)|Q(reply_user_id=user_id))
  1054. rst["consult_list"] = list(qset.filter(user_id__gt=0).order_by("ctime").values())
  1055. rst["all_consult_list"] = list(qset.order_by("ctime").values())
  1056. rst["user_id"] = user_id
  1057. cm.Consult.objects.filter(user_id=user_id).update(view_status=1)
  1058. return rst
  1059. def delete_consult(**kwargs):
  1060. """
  1061. """
  1062. user_id = kwargs.get("id")
  1063. cm.Consult.objects.filter(user_id=user_id).delete()
  1064. cm.WanzhuConsult.objects.filter(Q(user_id=user_id)|Q(reply_user_id=user_id)).delete()
  1065. def batch_update_player_latest(record):
  1066. """更新选手最后一次数据
  1067. """
  1068. user_id = record["user_id"]
  1069. match_id = record["match_id"]
  1070. stock_date = record["stock_date"]
  1071. match_group = record["match_group"]
  1072. #wincnt = record["wincnt"]
  1073. #total = record["totalcnt"]
  1074. key = "PLAYER_LATEST_{}".format(user_id)
  1075. if record and match_group:
  1076. ccc.cache.hset(key,"stock_date",stock_date)
  1077. ccc.cache.hset(key,"match_id",match_id)
  1078. ccc.cache.hset(key,"match_group",match_group)
  1079. ##更新胜率
  1080. #qset = cm.PlayerRecord.get_db_model(match_id).objects.filter(match_id=match_id,user_id=user_id)
  1081. ##win_rate = qset.filter(today_income__gte=0).count()/float(qset.count()) if qset else 0.0
  1082. #win_rate = wincnt/float(total) if total else 0.0
  1083. #win_rate = round(win_rate,3)
  1084. #ccc.cache.hset(key,"win_rate",win_rate)
  1085. def get_sub_category(pid):
  1086. """
  1087. """
  1088. qset = cm.BaikeCategory.objects.filter(parent_id=pid)
  1089. qdata = list(qset.values())
  1090. if qdata:
  1091. for subitem in qdata:
  1092. subitem["type"] = "ent"
  1093. subitem["label"] = subitem["name"]
  1094. subitem["children"] = get_sub_category(subitem["id"])
  1095. return qdata
  1096. def get_baike_category_tree(**kwargs):
  1097. parents = list(cm.BaikeCategory.objects.filter(parent_id__isnull=True).values())
  1098. for item in parents:
  1099. item["label"] = item["name"]
  1100. item["children"] = get_sub_category(item["id"])
  1101. return parents
  1102. def get_baike_list_by_category(**kwargs):
  1103. category_id = kwargs.get("category_id")
  1104. qset = cm.BaikeDetail.objects.filter(category_id=category_id)
  1105. data = list(qset.values("id","name"))
  1106. return data
  1107. def push_article(**kwargs):
  1108. _id = kwargs.get("id")
  1109. article = cm.Article.objects.filter(id=_id).first()
  1110. #发送app推送消息
  1111. title = u"顽主杯"
  1112. body = article.name
  1113. user_id = "ALL"
  1114. params = {"type":"article","id":_id}
  1115. aliyunpush.push_notice_by_userid(user_id,title,body,params,"ios")
  1116. aliyunpush.push_notice_by_userid(user_id,title,body,params,"android")
  1117. def spider_article(**kwargs):
  1118. data = startSpider(kwargs.get("url"))
  1119. return data
  1120. def import_article(**kwargs):
  1121. """
  1122. """
  1123. obj = cm.Article.objects.create(**kwargs)
  1124. def refund_out_trade_order(**kwargs):
  1125. """
  1126. """
  1127. _id = kwargs.get("id")
  1128. signup_order = cm.SignupOrder.objects.filter(id=_id).first()
  1129. if signup_order:
  1130. out_trade_no = signup_order.out_trade_no
  1131. transaction_id = signup_order.transaction_id
  1132. total_fee = signup_order.total_fee
  1133. amount = str(int(total_fee*100))
  1134. print(amount)
  1135. result = refund_order(transaction_id,out_trade_no,out_trade_no,amount)
  1136. if result.get("result_code") == "SUCCESS":
  1137. signup_order.order_status = -1
  1138. signup_order.save()
  1139. return "success"
  1140. else:
  1141. raise ce.TipException(result.get("err_code_des"))
  1142. else:
  1143. raise ce.TipException(u"该订单不存在退款失败!")
  1144. # 计算出入资金
  1145. def caculate_inout(**kwargs):
  1146. """
  1147. """
  1148. fundin = kwargs.get("in_fund")
  1149. fundout = kwargs.get("out_fund")
  1150. init_fund = kwargs.get("init_fund")
  1151. today_fund = kwargs.get("today_fund")
  1152. yesterday_fund = kwargs.get("yesterday_fund")
  1153. fundin = float(fundin) if fundin else 0.0
  1154. fundout = float(fundout) if fundout else 0.0
  1155. init_fund = float(init_fund) if init_fund else 0.0
  1156. today_fund = float(today_fund) if today_fund else 0.0
  1157. yesterday_fund = float(yesterday_fund) if yesterday_fund else 0.0
  1158. #入资金
  1159. if fundin > fundout:
  1160. fundin = fundin - fundout
  1161. today_income = round((today_fund-(yesterday_fund+fundin))/(float(yesterday_fund) + fundin),4)
  1162. yesterday_fund = yesterday_fund + fundin
  1163. total_income = round((today_fund-init_fund-fundin)/(init_fund+fundin),4)
  1164. init_fund = init_fund + fundin
  1165. #出资金
  1166. else:
  1167. fundout = fundout - fundin
  1168. today_income = round((today_fund-(yesterday_fund-fundout))/float(yesterday_fund),4)
  1169. yesterday_fund = round(today_fund/(1+today_income),4)
  1170. total_income = round((today_fund+fundout-init_fund)/init_fund,4)
  1171. init_fund = round(today_fund/(1+total_income),4)
  1172. today_income = "{}%".format(today_income*100)
  1173. total_income = "{}%".format(total_income*100)
  1174. yesterday_fund = round(yesterday_fund,4)
  1175. init_fund = round(init_fund,4)
  1176. return {"yesterday_fund":yesterday_fund,
  1177. "today_income":today_income,
  1178. "total_income":total_income,
  1179. "init_fund":init_fund}
  1180. def baike_detail_top(**kwargs):
  1181. """
  1182. """
  1183. _id = kwargs.get("id")
  1184. action = kwargs.get("action")
  1185. if action == "top":
  1186. cm.BaikeDetail.objects.filter(id=_id).update(istop=1)
  1187. else:
  1188. cm.BaikeDetail.objects.filter(id=_id).update(istop=0)
  1189. def download_top3_imgs(request):
  1190. """
  1191. """
  1192. kwargs = request.json
  1193. match_id = kwargs.get("match_id",0)
  1194. qset = cm.PlayerRecord.get_db_model(match_id).objects.all()
  1195. if kwargs.get("match_id"):
  1196. qset = qset.filter(match_id=kwargs.get("match_id"))
  1197. #if kwargs.get("groupId"):
  1198. # qset = qset.filter(match_group=kwargs.get("groupId"))
  1199. if kwargs.get("stock_date"):
  1200. qset = qset.filter(stock_date=kwargs.get("stock_date"))
  1201. group_ids = list(cm.MatchGroup.objects.filter(match_id=match_id).values_list("id",flat=True))
  1202. imgData = []
  1203. for group_id in group_ids:
  1204. data = list(qset.filter(match_group=group_id).order_by("group_rank").values())
  1205. for item in data[:3]:
  1206. today_stock_img = json.loads(item["today_stock_img"]) if item["today_stock_img"] else []
  1207. ilist = []
  1208. for img in today_stock_img:
  1209. nlist = img.split('?')
  1210. if len(nlist) > 0:
  1211. ilist.append(nlist[0])
  1212. match_group = item["match_group"]
  1213. match_group_name = cm.MatchGroup.objects.filter(id=match_group).first().name
  1214. group_rank = item["group_rank"]
  1215. username = item["username"]
  1216. imgName = u"%s-%s-%s" % (match_group_name,group_rank,username)
  1217. imgData.append({"imgName":imgName,"imgList":ilist})
  1218. #下载截图
  1219. zip_files = []
  1220. for imgInfo in imgData:
  1221. imgName = imgInfo["imgName"]
  1222. imgList = imgInfo["imgList"]
  1223. for i,url in enumerate(imgList):
  1224. tmpFile = "/tmp/%s-%s%s" % (imgName,i+1,os.path.splitext(url)[-1])
  1225. res = requests.get(url)
  1226. if res.status_code == 200:
  1227. with open(tmpFile,"wb+") as f:
  1228. f.write(res.content)
  1229. zip_files.append(tmpFile)
  1230. #生成zip文件
  1231. tmpzip = get_zipfile(zip_files)
  1232. return tmpzip
  1233. #return imgData
  1234. def update_stock_usernum(**kwargs):
  1235. """
  1236. """
  1237. stock_date = kwargs.get("stock_date")
  1238. qset = cm.UserStock.objects.filter(stock_date=stock_date,fund__gt=0)
  1239. qset = qset.values("stock_id","stock_name").annotate(count=Count("stock_id"),total_fund=Sum("fund")).order_by("-count")
  1240. cm.Stock.objects.update(stock_date=None)
  1241. for q in qset:
  1242. stock_id = q["stock_id"]
  1243. count = q["count"]
  1244. cm.Stock.objects.filter(id=stock_id).update(user_num=count,stock_date=stock_date)
  1245. def get_stock_list(**kwargs):
  1246. """
  1247. """
  1248. name = kwargs.get("name")
  1249. stock_date = kwargs.get("stock_date")
  1250. qset = cm.Stock.objects.all()
  1251. if name:
  1252. qset = qset.filter(name__icontains=name)
  1253. if stock_date:
  1254. qset = qset.filter(stock_date=stock_date)
  1255. data = list(qset.order_by("-user_num").values())
  1256. page = int(kwargs.get("page",1))
  1257. page_size = int(kwargs.get("page_size",20))
  1258. if page and page_size:
  1259. total,data = ccf.get_page_list(data,page,page_size)
  1260. return total,data
  1261. else:
  1262. return len(data),data