controls.py 41 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084
  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. data = list(qset.order_by("-total_income").values())
  400. else:
  401. data = list(qset.order_by("-id").values())
  402. page = int(kwargs.get("page",1))
  403. page_size = int(kwargs.get("page_size",20))
  404. if page and page_size:
  405. total,data = ccf.get_page_list(data,page,page_size)
  406. if model_name == "Player":
  407. for item in data:
  408. user = get_user_info(item["user_id"])
  409. if user:
  410. match = get_match_info(item["match_id"])
  411. item["nickname"] = user["nickname"]
  412. item["match_name"] = match["name"] if match else ""
  413. item["openid"] = user["openid"]
  414. item["player_type"] = user["player_type"]
  415. item["phone"] = user["phone"]
  416. if model_name == "PlayerRecord":
  417. for item in data:
  418. user = cm.UserInfo.objects.filter(id=item["user_id"]).first()
  419. item["username"] = user.username if user else ""
  420. item["usercode"] = user.usercode if user else ""
  421. today_stock = json.loads(item["today_stock"]) if item["today_stock"] else []
  422. today_stock = filter(lambda x:x["name"],today_stock if today_stock else [])
  423. today_stock_img = json.loads(item["today_stock_img"]) if item["today_stock_img"] else []
  424. item["today_stock"] = today_stock
  425. item["today_stock_img"] = today_stock_img
  426. yesterday_stock = json.loads(item["yesterday_stock"]) if item["yesterday_stock"] else []
  427. yesterday_stock = filter(lambda x:x["name"],yesterday_stock if yesterday_stock else [])
  428. item["yesterday_stock"] = yesterday_stock
  429. item["yesterday_stock_img"] = json.loads(item["yesterday_stock_img"]) if item["yesterday_stock_img"] else []
  430. if item["today_income"]:
  431. item["today_income"] = "{}%".format(item["today_income"]*100)
  432. if item["total_income"]:
  433. item["total_income"] = "{}%".format(item["total_income"]*100)
  434. item["is_markt"] = bool(item["is_markt"])
  435. #资金出入
  436. if cm.FundInOut.objects.filter(user_id=item["user_id"],stock_date=item["stock_date"]).exists():
  437. item["fund_inout"] = cm.FundInOut.objects.filter(user_id=item["user_id"],stock_date=item["stock_date"]).values().first()
  438. if kwargs.get("groupId"):
  439. data = sorted(data,key=lambda x:x["group_rank"])
  440. if model_name == "Match":
  441. for item in data:
  442. groups = ",".join(list(cm.MatchGroup.objects.filter(match_id=item["id"]).values_list("name",flat=True)))
  443. item["join_count"] = cm.Player.objects.filter(match_id=item["id"]).count()
  444. item["out_count"] = cm.Player.objects.filter(match_id=item["id"],match_status=-1).count()
  445. item["groups"] = groups
  446. if model_name == "MatchGroup":
  447. for item in data:
  448. match = cm.Match.objects.filter(id=item["match_id"]).first()
  449. match_name = match.name if match else ""
  450. item["join_count"] = cm.Player.objects.filter(match_group=item["id"]).count()
  451. item["out_count"] = cm.Player.objects.filter(match_group=item["id"],match_status=-1).count()
  452. item["match_name"] = match_name
  453. if model_name == "SignupOrder":
  454. for item in data:
  455. user = cm.UserInfo.objects.filter(id=item["user_id"]).first()
  456. try:
  457. item["match_group_name"] = cm.MatchGroup.objects.filter(id=item["match_group"]).first().name
  458. except Exception as e:
  459. print(e)
  460. item["match_group_name"] = ""
  461. if user:
  462. item["usercode"] = user.usercode
  463. item["nickname"] = user.nickname
  464. item["openid"] = user.openid
  465. if model_name == "FundInOut":
  466. for item in data:
  467. user = cm.UserInfo.objects.filter(id=item["user_id"]).first()
  468. item["username"] = user.username
  469. item["usercode"] = user.usercode
  470. if kwargs.get("username"):
  471. data = filter(lambda x:x["username"] == str(kwargs.get("username")) or kwargs.get("username") ==x["usercode"],data)
  472. return (total,data)
  473. else:
  474. return len(data),data
  475. return rst
  476. def download_records(request):
  477. """
  478. """
  479. kwargs = request.json
  480. header = [u"排名",u"选手",u"初始资产(万)",u"昨日资产(万)",u"今日资产",u"今日盈亏",u"总盈亏",u"昨日持股",u"今日持股"]
  481. match_id = kwargs.get("match_id")
  482. qset = cm.PlayerRecord.get_db_model(match_id).objects.all()
  483. if kwargs.get("match_id"):
  484. qset = qset.filter(match_id=kwargs.get("match_id"))
  485. if kwargs.get("groupId"):
  486. qset = qset.filter(match_group=kwargs.get("groupId"))
  487. if kwargs.get("username"):
  488. qset = qset.filter(Q(username__icontains=kwargs.get("username"))|Q(usercode__icontains=kwargs.get("username")))
  489. if kwargs.get("stock_date"):
  490. qset = qset.filter(stock_date=kwargs.get("stock_date"))
  491. qdata = list(qset.order_by("-total_income").values())
  492. if kwargs.get("groupId"):
  493. qdata = sorted(qdata,key=lambda x:x["group_rank"])
  494. data = []
  495. for item in qdata:
  496. user = cm.UserInfo.objects.filter(id=item["user_id"]).first()
  497. item["username"] = user.username
  498. item["usercode"] = user.usercode
  499. item["today_income"] = "{}%".format(item["today_income"]*100)
  500. item["total_income"] = "{}%".format(item["total_income"]*100)
  501. today_stock = json.loads(item["today_stock"]) if item["today_stock"] else []
  502. today_stock = list(filter(lambda x:x["name"],today_stock if today_stock else []))
  503. today_stock = ",".join([x["name"]+str(x["fund"]) for x in today_stock]) if today_stock else ""
  504. yesterday_stock = json.loads(item["yesterday_stock"]) if item["yesterday_stock"] else []
  505. yesterday_stock = list(filter(lambda x:x["name"],yesterday_stock if yesterday_stock else []))
  506. yesterday_stock = ",".join([x["name"]+str(x["fund"]) for x in yesterday_stock]) if yesterday_stock else ""
  507. #空仓、开超市、请假判断
  508. if item["auto_complete"] > 0:
  509. today_stock = u"请假"
  510. else:
  511. if item["is_markt"] == 1:
  512. today_stock = u"开超市"
  513. else:
  514. if today_stock == "":
  515. today_stock = u"空仓"
  516. #昨日
  517. if item["yesterday_auto_complete"] > 0:
  518. yesterday_stock = u"请假"
  519. else:
  520. if item["yesterday_is_markt"] == 1:
  521. yesterday_stock = u"开超市"
  522. else:
  523. if yesterday_stock == "":
  524. yesterday_stock = u"空仓"
  525. data.append(
  526. [
  527. item["group_rank"],
  528. item["username"],
  529. item["init_fund"],
  530. item["yesterday_fund"],
  531. item["today_fund"],
  532. item["today_income"],
  533. item["total_income"],
  534. yesterday_stock,
  535. today_stock
  536. ])
  537. xlsfile = os.path.join(settings.BASE_DIR,"demo.xls")
  538. exceltool = ExcelTool(xlsfile)
  539. xlsdata = exceltool.save_data(u"学员报名数据表",header,data)
  540. return xlsdata
  541. def fast_save_player(**kwargs):
  542. """
  543. """
  544. username = kwargs.get("username")
  545. usercode = kwargs.get("usercode")
  546. if cm.UserInfo.objects.filter(username=username).exists():
  547. raise ce.TipException(u"用户名重复!")
  548. if cm.UserInfo.objects.filter(usercode=usercode).exists():
  549. raise ce.TipException(u"用户代码重复!")
  550. user = cm.UserInfo.objects.create(username=username,usercode=usercode)
  551. #新建选手
  552. user_id = user.id
  553. kwargs["user_id"] = user_id
  554. cm.Player.objects.create(**kwargs)
  555. def update_player_latest(record):
  556. """更新选手最后一次数据
  557. """
  558. user_id = record["user_id"]
  559. match_id = record["match_id"]
  560. stock_date = record["stock_date"]
  561. match_group = record["match_group"]
  562. wincnt = record["wincnt"]
  563. total = record["totalcnt"]
  564. key = "PLAYER_LATEST_{}".format(user_id)
  565. if record:
  566. ccc.cache.hset(key,"stock_date",stock_date)
  567. ccc.cache.hset(key,"match_id",match_id)
  568. ccc.cache.hset(key,"match_group",match_group)
  569. #更新胜率
  570. qset = cm.PlayerRecord.get_db_model(match_id).objects.filter(match_id=match_id,user_id=user_id)
  571. #win_rate = qset.filter(today_income__gte=0).count()/float(qset.count()) if qset else 0.0
  572. win_rate = wincnt/float(total) if total else 0.0
  573. win_rate = round(win_rate,3)
  574. ccc.cache.hset(key,"win_rate",win_rate)
  575. #badest = cm.PlayerRecord.get_db_model(match_id).objects.filter(user_id=user_id,match_id=match_id).order_by("today_income").first()
  576. #if badest:
  577. # ccc.cache.hset(key,"badest_income",badest.today_income)
  578. def update_group_rank(stock_date=None,match_id=None,group_id=None):
  579. """
  580. """
  581. if stock_date:
  582. stock_date = stock_date
  583. else:
  584. stock_date = datetime.datetime.now().strftime("%Y-%m-%d")
  585. del_keys = ccc.cache.keys("*_%s_%s_%s" % (match_id,group_id,stock_date))
  586. if del_keys:
  587. ccc.cache.delete(*del_keys)
  588. groups = cm.MatchGroup.objects.filter(id=group_id)
  589. for gp in groups:
  590. match_id = gp.match_id
  591. match_group = gp.id
  592. 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")
  593. if prset:
  594. records = prset.values()
  595. case_id = " case id "
  596. cases = []
  597. where = []
  598. for index,pr in enumerate(records):
  599. case = "WHEN %s THEN %s" % (pr["id"],index+1)
  600. cases.append(case)
  601. where.append(str(pr["id"]))
  602. #
  603. key = "%s_%s_%s_%s" % (pr["user_id"],match_id,match_group,stock_date)
  604. pr.update({"group_rank":index+1})
  605. ccc.pl.set(key,json.dumps(pr,cls=ccc.CusJSONEncoder))
  606. ccc.pl.execute()
  607. winset = cm.PlayerRecord.get_db_model(match_id).objects.filter(
  608. match_id=match_id,match_group=match_group,today_income__gte=0)\
  609. .values("user_id").annotate(wincnt=Count("user_id")).values("user_id","match_id","match_group","wincnt")
  610. totalset = cm.PlayerRecord.get_db_model(match_id).objects.filter(
  611. match_id=match_id,match_group=match_group)\
  612. .values("user_id").annotate(totalcnt=Count("user_id")).values("user_id","totalcnt")
  613. totallist = list(totalset)
  614. for item in winset:
  615. item["stock_date"] = stock_date
  616. user_id = item["user_id"]
  617. total = filter(lambda x:x["user_id"]==user_id,totallist)
  618. if total:
  619. item.update(total[0])
  620. else:
  621. item.update({"totalcnt":0})
  622. if total:
  623. #更新最后一次数据
  624. update_player_latest(item)
  625. if cases and where:
  626. case = case_id + " ".join(cases)
  627. where = ",".join(where)
  628. sql = "update player_record_%s set group_rank = %s ELSE 0 END where id in (%s)" % (match_id,case,where)
  629. cursor = connection.cursor()
  630. cursor.execute(sql)
  631. cursor.close()
  632. ##更新最后一次数据
  633. #update_player_latest(pr)
  634. def update_comment(**kwargs):
  635. """
  636. """
  637. id = kwargs.get("id")
  638. match_id = kwargs.get("match_id")
  639. wanzhu_comment = kwargs.get("wanzhu_comment")
  640. cm.PlayerRecord.get_db_model(match_id).objects.filter(id=id).update(wanzhu_comment=wanzhu_comment)
  641. def reset_initfund(**kwargs):
  642. """
  643. """
  644. match_id = kwargs.get("match_id")
  645. cm.Player.objects.filter(match_id=match_id).update(fund=1)
  646. #cm.Player.objects.filter(match_id=match_id,id=3011).update(fund=None)
  647. def get_stock_comments_list(**kwargs):
  648. """
  649. """
  650. qset = cm.Comments.objects.filter(stock_id__isnull=False).order_by(*["-istop","-id"])
  651. if kwargs.get("name"):
  652. qset = qset.filter(stock_name__icontains=kwargs.get("name"))
  653. data = list(qset.values())
  654. for item in data:
  655. item["stock_name"] = cm.Stock.objects.filter(id=item["stock_id"]).first().name
  656. #分页
  657. page = int(kwargs.get("page",1))
  658. page_size = int(kwargs.get("page_size",20))
  659. if page and page_size:
  660. total,data = ccf.get_page_list(data,page,page_size)
  661. else:
  662. total = len(data)
  663. return total,data
  664. def get_record_comments_list(**kwargs):
  665. """
  666. """
  667. match_id = kwargs.get("match_id")
  668. qset = cm.Comments.objects.filter(record_id__isnull=False).order_by(*["-istop","-id"])
  669. if kwargs.get("user_name"):
  670. qset = qset.filter(user_name__icontains=kwargs.get("name"))
  671. data = list(qset.values())
  672. for item in data:
  673. item["record_info"] = cm.PlayerRecord.get_db_model(match_id).objects.filter(id=item["record_id"]).values().first()
  674. #分页
  675. page = int(kwargs.get("page",1))
  676. page_size = int(kwargs.get("page_size",20))
  677. if page and page_size:
  678. total,data = ccf.get_page_list(data,page,page_size)
  679. else:
  680. total = len(data)
  681. return total,data
  682. def get_wanzhu_consult_list(**kwargs):
  683. """
  684. """
  685. qset = cm.Consult.objects.all().order_by("-ctime")
  686. if kwargs.get("username"):
  687. qset = qset.filter(user_name__icontains=kwargs.get("username"))
  688. if kwargs.get("reply_status"):
  689. qset = qset.filter(reply_status=kwargs.get("reply_status"))
  690. if kwargs.get("content"):
  691. qset = qset.filter(Q(content__icontains=kwargs.get("content"))|Q(reply_content__icontains=kwargs.get("content")))
  692. data = list(qset.values())
  693. for item in data:
  694. user_id = item["user_id"]
  695. userinfo = get_user_info(user_id)
  696. if userinfo:
  697. item["user_code"] = userinfo.get("usercode")
  698. #分页
  699. page = int(kwargs.get("page",1))
  700. page_size = int(kwargs.get("page_size",20))
  701. if page and page_size:
  702. total,data = ccf.get_page_list(data,page,page_size)
  703. else:
  704. total = len(data)
  705. return total,data
  706. def calc_reply(v):
  707. """计算未回复数量
  708. """
  709. user_ids = [str(x["user_id"])[0] for x in v]
  710. user_id_str = "".join(user_ids)
  711. if v[-1]["user_id"]<=0:
  712. return 0
  713. else:
  714. index = user_id_str.rfind("0")
  715. return len(v) - index-1
  716. def reply_wanzhu_consult(**kwargs):
  717. """
  718. """
  719. pid = kwargs.get("pid")
  720. content = kwargs.get("reply_content")
  721. user_id = kwargs.get("user_id")
  722. action = kwargs.get("action")
  723. if action == "send":
  724. user = cm.UserInfo.objects.filter(id=user_id).first()
  725. user_name = user.username
  726. user_avatar = user.avatar
  727. obj,flag = cm.Consult.objects.get_or_create(
  728. user_id = user_id,
  729. )
  730. obj.user_name = user_name
  731. obj.user_avatar = user_avatar
  732. #obj.reply_content = content
  733. obj.view_status = 1
  734. obj.reply_status = 0
  735. obj.ctime = datetime.datetime.now()
  736. obj.save()
  737. wanzhu_obj = cm.WanzhuConsult.objects.create(
  738. user_id = user_id,
  739. user_name = user_name,
  740. user_avatar = user_avatar
  741. )
  742. wanzhu_obj_1 = cm.WanzhuConsult.objects.create(
  743. reply_content = content,
  744. user_id = 0,
  745. player_id = 0,
  746. reply_user_id = user_id,
  747. pid = wanzhu_obj.id
  748. )
  749. cm.Consult.objects.filter(user_id=user_id).update(reply_status=1,reply_content=content,ctime=datetime.datetime.now())
  750. else:
  751. cm.WanzhuConsult.objects.create(
  752. pid = pid,
  753. reply_content = content,
  754. user_id = 0,
  755. player_id = 0,
  756. reply_user_id = user_id
  757. )
  758. cm.Consult.objects.filter(user_id=user_id).update(reply_status=1,reply_content=content,ctime=datetime.datetime.now())
  759. #发送消息提醒
  760. if cm.UserInfo.objects.filter(id=user_id).exists():
  761. user = cm.UserInfo.objects.filter(id=user_id).first()
  762. openid = user.openid
  763. username = user.username
  764. try:
  765. send_consult_reply_message(openid,username)
  766. except Exception as e:
  767. print(e)
  768. #发送app推送消息
  769. title = u"顽主杯"
  770. body = u"你收到一条顽主的消息"
  771. params = {"type":"message","id":str(user_id)}
  772. aliyunpush.push_notice_by_userid(user_id,title,body,params,"ios")
  773. aliyunpush.push_notice_by_userid(user_id,title,body,params,"android")
  774. def update_player_type(**kwargs):
  775. """
  776. """
  777. player_id = kwargs.get("player_id")
  778. user_id = kwargs.get("user_id")
  779. player_type = kwargs.get("player_type")
  780. phone = kwargs.get("phone")
  781. cm.UserInfo.objects.filter(id=user_id).update(player_type=player_type,phone=phone)
  782. def update_player_role(**kwargs):
  783. """
  784. """
  785. player_id = kwargs.get("player_id")
  786. user_id = kwargs.get("user_id")
  787. role = kwargs.get("role")
  788. cm.Player.objects.filter(id=player_id).update(role=role)
  789. def comments_top(**kwargs):
  790. """
  791. """
  792. _id = kwargs.get("id")
  793. istop = 1-cm.Comments.objects.filter(id=_id).first().istop
  794. cm.Comments.objects.filter(id=_id).update(istop=istop)
  795. def comments_black(**kwargs):
  796. """
  797. """
  798. _id = kwargs.get("id")
  799. comment_status = 1-cm.UserInfo.objects.filter(id=_id).first().comment_status
  800. cm.UserInfo.objects.filter(id=_id).update(comment_status=comment_status)
  801. def relation_old_user(**kwargs):
  802. """
  803. """
  804. old_user_id = kwargs.get("old_user_id")
  805. user_id = kwargs.get("user_id")
  806. print(kwargs,1111111111111)
  807. user = cm.UserInfo.objects.filter(id=user_id).first()
  808. if user:
  809. if user.openid:
  810. cm.UserInfo.objects.filter(id=old_user_id).update(openid=user.openid)
  811. cm.UserInfo.objects.filter(id=user_id).delete()
  812. def get_openid_miss(**kwargs):
  813. """
  814. """
  815. qset = cm.UserInfo.objects.filter(openid__isnull=True)
  816. data = list(qset.values())
  817. return data
  818. def signup_old_user(**kwargs):
  819. """
  820. """
  821. order_id = kwargs.get("id")
  822. signup_type = kwargs.get("signup_type")
  823. match_group = kwargs.get("match_group")
  824. player_type = kwargs.get("player_type")
  825. role_type = kwargs.get("role_type")
  826. order = cm.SignupOrder.objects.filter(id=order_id).first()
  827. if order:
  828. user_id = order.user_id
  829. match_id = order.match_id
  830. signup_name = order.signup_name
  831. user = cm.UserInfo.objects.filter(id=user_id).first()
  832. match_group_name = cm.MatchGroup.objects.filter(id=match_group).first().name
  833. match = cm.Match.objects.filter(id=match_id).first()
  834. user.player_type = player_type
  835. user.phone = order.phone
  836. user.username = signup_name
  837. user.save()
  838. order.signup_type = signup_type
  839. order.match_group = match_group
  840. order.match_group_name = match_group_name
  841. order.order_status = 1
  842. order.save()
  843. #生产选手信息
  844. player,flag = cm.Player.objects.get_or_create(
  845. user_id = user_id,
  846. match_id = match_id
  847. )
  848. player.match_name = match.name
  849. player.match_group = match_group
  850. player.match_group_name = match_group_name
  851. player.username = signup_name
  852. player.usercode = user.usercode
  853. player.role = role_type
  854. player.match_status = 1
  855. player.save()
  856. def get_consult_detail(**kwargs):
  857. """
  858. """
  859. rst = {}
  860. user_id = kwargs.get("user_id")
  861. qset = cm.WanzhuConsult.objects.filter(Q(user_id=user_id)|Q(reply_user_id=user_id))
  862. rst["consult_list"] = list(qset.filter(user_id__gt=0).order_by("ctime").values())
  863. rst["all_consult_list"] = list(qset.order_by("ctime").values())
  864. rst["user_id"] = user_id
  865. cm.Consult.objects.filter(user_id=user_id).update(view_status=1)
  866. return rst
  867. def delete_consult(**kwargs):
  868. """
  869. """
  870. user_id = kwargs.get("id")
  871. cm.Consult.objects.filter(user_id=user_id).delete()
  872. cm.WanzhuConsult.objects.filter(Q(user_id=user_id)|Q(reply_user_id=user_id)).delete()
  873. def batch_update_player_latest(record):
  874. """更新选手最后一次数据
  875. """
  876. user_id = record["user_id"]
  877. match_id = record["match_id"]
  878. stock_date = record["stock_date"]
  879. match_group = record["match_group"]
  880. #wincnt = record["wincnt"]
  881. #total = record["totalcnt"]
  882. key = "PLAYER_LATEST_{}".format(user_id)
  883. if record and match_group:
  884. ccc.cache.hset(key,"stock_date",stock_date)
  885. ccc.cache.hset(key,"match_id",match_id)
  886. ccc.cache.hset(key,"match_group",match_group)
  887. ##更新胜率
  888. #qset = cm.PlayerRecord.get_db_model(match_id).objects.filter(match_id=match_id,user_id=user_id)
  889. ##win_rate = qset.filter(today_income__gte=0).count()/float(qset.count()) if qset else 0.0
  890. #win_rate = wincnt/float(total) if total else 0.0
  891. #win_rate = round(win_rate,3)
  892. #ccc.cache.hset(key,"win_rate",win_rate)
  893. def get_sub_category(pid):
  894. """
  895. """
  896. qset = cm.BaikeCategory.objects.filter(parent_id=pid)
  897. qdata = list(qset.values())
  898. if qdata:
  899. for subitem in qdata:
  900. subitem["type"] = "ent"
  901. subitem["label"] = subitem["name"]
  902. subitem["children"] = get_sub_category(subitem["id"])
  903. return qdata
  904. def get_baike_category_tree(**kwargs):
  905. parents = list(cm.BaikeCategory.objects.filter(parent_id__isnull=True).values())
  906. for item in parents:
  907. item["label"] = item["name"]
  908. item["children"] = get_sub_category(item["id"])
  909. return parents
  910. def get_baike_list_by_category(**kwargs):
  911. category_id = kwargs.get("category_id")
  912. qset = cm.BaikeDetail.objects.filter(category_id=category_id)
  913. data = list(qset.values("id","name"))
  914. return data
  915. def push_article(**kwargs):
  916. _id = kwargs.get("id")
  917. article = cm.Article.objects.filter(id=_id).first()
  918. #发送app推送消息
  919. title = u"顽主杯"
  920. body = article.name
  921. user_id = "ALL"
  922. params = {"type":"article","id":_id}
  923. aliyunpush.push_notice_by_userid(user_id,title,body,params,"ios")
  924. aliyunpush.push_notice_by_userid(user_id,title,body,params,"android")
  925. def import_article(**kwargs):
  926. """
  927. """
  928. obj = cm.Article.objects.create(**kwargs)
  929. def refund_out_trade_order(**kwargs):
  930. """
  931. """
  932. _id = kwargs.get("id")
  933. signup_order = cm.SignupOrder.objects.filter(id=_id).first()
  934. if signup_order:
  935. out_trade_no = signup_order.out_trade_no
  936. transaction_id = signup_order.transaction_id
  937. total_fee = signup_order.total_fee
  938. amount = str(int(total_fee*100))
  939. print(amount)
  940. result = refund_order(transaction_id,out_trade_no,out_trade_no,amount)
  941. if result.get("result_code") == "SUCCESS":
  942. signup_order.order_status = -1
  943. signup_order.save()
  944. return "success"
  945. else:
  946. raise ce.TipException(result.get("err_code_des"))
  947. else:
  948. raise ce.TipException(u"该订单不存在退款失败!")