shareUserPermission.py 18 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392
  1. #!/usr/bin/env python
  2. # -*- coding: utf-8 -*-
  3. from django.core import serializers
  4. import traceback, simplejson as json
  5. from django.views.generic.base import View
  6. from django.core.exceptions import FieldError
  7. from django.views.decorators.csrf import csrf_exempt
  8. from django.utils.decorators import method_decorator
  9. from Model.models import Device_User, Device_Info, Role
  10. from Service.CommonService import CommonService
  11. from Object.ResponseObject import ResponseObject
  12. from Object.TokenObject import TokenObject
  13. from django.db.models import Q
  14. class searchUserView(View):
  15. @method_decorator(csrf_exempt)
  16. def dispatch(self, *args, **kwargs):
  17. return super(searchUserView, self).dispatch(*args, **kwargs)
  18. def post(self, request, *args, **kwargs):
  19. request.encoding = 'utf-8'
  20. fieldDict = request.POST
  21. return self.searchUser(fieldDict, args, kwargs)
  22. def get(self, request, *args, **kwargs):
  23. request.encoding = 'gb2312'
  24. fieldDict = request.GET
  25. return self.searchUser(fieldDict, args, kwargs)
  26. def searchUserSQL(self, fieldDict, response, *args, **kwargs):
  27. exact = fieldDict.get('exact', True)
  28. if exact == '0' or exact == 'False':
  29. exact = False
  30. else:
  31. exact = True
  32. if 'exact' in fieldDict.keys():
  33. fieldDict.pop('exact')
  34. try:
  35. if exact == 'True' or exact == 1:
  36. User = Device_User.objects.filter(**fieldDict).order_by('-data_joined')
  37. # elif exact == 'False' or exact == 0:
  38. else:
  39. if 'username' in fieldDict.keys():
  40. username = fieldDict.get('username',None)
  41. User = Device_User.objects.filter(Q(username=username)|Q(phone=username)|Q(userEmail=username)).order_by('-data_joined')
  42. elif 'userEmail' in fieldDict.keys():
  43. email = fieldDict.get('userEmail',None)
  44. User = Device_User.objects.filter(Q(userEmail=email)|Q(username=email)).order_by('-data_joined')
  45. elif 'userID' in fieldDict.keys():
  46. userID = fieldDict.get('userID',None)
  47. User = Device_User.objects.filter(userID=userID).order_by('-data_joined')
  48. else:
  49. User = Device_User.objects.filter(**fieldDict).order_by('-data_joined')
  50. # else:
  51. # User = Device_User.objects.filter(**fieldDict).order_by('-data_joined')
  52. except FieldError:
  53. return response.json(502)
  54. except Exception as e:
  55. errorInfo = traceback.format_exc()
  56. print('查询数据库错误: %s' % errorInfo)
  57. return response.json(500, {'details': repr(e)})
  58. else:
  59. if User:
  60. ddUser = User[0].device_info_set.all()
  61. sqlJSON = serializers.serialize('json', User)
  62. sqlList = json.loads(sqlJSON)
  63. if ddUser:
  64. sqlJSON1 = serializers.serialize('json', ddUser)
  65. sqlList1 = json.loads(sqlJSON1)
  66. device_Info_dict = {}
  67. device_Info_list = []
  68. for eachJson1 in sqlList1:
  69. device_Info_dict['primaryUserID'] = eachJson1['fields']['primaryUserID']
  70. device_Info_dict['Type'] = eachJson1['fields']['Type']
  71. device_Info_dict['UID'] = eachJson1['fields']['UID']
  72. device_Info_dict['pk'] = eachJson1['pk']
  73. device_Info_dict['NickName'] = eachJson1['fields']['NickName']
  74. device_Info_dict['View_Password'] = eachJson1['fields']['View_Password']
  75. device_Info_dict['View_Account'] = eachJson1['fields']['View_Account']
  76. device_Info_dict['Online'] = eachJson1['fields']['Online']
  77. device_Info_dict['EventNotification'] = eachJson1['fields']['EventNotification']
  78. device_Info_dict['ChannelIndex'] = eachJson1['fields']['ChannelIndex']
  79. device_Info_dict['EventNotification'] = eachJson1['fields']['EventNotification']
  80. device_Info_dict['NotificationMode'] = eachJson1['fields']['NotificationMode']
  81. device_Info_list.append(device_Info_dict)
  82. device_Info_dict = {}
  83. sqlList[0]['device_info'] = device_Info_list
  84. sqlDict = dict(zip(["datas"], [sqlList]))
  85. return response.json(0, sqlDict)
  86. else:
  87. return response.json(102)
  88. def searchUserPCSQL(self, fieldDict, response):
  89. try:
  90. page = int(fieldDict['page'])
  91. line = int(fieldDict['line'])
  92. fieldDict.pop('page')
  93. fieldDict.pop('line')
  94. fieldDict.pop('type')
  95. if len(fieldDict) > 0:
  96. searchCondition = CommonService.get_kwargs(data=fieldDict)
  97. device_user_queryset = Device_User.objects.filter(**searchCondition).order_by('-data_joined')
  98. else:
  99. device_user_queryset = Device_User.objects.all().order_by('-data_joined')
  100. except FieldError:
  101. return response.json(502)
  102. except Exception as e:
  103. errorInfo = traceback.format_exc()
  104. print('查询数据库错误: %s' % errorInfo)
  105. return response.json(500, {"details": repr(e)})
  106. else:
  107. if device_user_queryset:
  108. device_user_count = device_user_queryset.count()
  109. device_user_res = device_user_queryset[(page - 1) * line:page * line]
  110. sqlDict = CommonService.qs_to_dict(device_user_res)
  111. for k, v in enumerate(sqlDict["datas"]):
  112. for val in device_user_res:
  113. if v['pk'] == val.userID:
  114. device_info_query_set = val.device_info_set.all()
  115. device_info_list = CommonService.qs_to_dict(device_info_query_set)
  116. # device_user关联到device_info
  117. sqlDict["datas"][k]['device_info'] = device_info_list
  118. if len(v['fields']['role']) > 0:
  119. role_query_set = Role.objects.get(rid=v['fields']['role'][0])
  120. sqlDict["datas"][k]['fields']['role'].append(role_query_set.roleName)
  121. sqlDict['count'] = device_user_count
  122. return response.json(0, sqlDict)
  123. else:
  124. return response.json(0, {"datas": ""})
  125. def searchUser(self, fieldDict, *args, **kwargs):
  126. response = ResponseObject()
  127. token = fieldDict.get('token', None)
  128. if token != None:
  129. tko = TokenObject(token)
  130. tko.valid()
  131. response.lang = tko.lang
  132. if tko.code == 0:
  133. fieldDict = fieldDict.dict()
  134. fieldDict.pop('token')
  135. type = fieldDict.get('type', None)
  136. if type == 'PC':
  137. return self.searchUserPCSQL(fieldDict, response)
  138. else:
  139. return self.searchUserSQL(fieldDict, response, args, kwargs)
  140. else:
  141. return response.json(tko.code)
  142. else:
  143. return response.json(444)
  144. class shareUserEquipmentView(View):
  145. @method_decorator(csrf_exempt)
  146. def dispatch(self, *args, **kwargs):
  147. return super(shareUserEquipmentView, self).dispatch(*args, **kwargs)
  148. def post(self, request, *args, **kwargs):
  149. request.encoding = 'utf-8'
  150. queryDict = request.POST
  151. return self.shareUser(queryDict, args, kwargs)
  152. def get(self, request, *args, **kwargs):
  153. request.encoding = 'gb2312'
  154. queryDict = request.GET
  155. return self.shareUser(queryDict, args, kwargs)
  156. def shareUser(self, queryDict, *args, **kwargs):
  157. token = queryDict.get('token', None)
  158. GuestID = queryDict.get('guestID', None)
  159. content = queryDict.get('content', None)
  160. sharedAll = queryDict.get('sharedAll', False)
  161. if sharedAll in ('1', '0'):
  162. sharedAll = bool(int(sharedAll))
  163. elif sharedAll in ('true', 'false'):
  164. if sharedAll == 'true':
  165. sharedAll = 1
  166. else:
  167. sharedAll = 0
  168. response = ResponseObject()
  169. if token != None and GuestID != None:
  170. tko = TokenObject(token)
  171. tko.valid()
  172. response.lang = tko.lang
  173. if tko.code == 0:
  174. MasterID = tko.userID
  175. if sharedAll and MasterID != None:
  176. return self.shareUserSQL(MasterID, GuestID, True, response, args, kwargs)
  177. if content != None and MasterID != None:
  178. return self.shareUserSQL(MasterID, GuestID, False, response, args, content=content)
  179. else:
  180. return response.json(804)
  181. else:
  182. return response.json(tko.code)
  183. else:
  184. return response.json(800)
  185. def shareUserSQL(self, MasterID, GuestID, sharedAll, response, *args, **kwargs):
  186. try:
  187. Guest = Device_User.objects.filter(userID=GuestID).order_by('-data_joined')
  188. Master = Device_Info.objects.filter(userID_id=MasterID).order_by('-data_joined')
  189. except Exception as e:
  190. errorInfo = traceback.format_exc()
  191. print('查询数据库错误: %s' % errorInfo)
  192. return response.json(500, {"details": repr(e)})
  193. else:
  194. if Master:
  195. if Guest:
  196. querysetList = []
  197. dictLen = 0
  198. if sharedAll:
  199. for equipment in Master:
  200. eqDict = equipment.model_to_dict(exclude=['id', 'data_joined', 'primaryUserID'])
  201. shareEquipment = Device_Info.objects.filter(userID_id=GuestID, primaryUserID= \
  202. MasterID, UID=eqDict.get('UID', None)).order_by('-data_joined')
  203. if shareEquipment:
  204. dictLen += 1
  205. continue
  206. else:
  207. eqDict['primaryUserID'] = eqDict.pop('userID')
  208. eqDict['isShare'] = True
  209. eqDict['userID_id'] = GuestID
  210. eqDict['id'] = CommonService.getUserID(getUser=False)
  211. querysetList.append(Device_Info(**eqDict))
  212. else:
  213. content = kwargs.get('content', None)
  214. if content != None:
  215. contentDict = json.loads(content)
  216. uidlist = UID = contentDict.get('UID', None)
  217. print(uidlist)
  218. for equipment in Master:
  219. eqDict = equipment.model_to_dict(exclude=['id', 'data_joined', 'primaryUserID'])
  220. if eqDict['UID'] in uidlist:
  221. UID.remove(eqDict['UID'])
  222. shareEquipment = Device_Info.objects.filter(userID_id=GuestID, primaryUserID= \
  223. MasterID, UID=eqDict.get('UID', None)).order_by('-data_joined')
  224. if shareEquipment:
  225. dictLen += 1
  226. else:
  227. eqDict['primaryUserID'] = eqDict.pop('userID')
  228. eqDict['isShare'] = True
  229. eqDict['userID_id'] = GuestID
  230. eqDict['id'] = CommonService.getUserID(getUser=False)
  231. querysetList.append(Device_Info(**eqDict))
  232. else:
  233. continue
  234. if len(querysetList) == 0:
  235. if sharedAll:
  236. return response.json(160)
  237. else:
  238. if len(UID) > 0:
  239. return response.json(160, {'error_UID': UID})
  240. else:
  241. return response.json(160)
  242. else:
  243. try:
  244. equipmentCount = Device_Info.objects.bulk_create(querysetList)
  245. except Exception as e:
  246. errorInfo = traceback.format_exc()
  247. print('添加数据库记录错误: %s' % errorInfo)
  248. return response.json(500, {"details": repr(e)})
  249. else:
  250. if dictLen > 0:
  251. return response.json(0, {
  252. 'Shared': dictLen,
  253. 'Sharing': len(querysetList),
  254. 'errormsg': u'A part of the equipment has been shared!',
  255. })
  256. else:
  257. if sharedAll:
  258. return response.json(0, {
  259. 'Sharing': len(querysetList)
  260. })
  261. else:
  262. if len(UID) > 0:
  263. return response.json(0, {
  264. 'error_UID': UID,
  265. 'Sharing': len(querysetList),
  266. })
  267. else:
  268. return response.json(0, {
  269. 'Sharing': len(querysetList),
  270. })
  271. else:
  272. return response.json(113)
  273. else:
  274. return response.json(172)
  275. class unsharedUserEquipmentView(View):
  276. @method_decorator(csrf_exempt)
  277. def dispatch(self, *args, **kwargs):
  278. return super(unsharedUserEquipmentView, self).dispatch(*args, **kwargs)
  279. def post(self, request, *args, **kwargs):
  280. request.encoding = 'utf-8'
  281. queryset = request.POST
  282. return self.unsharedUserEquipment(queryset, args, kwargs)
  283. def get(self, request, *args, **kwargs):
  284. request.encoding = 'gb2312'
  285. queryset = request.GET
  286. return self.unsharedUserEquipment(queryset, args, kwargs)
  287. def unsharedUserEquipment(self, queryset, *args, **kwargs):
  288. token = queryset.get('token', None)
  289. GuestID = queryset.get('guestID', None)
  290. content = queryset.get('content', None)
  291. unsharedAll = queryset.get('unsharedAll', False)
  292. response = ResponseObject()
  293. if unsharedAll in ('1', '0'):
  294. unsharedAll = bool(int(unsharedAll))
  295. elif unsharedAll in ('true', 'false'):
  296. if unsharedAll == 'true':
  297. unsharedAll = 1
  298. else:
  299. unsharedAll = 0
  300. if token != None and GuestID != None and len(GuestID) > 0:
  301. tko = TokenObject(token)
  302. tko.valid()
  303. response.lang = tko.lang
  304. if tko.code == 0:
  305. MasterID = tko.userID
  306. if unsharedAll and MasterID != None:
  307. return self.unsharedUserEquipmentSQL(MasterID, GuestID, True, response, args, kwargs)
  308. else:
  309. if content != None and MasterID != None:
  310. return self.unsharedUserEquipmentSQL(MasterID, GuestID, False, response, args, content=content)
  311. else:
  312. return response.json(805)
  313. else:
  314. return response.json(tko.code)
  315. else:
  316. return response.json(800)
  317. def unsharedUserEquipmentSQL(self, MasterID, GuestID, unsharedAll, response, *args, **kwargs):
  318. if unsharedAll:
  319. try:
  320. equipmentCount = Device_Info.objects.filter(userID_id=GuestID, primaryUserID=MasterID).delete()
  321. except Exception as e:
  322. errorInfo = traceback.format_exc()
  323. print('删除设备更新数据库错误: %s' % errorInfo)
  324. return response.json(171, {"details": repr(e)})
  325. else:
  326. return response.json(0, {'removeCount': equipmentCount[0]})
  327. else:
  328. content = kwargs.get('content', None)
  329. if content != None:
  330. removeCount = 0
  331. errorRemove = []
  332. errorUID = []
  333. contentDict = json.loads(content)
  334. uidlist = contentDict.get('UID', None)
  335. for index in range(len(uidlist)):
  336. uid = uidlist[index]
  337. try:
  338. equipment = Device_Info.objects.filter(userID_id=GuestID, primaryUserID=MasterID, UID=uid)
  339. if equipment:
  340. equipmentCount = equipment.delete()
  341. else:
  342. errorUID.append(uid)
  343. continue
  344. except Exception as e:
  345. errorInfo = traceback.format_exc()
  346. print('查询数据库错误: %s' % errorInfo)
  347. errorRemove.append(uid)
  348. continue
  349. else:
  350. removeCount += equipmentCount[0]
  351. if len(errorRemove) > 0:
  352. return response.json(171, {'removeCount': removeCount,'error_UID': errorRemove})
  353. else:
  354. if len(errorUID) > 0:
  355. return response.json(173, {'removeCount': removeCount,'error_UID': errorUID})
  356. else:
  357. return response.json(0, {'removeCount': removeCount})