RegionCountryController.py 2.2 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667
  1. #!/usr/bin/env python3
  2. # -*- coding: utf-8 -*-
  3. import time
  4. from django.views import View
  5. from Model.models import Device_User, RegionCountryModel
  6. from Object.ResponseObject import ResponseObject
  7. from Object.TokenObject import TokenObject
  8. import json
  9. class RegionCountryView(View):
  10. def get(self, request, *args, **kwargs):
  11. request.encoding = 'utf-8'
  12. operation = kwargs.get('operation', None)
  13. request_dict = request.GET
  14. return self.validate(request_dict, operation)
  15. def post(self, request, *args, **kwargs):
  16. request.encoding = 'utf-8'
  17. operation = kwargs.get('operation', None)
  18. request_dict = request.POST
  19. return self.validate(request_dict, operation)
  20. def validate(self, request_dict, operation):
  21. token = TokenObject(request_dict.get('token', None))
  22. response = ResponseObject()
  23. if token.code != 0:
  24. return response.json(token.code)
  25. if operation == 'initUserRegion':
  26. return self.do_initUserRegion(token.userID, request_dict, response)
  27. elif operation == 'initRegionDate':
  28. return self.do_initRegionDate(request_dict, response)
  29. else:
  30. return response.json(404)
  31. def do_initUserRegion(self, userID, request_dict, response):
  32. user_qs = Device_User.objects.filter(userID=userID)
  33. number = request_dict.get('number', None)
  34. if not user_qs.exists():
  35. return response.json(309)
  36. user_qs[0].update(regionCountry = number)
  37. def do_initRegionDate(self, request_dict, response):
  38. filepath = request_dict.get('filepath')
  39. with open(filepath, 'r', encoding='UTF-8') as f:
  40. region_list = json.load(f)
  41. #region_list = json.loads(fobj)
  42. now_time = int(time.time())
  43. for region in region_list:
  44. region_id = region['region_id']
  45. countries_list = region['countries']
  46. for countries in countries_list:
  47. cn = countries['cn']
  48. en = countries['en']
  49. number = countries['number']
  50. name = countries['name']
  51. RegionCountryModel.objects.create(number=number, region_id=region_id, name=name, cn=cn, en=en, add_time=now_time, update_time=now_time)