controls.py 41 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093
  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
  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. import xlrd
  32. import xlwt
  33. from xlutils.copy import copy
  34. from xltpl.writer import BookWriter
  35. from weixin.wzhifuSDK_V3 import refund_order
  36. @ccc.cache_data()
  37. def get_user_info(uid):
  38. user = cm.UserInfo.objects.filter(id=uid).values().first()
  39. if user:
  40. user["style"] = []
  41. if user["zq"]:
  42. user["style"].append(user["zq"])
  43. if user["cw"]:
  44. user["style"].append(user["cw"])
  45. if user["df"]:
  46. user["style"].append(user["df"])
  47. return user
  48. @ccc.cache_data()
  49. def get_match_info(match_id):
  50. match = cm.Match.objects.filter(id=match_id).values().first()
  51. return match
  52. def async(f):
  53. def wrapper(*args, **kwargs):
  54. thr = Thread(target=f, args=args, kwargs=kwargs)
  55. thr.start()
  56. return wrapper
  57. def add_model(cls,**kwargs):
  58. """
  59. """
  60. model_name = re.search(r'.*\.(\w+)View',str(cls.__class__)).groups()[0]
  61. model = getattr(cm,model_name)
  62. if model_name == "UserInfo":
  63. username = kwargs.get("username")
  64. usercode = kwargs.get("usercode")
  65. phone = kwargs.get("phone")
  66. if cm.UserInfo.objects.filter(username=username).exists():
  67. raise ce.TipException(u"用户名重复!")
  68. if cm.UserInfo.objects.filter(usercode=usercode).exists():
  69. raise ce.TipException(u"用户代码重复!")
  70. if cm.UserInfo.objects.filter(phone=phone).exists():
  71. raise ce.TipException(u"手机号重复!")
  72. if model_name == "Match":
  73. name = kwargs.get("name")
  74. if cm.Match.objects.filter(name=name).exists():
  75. raise ce.TipException(u"比赛名称重复!")
  76. if model_name == "MatchGoup":
  77. name = kwargs.get("name")
  78. match_id = kwargs.get("match_id")
  79. if cm.MatchGroup.objects.filter(name=name,match_id=match_id).exists():
  80. raise ce.TipException(u"分组名称重复!")
  81. if model_name == "Player":
  82. user_id = kwargs.get("user_id")
  83. match_id = kwargs.get("match_id")
  84. if cm.Player.objects.filter(user_id=user_id,match_id=match_id).exists():
  85. raise ce.TipException(u"重复添加选手!")
  86. user =cm.UserInfo.objects.filter(id=user_id).first()
  87. username = user.username if user else ""
  88. usercode = user.usercode if user else ""
  89. kwargs["username"] = username
  90. kwargs["usercode"] = usercode
  91. if model_name == "PlayerRecord":
  92. #return None
  93. match_id = kwargs.get("match_id")
  94. stock_date = kwargs.get("stock_date")
  95. today_stock = json.dumps(kwargs.get("today_stock"))
  96. today_stock_img = json.dumps(kwargs.get("today_stock_img"))
  97. player_id = kwargs.get("player_id")
  98. today_fund = float(kwargs.get("today_fund"))
  99. is_markt = int(kwargs.get("is_markt",0))
  100. player = cm.Player.objects.filter(id=player_id).first()
  101. user_id = player.user_id
  102. init_fund = player.fund
  103. user = cm.UserInfo.objects.filter(id=user_id).first()
  104. username = user.username
  105. usercode = user.usercode
  106. match_group = player.match_group
  107. match = cm.Match.objects.filter(id=match_id).first()
  108. if player.match_status < 1:
  109. raise ce.TipException(u"该账号已暂停/退出比赛,如有疑问请联系管理员获取详情信息!")
  110. now_date = datetime.datetime.now().strftime("%Y-%m-%d")
  111. if match.end_time < now_date:
  112. raise ce.TipException(u"该比赛已结束!")
  113. #yesterday_date = (datetime.datetime.now()-datetime.timedelta(days=1)).strftime("%Y-%m-%d")
  114. yesterday = cm.PlayerRecord.get_db_model(match_id).objects.filter(
  115. match_id=match_id,player_id=player_id).order_by("-stock_date").first()
  116. if yesterday:
  117. yesterday_fund = yesterday.today_fund
  118. yesterday_stock = yesterday.today_stock
  119. yesterday_stock_img = yesterday.today_stock_img
  120. yesterday_is_markt = yesterday.is_markt
  121. else:
  122. yesterday_fund = init_fund
  123. yesterday_stock = ""
  124. yesterday_stock_img = ""
  125. yesterday_is_markt = 0
  126. obj,flag = cm.PlayerRecord.get_db_model(match_id).objects.get_or_create(
  127. player_id=player_id,
  128. match_id=match_id,
  129. stock_date=stock_date)
  130. obj.init_fund = init_fund
  131. obj.yesterday_fund = yesterday_fund
  132. obj.yesterday_stock = yesterday_stock
  133. obj.yesterday_stock_img = yesterday_stock_img
  134. obj.today_fund = today_fund
  135. obj.today_stock = today_stock
  136. obj.today_stock_img = today_stock_img
  137. obj.user_id = user_id
  138. obj.username = username
  139. obj.usercode = usercode
  140. obj.match_group = match_group
  141. obj.is_markt = is_markt
  142. obj.yesterday_is_markt = yesterday_is_markt
  143. #计算今日和昨日盈亏
  144. today_income = (today_fund - yesterday_fund)/float(yesterday_fund) if yesterday_fund else 0.0
  145. total_income = (today_fund - init_fund)/float(init_fund) if init_fund else 0.0
  146. obj.today_income = round(today_income,4)
  147. obj.total_income = round(total_income,4)
  148. if not flag:
  149. obj.ctime = datetime.datetime.now()
  150. obj.save()
  151. #更新group_rank
  152. return obj.id
  153. if model_name == "PlayerRecord":
  154. match_id = kwargs.get("match_id")
  155. obj = model.get_db_model(match_id).objects.create(**kwargs)
  156. else:
  157. obj = model.objects.create(**kwargs)
  158. #
  159. if model_name == "Match":
  160. cm.MatchGroup.objects.create(
  161. match_id = obj.id,
  162. name = obj.name+u"游客组",
  163. is_active = 0,
  164. charge = 0
  165. )
  166. return obj.id
  167. def update_model(cls,**kwargs):
  168. """
  169. """
  170. model_name = re.search(r'.*\.(\w+)View',str(cls.__class__)).groups()[0]
  171. model = getattr(cm,model_name)
  172. id = kwargs.pop("id")
  173. if model_name == "UserInfo":
  174. username = kwargs.get("username")
  175. usercode = kwargs.get("usercode")
  176. phone = kwargs.get("phone")
  177. if cm.UserInfo.objects.exclude(id=id).filter(username=username).exists():
  178. raise ce.TipException(u"用户名重复!")
  179. if cm.UserInfo.objects.exclude(id=id).filter(usercode=usercode).exists():
  180. raise ce.TipException(u"用户代码重复!")
  181. if phone and cm.UserInfo.objects.exclude(id=id).filter(phone=phone).exists():
  182. raise ce.TipException(u"手机号重复!")
  183. if model_name == "Player":
  184. user_id = kwargs.get("user_id")
  185. match_id = kwargs.get("match_id")
  186. match_group = kwargs.get("match_group")
  187. if cm.Player.objects.exclude(id=id).filter(user_id=user_id,match_id=match_id).exists():
  188. raise ce.TipException(u"重复添加选手!")
  189. user =cm.UserInfo.objects.filter(id=user_id).first()
  190. username = user.username if user else ""
  191. usercode = user.usercode if user else ""
  192. try:
  193. match_group_name = cm.MatchGroup.objects.filter(id=match_group).first().name
  194. except Exception as e:
  195. match_group_name = ""
  196. kwargs["match_group_name"] = match_group_name
  197. if model_name == "PlayerRecord":
  198. is_markt = int(kwargs.get("is_markt",0))
  199. kwargs["today_stock"] = json.dumps(kwargs.get("today_stock"))
  200. kwargs["today_stock_img"] = json.dumps(kwargs.get("today_stock_img"))
  201. if model_name == "PlayerRecord":
  202. match_id = kwargs.get("match_id")
  203. rst = model.get_db_model(match_id).objects.filter(id=id).update(**kwargs)
  204. else:
  205. rst = model.objects.filter(id=id).update(**kwargs)
  206. if model_name == "PlayerRecord":
  207. match_id = kwargs.get("match_id")
  208. user_id = kwargs.get("user_id")
  209. obj = cm.PlayerRecord.get_db_model(match_id).objects.filter(id=id).first()
  210. today_fund = obj.today_fund
  211. yesterday_fund = obj.yesterday_fund
  212. init_fund = obj.init_fund
  213. today_income = (today_fund - yesterday_fund)/float(yesterday_fund)
  214. total_income = (today_fund - init_fund)/float(init_fund)
  215. obj.today_income = round(today_income,4)
  216. obj.total_income = round(total_income,4)
  217. #
  218. player = cm.Player.objects.filter(user_id=user_id,match_id=match_id).first()
  219. if player:
  220. obj.match_group = player.match_group
  221. obj.username = player.username
  222. obj.usercode = player.usercode
  223. obj.save()
  224. #更新player的init_fund
  225. player_id = player.id
  226. cm.Player.objects.filter(id=player_id).update(fund=init_fund)
  227. ##更新group_rank
  228. update_group_rank(obj.stock_date,obj.match_group,obj.match_id)
  229. if model_name == "UserInfo":
  230. ccc.del_cache("cdata_get_user_info_(%s,)"%id)
  231. return rst
  232. def delete_model(cls,**kwargs):
  233. """
  234. """
  235. model_name = re.search(r'.*\.(\w+)View',str(cls.__class__)).groups()[0]
  236. model = getattr(cm,model_name)
  237. ids = str(kwargs.get("id")).split(",")
  238. if model_name == "WanzhuConsult":
  239. cm.WanzhuConsult.objects.filter(Q(user_id__in=ids)|Q(reply_user_id__in=ids)).delete()
  240. else:
  241. if model_name == "PlayerRecord":
  242. match_id = kwargs.get("match_id")
  243. obj = cm.PlayerRecord.get_db_model(match_id).objects.filter(id__in=ids).first()
  244. ##更新group_rank
  245. update_group_rank(obj.stock_date,obj.match_group,obj.match_id)
  246. rst = model.get_db_model(match_id).objects.filter(id__in=ids).delete()
  247. else:
  248. rst = model.objects.filter(id__in=ids).delete()
  249. if model_name == "UserInfo":
  250. cm.Player.objects.filter(user_id__in=ids).delete()
  251. cm.PlayerRecord.objects.filter(user_id__in=ids).delete()
  252. if model_name == "Player":
  253. cm.PlayerRecord.objects.filter(player_id__in=ids).delete()
  254. if model_name == "Match":
  255. cm.PlayerRecord.objects.filter(match_id__in=ids).delete()
  256. cm.Player.objects.filter(match_id__in=ids).delete()
  257. cm.MatchGroup.objects.filter(match_id__in=ids).delete()
  258. if model_name == "MatchGroup":
  259. cm.PlayerRecord.objects.filter(match_group__in=ids).delete()
  260. cm.Player.objects.filter(match_group__in=ids).delete()
  261. if model_name == "BaikeCategory":
  262. cm.BaikeDetail.objects.filter(category_id__in=ids).delete()
  263. return ids
  264. def get_search_list(cls,**kwargs):
  265. """
  266. """
  267. model_name = re.search(r'.*\.(\w+)SearchView',str(cls.__class__)).groups()[0]
  268. model = getattr(cm,model_name)
  269. qset = model.objects.all().order_by("-id")
  270. if model_name == "Player":
  271. if kwargs.get("usercode"):
  272. qset = qset.filter(Q(usercode__icontains=kwargs.get("usercode"))|Q(username__icontains=kwargs.get("usercode"))|Q(phone__icontains=kwargs.get("usercode")))
  273. data = list(qset.values("id","user_id","username","usercode"))
  274. for item in data:
  275. item["label"] = "%s(%s)" % (item["username"],item["usercode"])
  276. else:
  277. data = []
  278. elif model_name == "Stock":
  279. data = list(qset.values("id","name","code"))
  280. for item in data:
  281. item["label"] = "%s(%s)" % (item["name"],item["code"])
  282. elif model_name == "UserInfo":
  283. data = list(qset.values("id","username","usercode"))
  284. for item in data:
  285. item["label"] = "%s(%s)" % (item["username"],item["usercode"])
  286. else:
  287. if kwargs.get("name"):
  288. qset = qset.filter(name__icontains=kwargs.get("name"))
  289. data = list(qset.values("id","name"))
  290. return data
  291. def get_detail_info(cls,**kwargs):
  292. """
  293. """
  294. model_name = re.search(r'.*\.(\w+)View',str(cls.__class__)).groups()[0]
  295. model = getattr(cm,model_name)
  296. id = kwargs.get("id")
  297. if model_name == "PlayerRecord":
  298. match_id = kwargs.get("match_id")
  299. rst = list(model.get_db_model(match_id).objects.filter(id=id).values())
  300. else:
  301. rst = list(model.objects.filter(id=id).values())
  302. rst = rst[0] if rst else {}
  303. if model_name == "Player":
  304. user = get_user_info(rst["user_id"])
  305. match = get_match_info(rst["match_id"])
  306. rst["username"] = user["username"]
  307. rst["usercode"] = user["usercode"]
  308. rst["match_name"] = match["name"] if match else ""
  309. rst["match_group"] = int(rst["match_group"]) if rst["match_group"] else 0
  310. if model_name == "PlayerRecord":
  311. rst["today_stock"] = json.loads(rst["today_stock"]) if rst["today_stock"] else []
  312. rst["today_stock_img"] = json.loads(rst["today_stock_img"]) if rst["today_stock_img"] else []
  313. rst["yesterday_stock"] = json.loads(rst["yesterday_stock"]) if rst["yesterday_stock"] else []
  314. rst["yesterday_stock_img"] = json.loads(rst["yesterday_stock_img"]) if rst["yesterday_stock_img"] else []
  315. rst["is_markt"] = bool(rst["is_markt"])
  316. return rst
  317. def get_group_income(match_id,match_group,stock_date=None):
  318. """
  319. """
  320. data = []
  321. if stock_date:
  322. today = stock_date
  323. else:
  324. today = datetime.datetime.now().strftime("%Y-%m-%d")
  325. player_ids = list(cm.Player.objects.filter(match_id=match_id,match_group=match_group).values_list("id",flat=True))
  326. for player_id in player_ids:
  327. 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()
  328. if today_record:
  329. data.append(today_record.total_income)
  330. data = list(set(data))
  331. data = sorted(data,key=lambda x:x,reverse=True)
  332. return data
  333. def get_list_info(cls,**kwargs):
  334. """
  335. """
  336. model_name = re.search(r'.*\.(\w+)ListView',str(cls.__class__)).groups()[0]
  337. model = getattr(cm,model_name)
  338. if model_name == "PlayerRecord":
  339. match_id = kwargs.get("match_id",0)
  340. qset = model.get_db_model(match_id).objects.all()
  341. else:
  342. qset = model.objects.all()
  343. if kwargs.get("name"):
  344. qset = qset.filter(name__icontains=kwargs.get("name"))
  345. if model_name == "UserInfo":
  346. if kwargs.get("username"):
  347. qset = qset.filter(Q(username__icontains=kwargs.get("username"))|Q(usercode__icontains=kwargs.get("username"))|Q(phone__icontains=kwargs.get("username")))
  348. if model_name == "Player":
  349. if kwargs.get("username"):
  350. qset = qset.filter(Q(username__icontains=kwargs.get("username"))|Q(usercode__icontains=kwargs.get("username")))
  351. if kwargs.get("match_id"):
  352. qset = qset.filter(match_id=kwargs.get("match_id"))
  353. if kwargs.get("groupId"):
  354. qset = qset.filter(match_group=kwargs.get("groupId"))
  355. if model_name == "MatchGroup":
  356. if kwargs.get("match_id"):
  357. qset = qset.filter(match_id=kwargs.get("match_id"))
  358. if model_name == "PlayerRecord":
  359. if kwargs.get("match_id"):
  360. qset = qset.filter(match_id=kwargs.get("match_id"))
  361. if kwargs.get("groupId"):
  362. qset = qset.filter(match_group=kwargs.get("groupId"))
  363. if kwargs.get("username"):
  364. qset = qset.filter(Q(username__icontains=kwargs.get("username"))|Q(usercode__icontains=kwargs.get("username")))
  365. if kwargs.get("stock_date"):
  366. qset = qset.filter(stock_date=kwargs.get("stock_date"))
  367. if kwargs.get("auto_complete"):
  368. if int(kwargs.get("auto_complete"))==1:
  369. qset = qset.filter(auto_complete__gte=kwargs.get("auto_complete"))
  370. else:
  371. qset = qset.filter(auto_complete=kwargs.get("auto_complete"))
  372. if model_name == "SignupOrder":
  373. if kwargs.get("match_id"):
  374. qset = qset.filter(match_id=kwargs.get("match_id"))
  375. if kwargs.get("username"):
  376. qset = qset.filter(Q(user_name__icontains=kwargs.get("username"))|Q(phone__icontains=kwargs.get("username")))
  377. if kwargs.get("signup_type"):
  378. qset = qset.filter(signup_type=kwargs.get("signup_type"))
  379. if kwargs.get("order_status") or kwargs.get("order_status")==0:
  380. qset = qset.filter(order_status=kwargs.get("order_status"))
  381. if kwargs.get("pay_status") or kwargs.get("pay_status")==0:
  382. qset = qset.filter(pay_status=kwargs.get("pay_status"))
  383. if model_name == "Article":
  384. if kwargs.get("type"):
  385. qset = qset.filter(type=kwargs.get("type"))
  386. if kwargs.get("category"):
  387. qset = qset.filter(category=kwargs.get("category"))
  388. if model_name == "Stock":
  389. if kwargs.get("username"):
  390. qset = qset.filter(name__icontains=kwargs.get("username"))
  391. if model_name == "BaikeDetail":
  392. if kwargs.get("name"):
  393. qset = qset.filter(name__icontains=kwargs.get("name"))
  394. if kwargs.get("status"):
  395. qset = qset.filter(status=kwargs.get("status"))
  396. if kwargs.get("category_id"):
  397. qset = qset.filter(category_id=kwargs.get("category_id"))
  398. if model_name == "PlayerRecord":
  399. if kwargs.get("username"):
  400. data = list(qset.order_by("-stock_date").values())
  401. else:
  402. data = list(qset.order_by("-total_income").values())
  403. else:
  404. data = list(qset.order_by("-id").values())
  405. page = int(kwargs.get("page",1))
  406. page_size = int(kwargs.get("page_size",20))
  407. if page and page_size:
  408. total,data = ccf.get_page_list(data,page,page_size)
  409. if model_name == "Player":
  410. for item in data:
  411. user = get_user_info(item["user_id"])
  412. if user:
  413. match = get_match_info(item["match_id"])
  414. item["nickname"] = user["nickname"]
  415. item["match_name"] = match["name"] if match else ""
  416. item["openid"] = user["openid"]
  417. item["player_type"] = user["player_type"]
  418. item["phone"] = user["phone"]
  419. if model_name == "PlayerRecord":
  420. for item in data:
  421. user = cm.UserInfo.objects.filter(id=item["user_id"]).first()
  422. item["username"] = user.username if user else ""
  423. item["usercode"] = user.usercode if user else ""
  424. today_stock = json.loads(item["today_stock"]) if item["today_stock"] else []
  425. today_stock = filter(lambda x:x["name"],today_stock if today_stock else [])
  426. today_stock_img = json.loads(item["today_stock_img"]) if item["today_stock_img"] else []
  427. item["today_stock"] = today_stock
  428. item["today_stock_img"] = today_stock_img
  429. yesterday_stock = json.loads(item["yesterday_stock"]) if item["yesterday_stock"] else []
  430. yesterday_stock = filter(lambda x:x["name"],yesterday_stock if yesterday_stock else [])
  431. item["yesterday_stock"] = yesterday_stock
  432. item["yesterday_stock_img"] = json.loads(item["yesterday_stock_img"]) if item["yesterday_stock_img"] else []
  433. if item["today_income"]:
  434. item["today_income"] = "{}%".format(item["today_income"]*100)
  435. if item["total_income"]:
  436. item["total_income"] = "{}%".format(item["total_income"]*100)
  437. item["is_markt"] = bool(item["is_markt"])
  438. #资金出入
  439. if cm.FundInOut.objects.filter(user_id=item["user_id"],stock_date=item["stock_date"]).exists():
  440. item["fund_inout"] = cm.FundInOut.objects.filter(user_id=item["user_id"],stock_date=item["stock_date"]).values().first()
  441. if kwargs.get("groupId"):
  442. data = sorted(data,key=lambda x:x["group_rank"])
  443. if model_name == "Match":
  444. for item in data:
  445. groups = ",".join(list(cm.MatchGroup.objects.filter(match_id=item["id"]).values_list("name",flat=True)))
  446. item["join_count"] = cm.Player.objects.filter(match_id=item["id"]).count()
  447. item["out_count"] = cm.Player.objects.filter(match_id=item["id"],match_status=-1).count()
  448. item["groups"] = groups
  449. if model_name == "MatchGroup":
  450. for item in data:
  451. match = cm.Match.objects.filter(id=item["match_id"]).first()
  452. match_name = match.name if match else ""
  453. item["join_count"] = cm.Player.objects.filter(match_group=item["id"]).count()
  454. item["out_count"] = cm.Player.objects.filter(match_group=item["id"],match_status=-1).count()
  455. item["match_name"] = match_name
  456. if model_name == "SignupOrder":
  457. for item in data:
  458. user = cm.UserInfo.objects.filter(id=item["user_id"]).first()
  459. try:
  460. item["match_group_name"] = cm.MatchGroup.objects.filter(id=item["match_group"]).first().name
  461. except Exception as e:
  462. print(e)
  463. item["match_group_name"] = ""
  464. if user:
  465. item["usercode"] = user.usercode
  466. item["nickname"] = user.nickname
  467. item["openid"] = user.openid
  468. if model_name == "FundInOut":
  469. for item in data:
  470. user = cm.UserInfo.objects.filter(id=item["user_id"]).first()
  471. item["username"] = user.username
  472. item["usercode"] = user.usercode
  473. if kwargs.get("username"):
  474. data = filter(lambda x:x["username"] == str(kwargs.get("username")) or kwargs.get("username") ==x["usercode"],data)
  475. return (total,data)
  476. else:
  477. return len(data),data
  478. return rst
  479. def download_records(request):
  480. """
  481. """
  482. kwargs = request.json
  483. header = [u"排名",u"选手",u"初始资产(万)",u"昨日资产(万)",u"今日资产",u"今日盈亏",u"总盈亏",u"昨日持股",u"今日持股"]
  484. match_id = kwargs.get("match_id")
  485. qset = cm.PlayerRecord.get_db_model(match_id).objects.all()
  486. if kwargs.get("match_id"):
  487. qset = qset.filter(match_id=kwargs.get("match_id"))
  488. if kwargs.get("groupId"):
  489. qset = qset.filter(match_group=kwargs.get("groupId"))
  490. if kwargs.get("username"):
  491. qset = qset.filter(Q(username__icontains=kwargs.get("username"))|Q(usercode__icontains=kwargs.get("username")))
  492. if kwargs.get("stock_date"):
  493. qset = qset.filter(stock_date=kwargs.get("stock_date"))
  494. qdata = list(qset.order_by("-total_income").values())
  495. if kwargs.get("groupId"):
  496. qdata = sorted(qdata,key=lambda x:x["group_rank"])
  497. data = []
  498. for item in qdata:
  499. try:
  500. user = cm.UserInfo.objects.filter(id=item["user_id"]).first()
  501. item["username"] = user.username
  502. item["usercode"] = user.usercode
  503. item["today_income"] = "{}%".format(item["today_income"]*100)
  504. item["total_income"] = "{}%".format(item["total_income"]*100)
  505. today_stock = json.loads(item["today_stock"]) if item["today_stock"] else []
  506. today_stock = list(filter(lambda x:x["name"],today_stock if today_stock else []))
  507. today_stock = ",".join([x["name"]+str(x["fund"]) for x in today_stock]) if today_stock else ""
  508. yesterday_stock = json.loads(item["yesterday_stock"]) if item["yesterday_stock"] else []
  509. yesterday_stock = list(filter(lambda x:x["name"],yesterday_stock if yesterday_stock else []))
  510. yesterday_stock = ",".join([x["name"]+str(x["fund"]) for x in yesterday_stock]) if yesterday_stock else ""
  511. #空仓、开超市、请假判断
  512. if item["auto_complete"] > 0:
  513. today_stock = u"请假"
  514. else:
  515. if item["is_markt"] == 1:
  516. today_stock = u"开超市"
  517. else:
  518. if today_stock == "":
  519. today_stock = u"空仓"
  520. #昨日
  521. if item["yesterday_auto_complete"] > 0:
  522. yesterday_stock = u"请假"
  523. else:
  524. if item["yesterday_is_markt"] == 1:
  525. yesterday_stock = u"开超市"
  526. else:
  527. if yesterday_stock == "":
  528. yesterday_stock = u"空仓"
  529. data.append(
  530. [
  531. item["group_rank"],
  532. item["username"],
  533. item["init_fund"],
  534. item["yesterday_fund"],
  535. item["today_fund"],
  536. item["today_income"],
  537. item["total_income"],
  538. yesterday_stock,
  539. today_stock
  540. ])
  541. except Exception as e:
  542. print(e)
  543. xlsfile = os.path.join(settings.BASE_DIR,"demo.xls")
  544. exceltool = ExcelTool(xlsfile)
  545. xlsdata = exceltool.save_data(u"学员报名数据表",header,data)
  546. return xlsdata
  547. def fast_save_player(**kwargs):
  548. """
  549. """
  550. username = kwargs.get("username")
  551. usercode = kwargs.get("usercode")
  552. if cm.UserInfo.objects.filter(username=username).exists():
  553. raise ce.TipException(u"用户名重复!")
  554. if cm.UserInfo.objects.filter(usercode=usercode).exists():
  555. raise ce.TipException(u"用户代码重复!")
  556. user = cm.UserInfo.objects.create(username=username,usercode=usercode)
  557. #新建选手
  558. user_id = user.id
  559. kwargs["user_id"] = user_id
  560. cm.Player.objects.create(**kwargs)
  561. def update_player_latest(record):
  562. """更新选手最后一次数据
  563. """
  564. user_id = record["user_id"]
  565. match_id = record["match_id"]
  566. stock_date = record["stock_date"]
  567. match_group = record["match_group"]
  568. wincnt = record["wincnt"]
  569. total = record["totalcnt"]
  570. key = "PLAYER_LATEST_{}".format(user_id)
  571. if record:
  572. ccc.cache.hset(key,"stock_date",stock_date)
  573. ccc.cache.hset(key,"match_id",match_id)
  574. ccc.cache.hset(key,"match_group",match_group)
  575. #更新胜率
  576. qset = cm.PlayerRecord.get_db_model(match_id).objects.filter(match_id=match_id,user_id=user_id)
  577. #win_rate = qset.filter(today_income__gte=0).count()/float(qset.count()) if qset else 0.0
  578. win_rate = wincnt/float(total) if total else 0.0
  579. win_rate = round(win_rate,3)
  580. ccc.cache.hset(key,"win_rate",win_rate)
  581. #badest = cm.PlayerRecord.get_db_model(match_id).objects.filter(user_id=user_id,match_id=match_id).order_by("today_income").first()
  582. #if badest:
  583. # ccc.cache.hset(key,"badest_income",badest.today_income)
  584. def update_group_rank(stock_date=None,match_id=None,group_id=None):
  585. """
  586. """
  587. if stock_date:
  588. stock_date = stock_date
  589. else:
  590. stock_date = datetime.datetime.now().strftime("%Y-%m-%d")
  591. del_keys = ccc.cache.keys("*_%s_%s_%s" % (match_id,group_id,stock_date))
  592. if del_keys:
  593. ccc.cache.delete(*del_keys)
  594. groups = cm.MatchGroup.objects.filter(id=group_id)
  595. for gp in groups:
  596. match_id = gp.match_id
  597. match_group = gp.id
  598. 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")
  599. if prset:
  600. records = prset.values()
  601. case_id = " case id "
  602. cases = []
  603. where = []
  604. for index,pr in enumerate(records):
  605. case = "WHEN %s THEN %s" % (pr["id"],index+1)
  606. cases.append(case)
  607. where.append(str(pr["id"]))
  608. #
  609. key = "%s_%s_%s_%s" % (pr["user_id"],match_id,match_group,stock_date)
  610. pr.update({"group_rank":index+1})
  611. ccc.pl.set(key,json.dumps(pr,cls=ccc.CusJSONEncoder))
  612. ccc.pl.execute()
  613. winset = cm.PlayerRecord.get_db_model(match_id).objects.filter(
  614. match_id=match_id,match_group=match_group,today_income__gte=0)\
  615. .values("user_id").annotate(wincnt=Count("user_id")).values("user_id","match_id","match_group","wincnt")
  616. totalset = cm.PlayerRecord.get_db_model(match_id).objects.filter(
  617. match_id=match_id,match_group=match_group)\
  618. .values("user_id").annotate(totalcnt=Count("user_id")).values("user_id","totalcnt")
  619. totallist = list(totalset)
  620. for item in winset:
  621. item["stock_date"] = stock_date
  622. user_id = item["user_id"]
  623. total = filter(lambda x:x["user_id"]==user_id,totallist)
  624. if total:
  625. item.update(total[0])
  626. else:
  627. item.update({"totalcnt":0})
  628. if total:
  629. #更新最后一次数据
  630. update_player_latest(item)
  631. if cases and where:
  632. case = case_id + " ".join(cases)
  633. where = ",".join(where)
  634. sql = "update player_record_%s set group_rank = %s ELSE 0 END where id in (%s)" % (match_id,case,where)
  635. cursor = connection.cursor()
  636. cursor.execute(sql)
  637. cursor.close()
  638. ##更新最后一次数据
  639. #update_player_latest(pr)
  640. def update_comment(**kwargs):
  641. """
  642. """
  643. id = kwargs.get("id")
  644. match_id = kwargs.get("match_id")
  645. wanzhu_comment = kwargs.get("wanzhu_comment")
  646. cm.PlayerRecord.get_db_model(match_id).objects.filter(id=id).update(wanzhu_comment=wanzhu_comment)
  647. def reset_initfund(**kwargs):
  648. """
  649. """
  650. match_id = kwargs.get("match_id")
  651. cm.Player.objects.filter(match_id=match_id).update(fund=1)
  652. #cm.Player.objects.filter(match_id=match_id,id=3011).update(fund=None)
  653. def get_stock_comments_list(**kwargs):
  654. """
  655. """
  656. qset = cm.Comments.objects.filter(stock_id__isnull=False).order_by(*["-istop","-id"])
  657. if kwargs.get("name"):
  658. qset = qset.filter(stock_name__icontains=kwargs.get("name"))
  659. data = list(qset.values())
  660. for item in data:
  661. item["stock_name"] = cm.Stock.objects.filter(id=item["stock_id"]).first().name
  662. #分页
  663. page = int(kwargs.get("page",1))
  664. page_size = int(kwargs.get("page_size",20))
  665. if page and page_size:
  666. total,data = ccf.get_page_list(data,page,page_size)
  667. else:
  668. total = len(data)
  669. return total,data
  670. def get_record_comments_list(**kwargs):
  671. """
  672. """
  673. match_id = kwargs.get("match_id")
  674. qset = cm.Comments.objects.filter(record_id__isnull=False).order_by(*["-istop","-id"])
  675. if kwargs.get("user_name"):
  676. qset = qset.filter(user_name__icontains=kwargs.get("name"))
  677. data = list(qset.values())
  678. for item in data:
  679. item["record_info"] = cm.PlayerRecord.get_db_model(match_id).objects.filter(id=item["record_id"]).values().first()
  680. #分页
  681. page = int(kwargs.get("page",1))
  682. page_size = int(kwargs.get("page_size",20))
  683. if page and page_size:
  684. total,data = ccf.get_page_list(data,page,page_size)
  685. else:
  686. total = len(data)
  687. return total,data
  688. def get_wanzhu_consult_list(**kwargs):
  689. """
  690. """
  691. qset = cm.Consult.objects.all().order_by("-ctime")
  692. if kwargs.get("username"):
  693. qset = qset.filter(user_name__icontains=kwargs.get("username"))
  694. if kwargs.get("reply_status"):
  695. qset = qset.filter(reply_status=kwargs.get("reply_status"))
  696. if kwargs.get("content"):
  697. qset = qset.filter(Q(content__icontains=kwargs.get("content"))|Q(reply_content__icontains=kwargs.get("content")))
  698. data = list(qset.values())
  699. for item in data:
  700. user_id = item["user_id"]
  701. userinfo = get_user_info(user_id)
  702. if userinfo:
  703. item["user_code"] = userinfo.get("usercode")
  704. item["phone"] = userinfo.get("phone")
  705. #分页
  706. page = int(kwargs.get("page",1))
  707. page_size = int(kwargs.get("page_size",20))
  708. if page and page_size:
  709. total,data = ccf.get_page_list(data,page,page_size)
  710. else:
  711. total = len(data)
  712. return total,data
  713. def calc_reply(v):
  714. """计算未回复数量
  715. """
  716. user_ids = [str(x["user_id"])[0] for x in v]
  717. user_id_str = "".join(user_ids)
  718. if v[-1]["user_id"]<=0:
  719. return 0
  720. else:
  721. index = user_id_str.rfind("0")
  722. return len(v) - index-1
  723. def reply_wanzhu_consult(**kwargs):
  724. """
  725. """
  726. pid = kwargs.get("pid")
  727. content = kwargs.get("reply_content")
  728. user_id = kwargs.get("user_id")
  729. action = kwargs.get("action")
  730. if action == "send":
  731. user = cm.UserInfo.objects.filter(id=user_id).first()
  732. user_name = user.username
  733. user_avatar = user.avatar
  734. obj,flag = cm.Consult.objects.get_or_create(
  735. user_id = user_id,
  736. )
  737. obj.user_name = user_name
  738. obj.user_avatar = user_avatar
  739. #obj.reply_content = content
  740. obj.view_status = 1
  741. obj.reply_status = 0
  742. obj.ctime = datetime.datetime.now()
  743. obj.save()
  744. wanzhu_obj = cm.WanzhuConsult.objects.create(
  745. user_id = user_id,
  746. user_name = user_name,
  747. user_avatar = user_avatar
  748. )
  749. wanzhu_obj_1 = cm.WanzhuConsult.objects.create(
  750. reply_content = content,
  751. user_id = 0,
  752. player_id = 0,
  753. reply_user_id = user_id,
  754. pid = wanzhu_obj.id
  755. )
  756. cm.Consult.objects.filter(user_id=user_id).update(reply_status=1,reply_content=content,ctime=datetime.datetime.now())
  757. else:
  758. cm.WanzhuConsult.objects.create(
  759. pid = pid,
  760. reply_content = content,
  761. user_id = 0,
  762. player_id = 0,
  763. reply_user_id = user_id
  764. )
  765. cm.Consult.objects.filter(user_id=user_id).update(reply_status=1,reply_content=content,ctime=datetime.datetime.now())
  766. #发送消息提醒
  767. if cm.UserInfo.objects.filter(id=user_id).exists():
  768. user = cm.UserInfo.objects.filter(id=user_id).first()
  769. openid = user.openid
  770. username = user.username
  771. try:
  772. send_consult_reply_message(openid,username)
  773. except Exception as e:
  774. print(e)
  775. #发送app推送消息
  776. title = u"顽主杯"
  777. body = u"你收到一条顽主的消息"
  778. params = {"type":"message","id":str(user_id)}
  779. aliyunpush.push_notice_by_userid(user_id,title,body,params,"ios")
  780. aliyunpush.push_notice_by_userid(user_id,title,body,params,"android")
  781. def update_player_type(**kwargs):
  782. """
  783. """
  784. player_id = kwargs.get("player_id")
  785. user_id = kwargs.get("user_id")
  786. player_type = kwargs.get("player_type")
  787. phone = kwargs.get("phone")
  788. cm.UserInfo.objects.filter(id=user_id).update(player_type=player_type,phone=phone)
  789. def update_player_role(**kwargs):
  790. """
  791. """
  792. player_id = kwargs.get("player_id")
  793. user_id = kwargs.get("user_id")
  794. role = kwargs.get("role")
  795. cm.Player.objects.filter(id=player_id).update(role=role)
  796. def comments_top(**kwargs):
  797. """
  798. """
  799. _id = kwargs.get("id")
  800. istop = 1-cm.Comments.objects.filter(id=_id).first().istop
  801. cm.Comments.objects.filter(id=_id).update(istop=istop)
  802. def comments_black(**kwargs):
  803. """
  804. """
  805. _id = kwargs.get("id")
  806. comment_status = 1-cm.UserInfo.objects.filter(id=_id).first().comment_status
  807. cm.UserInfo.objects.filter(id=_id).update(comment_status=comment_status)
  808. def relation_old_user(**kwargs):
  809. """
  810. """
  811. old_user_id = kwargs.get("old_user_id")
  812. user_id = kwargs.get("user_id")
  813. print(kwargs,1111111111111)
  814. user = cm.UserInfo.objects.filter(id=user_id).first()
  815. if user:
  816. if user.openid:
  817. cm.UserInfo.objects.filter(id=old_user_id).update(openid=user.openid)
  818. cm.UserInfo.objects.filter(id=user_id).delete()
  819. def get_openid_miss(**kwargs):
  820. """
  821. """
  822. qset = cm.UserInfo.objects.filter(openid__isnull=True)
  823. data = list(qset.values())
  824. return data
  825. def signup_old_user(**kwargs):
  826. """
  827. """
  828. order_id = kwargs.get("id")
  829. signup_type = kwargs.get("signup_type")
  830. match_group = kwargs.get("match_group")
  831. player_type = kwargs.get("player_type")
  832. role_type = kwargs.get("role_type")
  833. order = cm.SignupOrder.objects.filter(id=order_id).first()
  834. if order:
  835. user_id = order.user_id
  836. match_id = order.match_id
  837. signup_name = order.signup_name
  838. user = cm.UserInfo.objects.filter(id=user_id).first()
  839. match_group_name = cm.MatchGroup.objects.filter(id=match_group).first().name
  840. match = cm.Match.objects.filter(id=match_id).first()
  841. user.player_type = player_type
  842. user.phone = order.phone
  843. user.username = signup_name
  844. user.save()
  845. order.signup_type = signup_type
  846. order.match_group = match_group
  847. order.match_group_name = match_group_name
  848. order.order_status = 1
  849. order.save()
  850. #生产选手信息
  851. player,flag = cm.Player.objects.get_or_create(
  852. user_id = user_id,
  853. match_id = match_id
  854. )
  855. player.match_name = match.name
  856. player.match_group = match_group
  857. player.match_group_name = match_group_name
  858. player.username = signup_name
  859. player.usercode = user.usercode
  860. player.role = role_type
  861. player.match_status = 1
  862. player.save()
  863. def get_consult_detail(**kwargs):
  864. """
  865. """
  866. rst = {}
  867. user_id = kwargs.get("user_id")
  868. qset = cm.WanzhuConsult.objects.filter(Q(user_id=user_id)|Q(reply_user_id=user_id))
  869. rst["consult_list"] = list(qset.filter(user_id__gt=0).order_by("ctime").values())
  870. rst["all_consult_list"] = list(qset.order_by("ctime").values())
  871. rst["user_id"] = user_id
  872. cm.Consult.objects.filter(user_id=user_id).update(view_status=1)
  873. return rst
  874. def delete_consult(**kwargs):
  875. """
  876. """
  877. user_id = kwargs.get("id")
  878. cm.Consult.objects.filter(user_id=user_id).delete()
  879. cm.WanzhuConsult.objects.filter(Q(user_id=user_id)|Q(reply_user_id=user_id)).delete()
  880. def batch_update_player_latest(record):
  881. """更新选手最后一次数据
  882. """
  883. user_id = record["user_id"]
  884. match_id = record["match_id"]
  885. stock_date = record["stock_date"]
  886. match_group = record["match_group"]
  887. #wincnt = record["wincnt"]
  888. #total = record["totalcnt"]
  889. key = "PLAYER_LATEST_{}".format(user_id)
  890. if record and match_group:
  891. ccc.cache.hset(key,"stock_date",stock_date)
  892. ccc.cache.hset(key,"match_id",match_id)
  893. ccc.cache.hset(key,"match_group",match_group)
  894. ##更新胜率
  895. #qset = cm.PlayerRecord.get_db_model(match_id).objects.filter(match_id=match_id,user_id=user_id)
  896. ##win_rate = qset.filter(today_income__gte=0).count()/float(qset.count()) if qset else 0.0
  897. #win_rate = wincnt/float(total) if total else 0.0
  898. #win_rate = round(win_rate,3)
  899. #ccc.cache.hset(key,"win_rate",win_rate)
  900. def get_sub_category(pid):
  901. """
  902. """
  903. qset = cm.BaikeCategory.objects.filter(parent_id=pid)
  904. qdata = list(qset.values())
  905. if qdata:
  906. for subitem in qdata:
  907. subitem["type"] = "ent"
  908. subitem["label"] = subitem["name"]
  909. subitem["children"] = get_sub_category(subitem["id"])
  910. return qdata
  911. def get_baike_category_tree(**kwargs):
  912. parents = list(cm.BaikeCategory.objects.filter(parent_id__isnull=True).values())
  913. for item in parents:
  914. item["label"] = item["name"]
  915. item["children"] = get_sub_category(item["id"])
  916. return parents
  917. def get_baike_list_by_category(**kwargs):
  918. category_id = kwargs.get("category_id")
  919. qset = cm.BaikeDetail.objects.filter(category_id=category_id)
  920. data = list(qset.values("id","name"))
  921. return data
  922. def push_article(**kwargs):
  923. _id = kwargs.get("id")
  924. article = cm.Article.objects.filter(id=_id).first()
  925. #发送app推送消息
  926. title = u"顽主杯"
  927. body = article.name
  928. user_id = "ALL"
  929. params = {"type":"article","id":_id}
  930. aliyunpush.push_notice_by_userid(user_id,title,body,params,"ios")
  931. aliyunpush.push_notice_by_userid(user_id,title,body,params,"android")
  932. def import_article(**kwargs):
  933. """
  934. """
  935. obj = cm.Article.objects.create(**kwargs)
  936. def refund_out_trade_order(**kwargs):
  937. """
  938. """
  939. _id = kwargs.get("id")
  940. signup_order = cm.SignupOrder.objects.filter(id=_id).first()
  941. if signup_order:
  942. out_trade_no = signup_order.out_trade_no
  943. transaction_id = signup_order.transaction_id
  944. total_fee = signup_order.total_fee
  945. amount = str(int(total_fee*100))
  946. print(amount)
  947. result = refund_order(transaction_id,out_trade_no,out_trade_no,amount)
  948. if result.get("result_code") == "SUCCESS":
  949. signup_order.order_status = -1
  950. signup_order.save()
  951. return "success"
  952. else:
  953. raise ce.TipException(result.get("err_code_des"))
  954. else:
  955. raise ce.TipException(u"该订单不存在退款失败!")
  956. #test