拍爱

views.py 35KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084
  1. # -*- coding: utf-8 -*-
  2. from __future__ import division
  3. import os
  4. import random
  5. import records
  6. import shortuuid
  7. from curtail_uuid import CurtailUUID
  8. from django.conf import settings
  9. from django.core.files.storage import default_storage
  10. from django.db import connection, transaction
  11. from django.shortcuts import render
  12. from logit import logit
  13. from rest_framework import viewsets
  14. from TimeConvert import TimeConvert as tc
  15. from account.models import UserInfo
  16. from group.models import GroupInfo, GroupPhotoInfo, GroupUserInfo, PhotoCommentInfo, PhotoThumbUpInfo
  17. from group.serializers import GroupInfoSerializer, GroupPhotoInfoSerializer, GroupUserInfoSerializer
  18. from message.models import UserMessageInfo
  19. from utils.error.errno_utils import GroupPhotoStatusCode, GroupStatusCode, GroupUserStatusCode, UserStatusCode
  20. from utils.error.response_utils import response
  21. from utils.page_utils import pagination
  22. from utils.redis.rgroup import (del_group_photo_thumbup_flag, get_group_info, get_group_photo_comment_list,
  23. get_group_photo_data, get_group_photo_thumbup_flag, get_group_photo_thumbup_list,
  24. get_group_photo_watchers, get_group_users_info, set_group_info,
  25. set_group_photo_comment_list, set_group_photo_data, set_group_photo_thumbup_flag,
  26. set_group_photo_thumbup_list, set_group_users_info)
  27. from utils.redis.rkeys import (GROUP_LAST_PHOTO_PK, GROUP_PHOTO_WATCHER_SET, GROUP_USERS_APPLYING_SET,
  28. GROUP_USERS_DELETED_SET, GROUP_USERS_PASSED_SET, GROUP_USERS_QUIT_SET,
  29. GROUP_USERS_REFUSED_SET, LENSMAN_PHOTO_HAGGLE_TIMES, LENSMAN_PHOTO_PRICE)
  30. from utils.redis.rorder import get_lensman_order_record
  31. from utils.sql.raw import PAI2_HOME_API
  32. from utils.thumbnail_utils import make_thumbnail
  33. from utils.url_utils import img_url, share_url
  34. # db = records.Database(settings.DATABASE_URL['default'])
  35. r = settings.REDIS_CACHE
  36. @logit
  37. @transaction.atomic
  38. def group_create_api(request):
  39. """
  40. 群组创建
  41. :param request:
  42. :return:
  43. """
  44. user_id = request.POST.get('user_id', '')
  45. group_name = request.POST.get('group_name', '')
  46. group_default_avatar = int(request.POST.get('group_default_avatar', 0))
  47. # 用户校验
  48. try:
  49. user = UserInfo.objects.get(user_id=user_id)
  50. except UserInfo.DoesNotExist:
  51. return response(UserStatusCode.USER_NOT_FOUND)
  52. # 群组唯一标识
  53. group_id = CurtailUUID.uuid(GroupInfo, 'group_id')
  54. # 群组记录创建
  55. group = GroupInfo.objects.create(
  56. group_id=group_id,
  57. admin_id=user_id,
  58. group_name=group_name,
  59. group_default_avatar=group_default_avatar,
  60. group_from=GroupInfo.APP_GROUP,
  61. )
  62. # Redis 群组数据缓存
  63. group_info = set_group_info(group)
  64. # 群组用户记录创建
  65. GroupUserInfo.objects.create(
  66. group_id=group_id,
  67. user_id=user_id,
  68. nickname=user.final_nickname,
  69. avatar=user.avatar,
  70. admin=True,
  71. user_status=GroupUserInfo.PASSED,
  72. passed_at=tc.utc_datetime(),
  73. )
  74. # Redis 群组用户数据缓存
  75. group_users = set_group_users_info(group)
  76. # Redis 群组通过集合缓存
  77. r.sadd(GROUP_USERS_PASSED_SET % group_id, user_id)
  78. return response(200, 'Create Group Success', u'群组创建成功', {
  79. 'group_id': group_id,
  80. 'group': group_info,
  81. 'users': group_users,
  82. })
  83. @logit
  84. def group_detail_api(request):
  85. """
  86. 群组详情
  87. :param request:
  88. :return:
  89. """
  90. group_id = request.POST.get('group_id', '')
  91. user_id = request.POST.get('user_id', '')
  92. return response(200, 'Get Group Detail Info Success', u'获取群组详情成功', {
  93. 'group_id': group_id,
  94. 'group': get_group_info(group_id),
  95. 'users': get_group_users_info(group_id, user_id),
  96. })
  97. @logit
  98. def group_update_api(request):
  99. """
  100. 群组更新
  101. :param request:
  102. :return:
  103. """
  104. group_id = request.POST.get('group_id', '')
  105. admin_id = request.POST.get('admin_id', '') or request.POST.get('user_id', '')
  106. group_name = request.POST.get('group_name', '')
  107. group_desc = request.POST.get('group_desc', '')
  108. group_avatar = request.FILES.get('group_avatar', '')
  109. # 群组校验
  110. try:
  111. group = GroupInfo.objects.get(group_id=group_id)
  112. except GroupInfo.DoesNotExist:
  113. return response(GroupStatusCode.GROUP_NOT_FOUND)
  114. # 权限校验
  115. if group.admin_id != admin_id:
  116. return response(GroupStatusCode.NO_UPDATE_PERMISSION)
  117. # 群组名称更新
  118. if group_name:
  119. group.group_name = group_name
  120. # 群组描述更新
  121. if group_desc:
  122. group.group_desc = group_desc
  123. # 群组头像更新
  124. if group_avatar:
  125. _, extension = os.path.splitext(group_avatar.name)
  126. group_avatar_path = 'group/{uuid}_{extension}'.format(uuid=shortuuid.uuid(), extension=extension)
  127. if default_storage.exists(group_avatar_path):
  128. default_storage.delete(group_avatar_path)
  129. default_storage.save(group_avatar_path, group_avatar)
  130. group.group_avatar = group_avatar_path
  131. group.save()
  132. # Redis 群组数据缓存更新
  133. group_info = set_group_info(group)
  134. return response(200, 'Update Group Success', u'群组更新成功', {
  135. 'group_id': group_id,
  136. 'group': group_info,
  137. 'users': get_group_users_info(group_id, admin_id),
  138. })
  139. @logit
  140. def group_list_api(request):
  141. """
  142. 群组列表
  143. :param request:
  144. :return:
  145. """
  146. user_id = request.POST.get('user_id', '')
  147. page = int(request.POST.get('page', 1))
  148. num = int(request.POST.get('num', settings.GROUP_PER_PAGE))
  149. group_users = GroupUserInfo.objects.filter(user_id=user_id, user_status=GroupUserInfo.PASSED).order_by('-pk')
  150. group_users, left = pagination(group_users, page, num)
  151. groups = []
  152. for group_user in group_users:
  153. group_info = get_group_info(group_user.group_id)
  154. groups.append(group_info) if group_info else None
  155. return response(200, 'Get Group List Success', u'获取群组列表成功', {
  156. 'groups': groups,
  157. 'left': left,
  158. })
  159. @logit
  160. def group_join_api(request):
  161. """
  162. 申请加群
  163. :param request:
  164. :return:
  165. """
  166. group_id = request.POST.get('group_id', '')
  167. user_id = request.POST.get('user_id', '')
  168. nickname = request.POST.get('nickname', '')
  169. # 用户校验
  170. try:
  171. user = UserInfo.objects.get(user_id=user_id)
  172. except UserInfo.DoesNotExist:
  173. return response(UserStatusCode.USER_NOT_FOUND)
  174. # 群组校验
  175. try:
  176. group = GroupInfo.objects.get(group_id=group_id)
  177. except GroupInfo.DoesNotExist:
  178. return response(GroupStatusCode.GROUP_NOT_FOUND)
  179. # 群组锁定校验
  180. if group.group_lock:
  181. return response(GroupStatusCode.GROUP_HAS_LOCKED)
  182. # 群组用户记录创建,若记录不存在,则创建,若记录已存在,则更新
  183. group_user, created = GroupUserInfo.objects.get_or_create(
  184. group_id=group_id,
  185. user_id=user_id,
  186. )
  187. if group_user.user_status != GroupUserInfo.PASSED:
  188. group_user.current_id = int(r.get(GROUP_LAST_PHOTO_PK % group_id) or -1)
  189. group_user.nickname = nickname or user.final_nickname
  190. group_user.avatar = user.avatar
  191. # group_user.admin = False # Admin Field Default False, Should Not Assign
  192. group_user.user_status = GroupUserInfo.PASSED
  193. group_user.passed_at = tc.utc_datetime()
  194. group_user.save()
  195. # Redis 群组用户数据缓存
  196. set_group_users_info(group)
  197. # Redis 群组通过集合缓存
  198. r.srem(GROUP_USERS_REFUSED_SET % group_id, user_id)
  199. r.srem(GROUP_USERS_DELETED_SET % group_id, user_id)
  200. r.srem(GROUP_USERS_QUIT_SET % group_id, user_id)
  201. r.sadd(GROUP_USERS_PASSED_SET % group_id, user_id)
  202. return response(200, 'Apply Success', u'申请成功', {
  203. 'current_id': group_user.current_id,
  204. 'photos': [],
  205. 'group_id': group_id,
  206. 'group': get_group_info(group_id),
  207. 'user_id': user_id,
  208. 'users': get_group_users_info(group_id, user_id),
  209. })
  210. @logit
  211. def group_lock_api(request):
  212. """
  213. 群组锁定
  214. :param request:
  215. :return:
  216. """
  217. group_id = request.POST.get('group_id', '')
  218. admin_id = request.POST.get('admin_id', '') or request.POST.get('user_id', '')
  219. # 群组校验
  220. try:
  221. group = GroupInfo.objects.get(group_id=group_id)
  222. except GroupInfo.DoesNotExist:
  223. return response(GroupStatusCode.GROUP_NOT_FOUND)
  224. # 权限校验
  225. if group.admin_id != admin_id:
  226. return response(GroupStatusCode.NO_LOCK_PERMISSION)
  227. # 群组锁定
  228. group.group_lock = True
  229. group.save()
  230. # Redis 群组数据缓存更新
  231. set_group_info(group)
  232. return response(200, 'Lock Success', u'锁定成功')
  233. @logit
  234. def group_unlock_api(request):
  235. """
  236. 群组解锁
  237. :param request:
  238. :return:
  239. """
  240. group_id = request.POST.get('group_id', '')
  241. admin_id = request.POST.get('admin_id', '') or request.POST.get('user_id', '')
  242. # 群组校验
  243. try:
  244. group = GroupInfo.objects.get(group_id=group_id)
  245. except GroupInfo.DoesNotExist:
  246. return response(GroupStatusCode.GROUP_NOT_FOUND)
  247. # 权限校验
  248. if group.admin_id != admin_id:
  249. return response(GroupStatusCode.NO_UNLOCK_PERMISSION)
  250. # 群组解锁
  251. group.group_lock = False
  252. group.save()
  253. # Redis 群组数据缓存更新
  254. set_group_info(group)
  255. return response(200, u'Unlock Success', u'解锁成功')
  256. @logit
  257. def group_remove_api(request):
  258. """
  259. 成员移除
  260. :param request:
  261. :return:
  262. """
  263. group_id = request.POST.get('group_id', '')
  264. admin_id = request.POST.get('admin_id', '')
  265. user_id = request.POST.get('user_id', '')
  266. # 群组校验
  267. try:
  268. group = GroupInfo.objects.get(group_id=group_id)
  269. except GroupInfo.DoesNotExist:
  270. return response(GroupStatusCode.GROUP_NOT_FOUND)
  271. # 权限校验
  272. if group.admin_id != admin_id or group.admin_id == user_id: # 管理员也不允许将自己移除
  273. return response(GroupStatusCode.NO_REMOVE_PERMISSION)
  274. # 群组用户校验
  275. try:
  276. group_user = GroupUserInfo.objects.get(group_id=group_id, user_id=user_id, user_status=GroupUserInfo.PASSED)
  277. except GroupUserInfo.DoesNotExist:
  278. return response(GroupUserStatusCode.GROUP_USER_NOT_FOUND)
  279. # 群组用户移除
  280. group_user.user_status = GroupUserInfo.DELETED
  281. group_user.deleted_at = tc.utc_datetime()
  282. group_user.save()
  283. # Redis 群组数据缓存更新
  284. group_users = set_group_users_info(group)
  285. # Redis 群组删除集合缓存
  286. r.srem(GROUP_USERS_PASSED_SET % group_id, user_id)
  287. r.sadd(GROUP_USERS_DELETED_SET % group_id, user_id)
  288. return response(200, 'Remove Success', u'用户移除成功', {
  289. 'group_id': group_id,
  290. 'users': group_users,
  291. })
  292. @logit
  293. def group_quit_api(request):
  294. """
  295. 成员退出
  296. :param request:
  297. :return:
  298. """
  299. group_id = request.POST.get('group_id', '')
  300. user_id = request.POST.get('user_id', '')
  301. # 群组校验
  302. try:
  303. group = GroupInfo.objects.get(group_id=group_id)
  304. except GroupInfo.DoesNotExist:
  305. return response(GroupStatusCode.GROUP_NOT_FOUND)
  306. # 权限校验
  307. if group.admin_id == user_id: # 管理员也不允许自己退出
  308. return response(GroupStatusCode.NO_QUIT_PERMISSION)
  309. # 群组用户校验
  310. try:
  311. group_user = GroupUserInfo.objects.get(group_id=group_id, user_id=user_id, user_status=GroupUserInfo.PASSED)
  312. except GroupUserInfo.DoesNotExist:
  313. return response(GroupUserStatusCode.GROUP_USER_NOT_FOUND)
  314. # 群组用户移除
  315. group_user.user_status = GroupUserInfo.QUIT
  316. group_user.quit_at = tc.utc_datetime()
  317. group_user.save()
  318. # Redis 群组数据缓存更新
  319. group_users = set_group_info(group)
  320. # Redis 群组删除集合缓存
  321. r.srem(GROUP_USERS_PASSED_SET % group_id, user_id)
  322. r.sadd(GROUP_USERS_QUIT_SET % group_id, user_id)
  323. return response(200, 'Quit Success', u'用户退出成功', {
  324. 'group_id': group_id,
  325. 'users': group_users,
  326. })
  327. @logit
  328. def group_pass_api(request):
  329. """
  330. 申请通过
  331. :param request:
  332. :return:
  333. """
  334. group_id = request.POST.get('group_id', '')
  335. admin_id = request.POST.get('admin_id', '')
  336. user_id = request.POST.get('user_id', '')
  337. # 群组校验
  338. try:
  339. group = GroupInfo.objects.get(group_id=group_id)
  340. except GroupInfo.DoesNotExist:
  341. return response(GroupStatusCode.GROUP_NOT_FOUND)
  342. # 权限校验
  343. if group.admin_id != admin_id:
  344. return response(GroupStatusCode.NO_PASS_PERMISSION)
  345. # 群组用户校验
  346. try:
  347. group_user = GroupUserInfo.objects.get(group_id=group_id, user_id=user_id, user_status=GroupUserInfo.APPLYING)
  348. except GroupUserInfo.DoesNotExist:
  349. return response(GroupStatusCode.JOIN_REQUEST_NOT_FOUND)
  350. # 群组用户通过
  351. group_user.user_status = GroupUserInfo.PASSED
  352. group_user.passed_at = tc.utc_datetime()
  353. group_user.save()
  354. # Redis 群组数据缓存更新
  355. group_users = set_group_info(group)
  356. # Redis 群组通过集合缓存
  357. r.srem(GROUP_USERS_APPLYING_SET % group_id, user_id)
  358. r.sadd(GROUP_USERS_PASSED_SET % group_id, user_id)
  359. return response(200, 'Pass Success', u'申请通过成功', {
  360. 'group_id': group_id,
  361. 'users': group_users,
  362. })
  363. @logit
  364. def group_refuse_api(request):
  365. """
  366. 申请拒绝
  367. :param request:
  368. :return:
  369. """
  370. group_id = request.POST.get('group_id', '')
  371. admin_id = request.POST.get('admin_id', '')
  372. user_id = request.POST.get('user_id', '')
  373. # 群组校验
  374. try:
  375. group = GroupInfo.objects.get(group_id=group_id)
  376. except GroupInfo.DoesNotExist:
  377. return response(GroupStatusCode.GROUP_NOT_FOUND)
  378. # 权限校验
  379. if group.admin_id != admin_id:
  380. return response(GroupStatusCode.NO_REFUSE_PERMISSION)
  381. # 群组用户校验
  382. try:
  383. group_user = GroupUserInfo.objects.get(group_id=group_id, user_id=user_id, user_status=GroupUserInfo.APPLYING)
  384. except GroupUserInfo.DoesNotExist:
  385. return response(GroupStatusCode.JOIN_REQUEST_NOT_FOUND)
  386. # 群组用户拒绝
  387. group_user.user_status = GroupUserInfo.REFUSED
  388. group_user.refused_at = tc.utc_datetime()
  389. group_user.save()
  390. # Redis 群组数据缓存更新
  391. group_users = set_group_info(group)
  392. # Redis 群组拒绝集合缓存
  393. r.srem(GROUP_USERS_APPLYING_SET % group_id, user_id)
  394. r.sadd(GROUP_USERS_REFUSED_SET % group_id, user_id)
  395. return response(200, 'Refuse Success', u'申请拒绝成功', {
  396. 'group_id': group_id,
  397. 'users': group_users,
  398. })
  399. @logit
  400. def group_data_api(request):
  401. """
  402. 群组数据, 评论数, 点赞数
  403. :param request:
  404. :return:
  405. """
  406. group_id = request.POST.get('group_id', '')
  407. return response(200, 'Get Group Data Success', u'获取群组数据成功', {
  408. 'photo_datas': get_group_photo_data(group_id),
  409. })
  410. @logit
  411. def flyimg_upload_api(request):
  412. """
  413. 飞图上传/飞图列表
  414. :param request:
  415. :return:
  416. """
  417. group_id = request.POST.get('group_id', '')
  418. user_id = request.POST.get('user_id', '')
  419. nickname = request.POST.get('nickname', '')
  420. photo = request.FILES.get('photo', '')
  421. current_id = int(request.POST.get('current_id', -1))
  422. # 用户校验
  423. try:
  424. user = UserInfo.objects.get(user_id=user_id)
  425. except UserInfo.DoesNotExist:
  426. return response(UserStatusCode.USER_NOT_FOUND)
  427. # 群组用户校验
  428. try:
  429. group_user = GroupUserInfo.objects.get(group_id=group_id, user_id=user_id, user_status=GroupUserInfo.PASSED)
  430. except GroupUserInfo.DoesNotExist:
  431. return response(GroupUserStatusCode.GROUP_USER_NOT_FOUND)
  432. if photo:
  433. photo_path = 'fly/{uuid}{extension}'.format(uuid=shortuuid.uuid(), extension=os.path.splitext(photo.name)[1] or 'jpeg')
  434. photo_thumbnail_path = photo_path.replace('.', '_thumbnail.')
  435. photo_thumbnail2_path = photo_path.replace('.', '_thumbnail2.')
  436. if default_storage.exists(photo_path):
  437. default_storage.delete(photo_path)
  438. default_storage.save(photo_path, photo)
  439. # if default_storage.exists(photo_thumbnail_path):
  440. # default_storage.delete(photo_thumbnail_path)
  441. # default_storage.save(photo_thumbnail_path, photo)
  442. # 群组照片缩略图生成
  443. # 双列: 540, 40-50K
  444. photo_w, photo_h, photo_thumbnail_w, photo_thumbnail_h = make_thumbnail(
  445. os.path.join(settings.MEDIA_ROOT, photo_path).replace('\\', '/'),
  446. os.path.join(settings.MEDIA_ROOT, photo_thumbnail_path).replace('\\', '/'),
  447. settings.THUMBNAIL_MAX_WIDTH
  448. )
  449. # 单列: 1080, xx-100K
  450. photo_w, photo_h, photo_thumbnail2_w, photo_thumbnail2_h = make_thumbnail(
  451. os.path.join(settings.MEDIA_ROOT, photo_path).replace('\\', '/'),
  452. os.path.join(settings.MEDIA_ROOT, photo_thumbnail2_path).replace('\\', '/'),
  453. settings.THUMBNAIL_MAX_WIDTH2
  454. )
  455. # 群组照片记录创建
  456. group_photo = GroupPhotoInfo.objects.create(
  457. group_id=group_id,
  458. user_id=user_id,
  459. nickname=nickname or user.final_nickname,
  460. avatar=user.avatar,
  461. photo_path=photo_path,
  462. photo_w=photo_w,
  463. photo_h=photo_h,
  464. photo_thumbnail_path=photo_thumbnail_path,
  465. photo_thumbnail_w=photo_thumbnail_w,
  466. photo_thumbnail_h=photo_thumbnail_h,
  467. photo_thumbnail2_path=photo_thumbnail2_path,
  468. photo_thumbnail2_w=photo_thumbnail2_w,
  469. photo_thumbnail2_h=photo_thumbnail2_h,
  470. )
  471. # 设置群组最后一张照片PK
  472. r.set(GROUP_LAST_PHOTO_PK % group_id, group_photo.pk)
  473. # 获取从 current_id 到 now 的群组照片列表
  474. group_photos = GroupPhotoInfo.objects.filter(
  475. group_id=group_id,
  476. status=True,
  477. pk__gt=max(current_id, group_user.current_id),
  478. ).order_by(
  479. '-pk'
  480. )
  481. latest_photo = group_photos.first()
  482. return response(200, 'Flyimg Upload Success', u'飞图上传成功', {
  483. 'current_id': latest_photo and latest_photo.pk or current_id,
  484. 'photos': [photo.photo_info(user_id) for photo in group_photos],
  485. })
  486. @logit
  487. def comment_submit_api(request):
  488. """
  489. 飞图评论提交/飞图评论列表
  490. :param request:
  491. :return:
  492. """
  493. group_id = request.POST.get('group_id', '')
  494. user_id = request.POST.get('user_id', '')
  495. to_uid = request.POST.get('to_uid', '')
  496. photo_id = request.POST.get('photo_id', '')
  497. comment = request.POST.get('comment', '')
  498. # 群组用户校验
  499. try:
  500. group_user = GroupUserInfo.objects.get(group_id=group_id, user_id=user_id, user_status=GroupUserInfo.PASSED)
  501. except GroupUserInfo.DoesNotExist:
  502. return response(GroupUserStatusCode.GROUP_USER_NOT_FOUND)
  503. # 群组照片校验
  504. try:
  505. group_photo = GroupPhotoInfo.objects.get(pk=photo_id)
  506. except GroupPhotoInfo.DoesNotExist:
  507. return response(GroupPhotoStatusCode.GROUP_PHOTO_NOT_FOUND)
  508. if comment:
  509. # 群组照片评论记录创建
  510. PhotoCommentInfo.objects.create(
  511. photo_id=photo_id,
  512. user_id=user_id,
  513. nickname=group_user.nickname,
  514. avatar=group_user.avatar,
  515. to_uid=to_uid,
  516. comment=comment,
  517. )
  518. # 群组照片评论数更新
  519. group_photo.comment_num += 1
  520. group_photo.save()
  521. # Redis 群组照片数据缓存
  522. set_group_photo_data(group_id)
  523. # Redis 群组照片评论列表缓存刷新
  524. set_group_photo_comment_list(photo_id)
  525. r.sadd(GROUP_PHOTO_WATCHER_SET % photo_id, user_id)
  526. # 判断群组照片发布者是否已经被管理员移除/主动退出,如若移除/退出,则不给发布者提醒
  527. # 照片所有者(评论/点赞)不给自己提醒
  528. if r.sismember(GROUP_USERS_PASSED_SET % group_photo.group_id, group_photo.user_id) and group_photo.user_id != user_id:
  529. UserMessageInfo.objects.create(
  530. from_uid=user_id,
  531. from_nickname=group_user.nickname,
  532. from_avatar=group_user.avatar,
  533. to_uid=group_photo.user_id,
  534. group_id=group_photo.group_id,
  535. photo_id=group_photo.pk,
  536. msg_type=UserMessageInfo.COMMENT,
  537. msg_title=u'评论',
  538. msg_content=comment,
  539. )
  540. # 给所有关注者(评论/点赞)发送提醒
  541. watchers = get_group_photo_watchers(photo_id)
  542. # 从关注者中移除该(评论/点赞)者
  543. watchers.discard(user_id)
  544. # 从关注者中移除该照片所有者
  545. watchers.discard(group_photo.user_id)
  546. for watcher in watchers:
  547. UserMessageInfo.objects.create(
  548. from_uid=user_id,
  549. from_nickname=group_user.nickname,
  550. from_avatar=group_user.avatar,
  551. to_uid=watcher,
  552. group_id=group_photo.group_id,
  553. photo_id=group_photo.pk,
  554. msg_type=UserMessageInfo.COMMENT,
  555. msg_title=u'评论',
  556. msg_content=comment,
  557. )
  558. return response(200, 'Comment Success', u'评论成功', {
  559. 'comments': get_group_photo_comment_list(photo_id),
  560. })
  561. @logit
  562. def thumbup_submit_api(request):
  563. """
  564. 飞图点赞提交
  565. :param request:
  566. :return:
  567. """
  568. group_id = request.POST.get('group_id', '')
  569. user_id = request.POST.get('user_id', '')
  570. photo_id = request.POST.get('photo_id', '')
  571. # 群组用户校验
  572. try:
  573. group_user = GroupUserInfo.objects.get(group_id=group_id, user_id=user_id, user_status=GroupUserInfo.PASSED)
  574. except GroupUserInfo.DoesNotExist:
  575. return response(GroupUserStatusCode.GROUP_USER_NOT_FOUND)
  576. # 群组照片校验
  577. try:
  578. group_photo = GroupPhotoInfo.objects.get(pk=photo_id)
  579. except GroupPhotoInfo.DoesNotExist:
  580. return response(GroupPhotoStatusCode.GROUP_PHOTO_NOT_FOUND)
  581. # user_id 是否点赞 photo_id
  582. if get_group_photo_thumbup_flag(photo_id, user_id):
  583. return response(GroupPhotoStatusCode.DUPLICATE_THUMB_UP)
  584. # 群组照片点赞记录创建/更新
  585. photo_thumbup, created = PhotoThumbUpInfo.objects.get_or_create(
  586. photo_id=photo_id,
  587. user_id=user_id,
  588. )
  589. photo_thumbup.nickname = group_user.nickname
  590. photo_thumbup.avatar = group_user.avatar
  591. photo_thumbup.thumbup = True
  592. photo_thumbup.save()
  593. # Redis 群组照片点赞数据缓存
  594. set_group_photo_thumbup_flag(photo_id, user_id)
  595. # 群组照片点赞数更新
  596. group_photo.thumbup_num += 1
  597. group_photo.save()
  598. # Redis 群组照片数据缓存
  599. set_group_photo_data(group_id)
  600. # Redis 群组照片点赞列表缓存刷新
  601. set_group_photo_thumbup_list(photo_id)
  602. r.sadd(GROUP_PHOTO_WATCHER_SET % photo_id, user_id)
  603. # 判断群组照片发布者是否已经被管理员移除/主动退出,如若移除/退出,则不给发布者提醒
  604. # 照片所有者(评论/点赞)不给自己提醒
  605. if r.sismember(GROUP_USERS_PASSED_SET % group_photo.group_id, group_photo.user_id) and group_photo.user_id != user_id:
  606. UserMessageInfo.objects.create(
  607. from_uid=user_id,
  608. from_nickname=group_user.nickname,
  609. from_avatar=group_user.avatar,
  610. to_uid=group_photo.user_id,
  611. group_id=group_photo.group_id,
  612. photo_id=group_photo.pk,
  613. msg_type=UserMessageInfo.THUMBUP,
  614. msg_title=u'点赞',
  615. msg_content=u'点赞',
  616. )
  617. # 给所有关注者(评论/点赞)发送提醒
  618. watchers = get_group_photo_watchers(photo_id)
  619. # 从关注者中移除该(评论/点赞)者
  620. watchers.discard(user_id)
  621. # 从关注者中移除该照片所有者
  622. watchers.discard(group_photo.user_id)
  623. for watcher in watchers:
  624. UserMessageInfo.objects.create(
  625. from_uid=user_id,
  626. from_nickname=group_user.nickname,
  627. from_avatar=group_user.avatar,
  628. to_uid=watcher,
  629. group_id=group_photo.group_id,
  630. photo_id=group_photo.pk,
  631. msg_type=UserMessageInfo.THUMBUP,
  632. msg_title=u'点赞',
  633. msg_content=u'点赞',
  634. )
  635. return response(200, 'Thumbup Success', u'点赞提交成功', {
  636. 'thumbup': True,
  637. 'thumbups': get_group_photo_thumbup_list(photo_id),
  638. })
  639. @logit
  640. def thumbup_list_api(request):
  641. """
  642. 飞图点赞列表
  643. :param request:
  644. :return:
  645. """
  646. group_id = request.POST.get('group_id', '')
  647. user_id = request.POST.get('user_id', '')
  648. photo_id = request.POST.get('photo_id', '')
  649. return response(200, 'Get Thumbup List Success', u'获取点赞列表成功', {
  650. 'thumbup': get_group_photo_thumbup_flag(photo_id, user_id), # user_id 是否点赞 photo_id
  651. 'thumbups': get_group_photo_thumbup_list(photo_id), # 群组照片点赞列表
  652. })
  653. @logit
  654. def thumbup_cancel_api(request):
  655. """
  656. 飞图点赞取消
  657. :param request:
  658. :return:
  659. """
  660. group_id = request.POST.get('group_id', '')
  661. user_id = request.POST.get('user_id', '')
  662. photo_id = request.POST.get('photo_id', '')
  663. # 群组用户校验
  664. try:
  665. group_user = GroupUserInfo.objects.get(group_id=group_id, user_id=user_id, user_status=GroupUserInfo.PASSED)
  666. except GroupUserInfo.DoesNotExist:
  667. return response(GroupUserStatusCode.GROUP_USER_NOT_FOUND)
  668. # 群组照片校验
  669. try:
  670. group_photo = GroupPhotoInfo.objects.get(pk=photo_id)
  671. except GroupPhotoInfo.DoesNotExist:
  672. return response(GroupPhotoStatusCode.GROUP_PHOTO_NOT_FOUND)
  673. # user_id 是否点赞 photo_id
  674. if not get_group_photo_thumbup_flag(photo_id, user_id):
  675. return response(GroupPhotoStatusCode.THUMB_UP_NOT_FOUND)
  676. # 群组照片点赞取消
  677. photo_thumbup, created = PhotoThumbUpInfo.objects.get_or_create(
  678. photo_id=photo_id,
  679. user_id=user_id,
  680. )
  681. photo_thumbup.thumbup = False
  682. photo_thumbup.save()
  683. # Redis 群组照片点赞数据移除
  684. del_group_photo_thumbup_flag(photo_id, user_id)
  685. # 群组照片点赞数更新
  686. group_photo.thumbup_num -= 1
  687. group_photo.save()
  688. # Redis 群组照片数据缓存
  689. set_group_photo_data(group_id)
  690. # Redis 群组照片点赞列表缓存刷新
  691. set_group_photo_thumbup_list(photo_id)
  692. # 判断群组照片发布者是否已经被管理员移除/主动退出,如若移除/退出,则不给发布者提醒
  693. if r.sismember(GROUP_USERS_PASSED_SET % group_photo.group_id, group_photo.user_id):
  694. UserMessageInfo.objects.create(
  695. from_uid=user_id,
  696. from_nickname=group_user.nickname,
  697. from_avatar=group_user.avatar,
  698. to_uid=group_photo.user_id,
  699. group_id=group_photo.group_id,
  700. photo_id=group_photo.pk,
  701. msg_type=UserMessageInfo.THUMBUP,
  702. msg_title=u'取消点赞',
  703. msg_content=u'取消点赞',
  704. )
  705. # 群组照片点赞列表
  706. photo_thumbups = PhotoThumbUpInfo.objects.filter(
  707. photo_id=photo_id,
  708. thumbup=True,
  709. )
  710. return response(200, 'Thumbup Cancel Success', u'点赞取消成功', {
  711. 'thumbup': False,
  712. 'thumbups': [thumbup.thumbup_info for thumbup in photo_thumbups],
  713. })
  714. @logit
  715. def pai2_home_api(request):
  716. """
  717. 首页信息
  718. :param request:
  719. :return:
  720. """
  721. user_id = request.POST.get('user_id', '')
  722. page = int(request.POST.get('page', 1))
  723. num = int(request.POST.get('num', settings.PAI2_HOME_PER_PAGE))
  724. # 执行原生 SQL 语句,获取首页照片列表
  725. cursor = connection.cursor()
  726. cursor.execute(PAI2_HOME_API.format(
  727. user_id=user_id,
  728. offset=0,
  729. rows=settings.PAI2_HOME_MAX_ROWS,
  730. ))
  731. rows = cursor.fetchall()
  732. # 使用 records 执行原生 SQL 语句,获取首页照片列表
  733. # rows = db.query(PAI2_HOME_API.format(
  734. # user_id=user_id,
  735. # offset=0,
  736. # rows=settings.PAI2_HOME_MAX_ROWS,
  737. # )).all()
  738. # 首页照片分页
  739. rows, left = pagination(rows, page, num)
  740. # 首页照片信息
  741. rows = [{
  742. 'group_id': row[0],
  743. 'group_name': row[1],
  744. 'group_default_avatar': row[2],
  745. 'group_avatar': row[3],
  746. 'group_from': row[4],
  747. 'photo_id': row[5],
  748. 'photo_url': img_url(row[6]),
  749. 'photo_w': row[7],
  750. 'photo_h': row[8],
  751. 'photo_thumbnail_url': img_url(row[9]),
  752. 'photo_thumbnail_w': row[10],
  753. 'photo_thumbnail_h': row[11],
  754. 'photo_thumbnail2_url': img_url(row[12]),
  755. 'photo_thumbnail2_w': row[13],
  756. 'photo_thumbnail2_h': row[14],
  757. 'photo_share_url': share_url(row[5]), # Warning: Index of This Line is 5
  758. 'user_id': row[15],
  759. 'nickname': row[16],
  760. 'avatar': row[17],
  761. 'comment_num': row[18],
  762. 'thumbup_num': row[19],
  763. 'photo_from': row[20],
  764. 'created_at': row[21],
  765. 'thumbup': get_group_photo_thumbup_flag(row[5], user_id),
  766. 'porder': get_lensman_order_record(row[5], user_id) if row[20] == GroupPhotoInfo.SESSION_GROUP else {},
  767. } for row in rows]
  768. # rows = [dict(row) for row in rows]
  769. # [row.update({
  770. # 'photo_url': img_url(row['photo_path']),
  771. # 'photo_thumbnail_url': img_url(row['photo_thumbnail_path']),
  772. # 'photo_thumbnail2_url': img_url(row['photo_thumbnail2_path']),
  773. # 'photo_share_url': share_url(row['photo_id']),
  774. # 'thumbup': get_group_photo_thumbup_flag(row['photo_id'], user_id),
  775. # 'porder': get_lensman_order_record(row['photo_id'], user_id) if row['photo_from'] == GroupPhotoInfo.SESSION_GROUP else {},
  776. # }) for row in rows]
  777. return response(200, 'Get Home Data Success', u'获取首页数据成功', {
  778. 'photos': rows,
  779. 'left': left,
  780. })
  781. @logit
  782. def lensman_photo_price(request):
  783. """
  784. 摄影师照片价格获取
  785. :param request:
  786. :return:
  787. """
  788. user_id = request.POST.get('user_id', '')
  789. photo_id = request.POST.get('photo_id', '')
  790. photo_type = request.POST.get('photo_type', 'nomark') # nomark for 去除水印, origin for 获取高清图
  791. # 处理价格逻辑
  792. lensman_photo_price_key = LENSMAN_PHOTO_PRICE % (user_id, photo_id, photo_type)
  793. lensman_photo_haggle_times_key = LENSMAN_PHOTO_HAGGLE_TIMES % (user_id, photo_id, photo_type)
  794. # Redis 获取存储的价格
  795. price = int(r.get(lensman_photo_price_key) or 0)
  796. if price:
  797. haggle_times = int(r.get(lensman_photo_haggle_times_key) or 0)
  798. # 砍价逻辑
  799. if haggle_times < settings.LENSMAN_PHOTO_HAGGLE_MAX_TIMES:
  800. price -= random.choice([50, 100])
  801. r.incr(lensman_photo_haggle_times_key)
  802. else:
  803. # 获取摄影师定价
  804. # TODO, 此处需要完整的摄影师定价
  805. price = 999 if photo_type == 'origin' else 299
  806. r.set(lensman_photo_price_key, price)
  807. return response(200, 'Get Price Success', u'获取价格成功', {
  808. 'price': price
  809. })
  810. @logit
  811. def lensman_photo_bought(request):
  812. """
  813. 摄影师照片已购买
  814. :param request:
  815. :return:
  816. """
  817. user_id = request.POST.get('user_id', '')
  818. photo_id = request.POST.get('photo_id', '')
  819. return response(200, 'Get Bought Data Success', u'获取购买数据成功', {
  820. 'porder': get_lensman_order_record(photo_id, user_id)
  821. })
  822. @logit
  823. def group_photo_detail(request, photo_id):
  824. photo = GroupPhotoInfo.objects.get(pk=photo_id)
  825. return render(request, 'photo/photo_detail.html', {'photo_url': photo.photo_url, 'api_domain': settings.API_DOMAIN})
  826. @logit
  827. def group_detail(request, group_id):
  828. return render(request, 'page/{}_download.html'.format('ios' if request.iOS else 'adr'), {})
  829. class GroupInfoViewSet(viewsets.ModelViewSet):
  830. queryset = GroupInfo.objects.all().order_by('-pk')
  831. serializer_class = GroupInfoSerializer
  832. class GroupUserInfoViewSet(viewsets.ModelViewSet):
  833. queryset = GroupUserInfo.objects.all().order_by('-pk')
  834. serializer_class = GroupUserInfoSerializer
  835. class GroupPhotoInfoViewSet(viewsets.ModelViewSet):
  836. queryset = GroupPhotoInfo.objects.all().order_by('-pk')
  837. serializer_class = GroupPhotoInfoSerializer
  838. # Only Once Function
  839. def refresh_thumbnail():
  840. """ 刷新缩略图 """
  841. photos = GroupPhotoInfo.objects.filter(status=True)
  842. for photo in photos:
  843. try:
  844. photo_path = photo.photo_path
  845. photo_thumbnail_path = photo_path.replace('.', '_thumbnail.')
  846. photo_thumbnail2_path = photo_path.replace('.', '_thumbnail2.')
  847. # 群组照片缩略图生成
  848. # 双列: 540, 40-50K
  849. photo_w, photo_h, photo_thumbnail_w, photo_thumbnail_h = make_thumbnail(
  850. os.path.join(settings.MEDIA_ROOT, photo_path).replace('\\', '/'),
  851. os.path.join(settings.MEDIA_ROOT, photo_thumbnail_path).replace('\\', '/'),
  852. settings.THUMBNAIL_MAX_WIDTH
  853. )
  854. # 单列: 1080, xx-100K
  855. photo_w, photo_h, photo_thumbnail2_w, photo_thumbnail2_h = make_thumbnail(
  856. os.path.join(settings.MEDIA_ROOT, photo_path).replace('\\', '/'),
  857. os.path.join(settings.MEDIA_ROOT, photo_thumbnail2_path).replace('\\', '/'),
  858. settings.THUMBNAIL_MAX_WIDTH2
  859. )
  860. photo.photo_w = photo_w
  861. photo.photo_h = photo_h
  862. photo.photo_thumbnail_path = photo_thumbnail_path
  863. photo.photo_thumbnail_w = photo_thumbnail_w
  864. photo.photo_thumbnail_h = photo_thumbnail_h
  865. photo.photo_thumbnail2_path = photo_thumbnail2_path
  866. photo.photo_thumbnail2_w = photo_thumbnail2_w
  867. photo.photo_thumbnail2_h = photo_thumbnail2_h
  868. photo.save()
  869. except Exception as e:
  870. pass
  871. return 'Refresh Thumbnail OK'
  872. def statistic_thumbnail_size(pfrom):
  873. """
  874. 统计缩略图大小
  875. :param pfrom: 0 for APP_GROUP, 1 for SESSION_GROUP, -1 for ALL
  876. :return:
  877. """
  878. if pfrom == -1:
  879. photos = GroupPhotoInfo.objects.filter(status=True)
  880. else:
  881. photos = GroupPhotoInfo.objects.filter(photo_from=pfrom, status=True)
  882. photo_count = photos.count()
  883. photo_size = 0
  884. photo_thumbnail_size = 0
  885. photo_thumbnail2_size = 0
  886. for photo in photos:
  887. photo_size += os.path.getsize(os.path.join(settings.MEDIA_ROOT, photo.photo_path).replace('\\', '/'))
  888. photo_thumbnail_size += os.path.getsize(os.path.join(settings.MEDIA_ROOT, photo.photo_thumbnail_path).replace('\\', '/'))
  889. photo_thumbnail2_size += os.path.getsize(os.path.join(settings.MEDIA_ROOT, photo.photo_thumbnail2_path).replace('\\', '/'))
  890. print '>>> Photo Size: %.3f KB' % (photo_size / 1024 / photo_count)
  891. print '>>> Photo Thumbnail Size: %.3f KB' % (photo_thumbnail_size / 1024 / photo_count)
  892. print '>>> Photo Thumbnail2 Size: %.3f KB' % (photo_thumbnail2_size / 1024 / photo_count)
  893. return 'Statistic Thumbnail Size OK'