models.py 15 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366
  1. from itertools import chain
  2. from django.contrib.auth.models import BaseUserManager, AbstractBaseUser
  3. from django.db import models
  4. # from django.utils import six
  5. from django.utils.encoding import python_2_unicode_compatible
  6. from imagekit.models import ProcessedImageField
  7. from imagekit.processors import ResizeToFill
  8. import six
  9. from AnsjerPush.config import SERVER_DOMAIN
  10. class PermissionsManager(models.Manager):
  11. use_in_migrations = True
  12. def get_by_natural_key(self, permName):
  13. return self.get(
  14. permName=permName
  15. )
  16. class RoleManager(models.Manager):
  17. """
  18. The manager for the auth's Role model.
  19. """
  20. use_in_migrations = True
  21. def get_by_natural_key(self, roleName):
  22. return self.get(roleName=roleName)
  23. class UserManager(BaseUserManager):
  24. def create_user(self, username, password, userID, is_active,
  25. user_isValid, **extra_fields):
  26. user = self.model(
  27. username=username,
  28. userID=userID,
  29. is_active=is_active,
  30. user_isValid=user_isValid,
  31. **extra_fields
  32. )
  33. user.set_password(password)
  34. user.save(using=self._db)
  35. role = Role.objects.get(rid=1)
  36. user.role.add(role)
  37. return user
  38. def create_superuser(self, username, password, userID, is_active,
  39. user_isValid, **extra_fields):
  40. # extra_fields.setdefault('is_superuser', is_superuser)
  41. is_superuser = extra_fields.get('is_superuser', None)
  42. if is_superuser != 100 and is_superuser != 1:
  43. raise ValueError('Superuser must have is_superuser=1 or 100.')
  44. return self.create_user(username, password, userID, is_active,
  45. user_isValid, **extra_fields)
  46. @python_2_unicode_compatible
  47. class Permissions(models.Model):
  48. permName = models.CharField(blank=True, null=True, max_length=32, unique=True, verbose_name=u'权限名称')
  49. description = models.CharField(blank=True, null=True, max_length=128, verbose_name=u'描述信息', default='')
  50. objects = PermissionsManager()
  51. def __str__(self):
  52. return "%s" % (
  53. six.text_type(self.description))
  54. class Meta:
  55. ordering = ['permName']
  56. db_table = 'permissions'
  57. verbose_name = u'role permission'
  58. verbose_name_plural = verbose_name
  59. unique_together = (('permName'),)
  60. def natural_key(self):
  61. return (self.permName)
  62. class Role(models.Model):
  63. rid = models.SmallIntegerField(primary_key=True, unique=True, verbose_name=u'用户角色组ID')
  64. roleName = models.CharField(max_length=32, unique=True,
  65. default='User', verbose_name=u'角色名称')
  66. permission = models.ManyToManyField(to='Permissions', blank=True, verbose_name=u'权限', db_table='role_permissions')
  67. Description = models.TextField(blank=True, default='', verbose_name=u'描述信息')
  68. objects = RoleManager()
  69. class Meta:
  70. ordering = ['roleName']
  71. db_table = 'role'
  72. verbose_name = u'用户角色'
  73. verbose_name_plural = verbose_name
  74. def __str__(self):
  75. return self.roleName
  76. def natural_key(self):
  77. return (self.roleName,)
  78. def get_all_permission(self):
  79. perms = self.permission.all()
  80. permslist = []
  81. if self.rid == 100:
  82. perms = Permissions.objects.all()
  83. for perm in perms:
  84. permslist.append(perm.permName)
  85. return permslist
  86. else:
  87. permlist_tmp = []
  88. for perm in perms:
  89. permlist_tmp.append(perm.permName)
  90. permSet = set(permlist_tmp)
  91. permslist = list(permSet)
  92. permslist.sort()
  93. return permslist
  94. class Device_User(AbstractBaseUser):
  95. userID = models.CharField(blank=True, max_length=32, primary_key=True,
  96. verbose_name=u'用户ID', unique=True)
  97. role = models.ManyToManyField(to='Role', blank=True, verbose_name=u'用户角色', db_table='user_role')
  98. username = models.CharField(max_length=64, verbose_name=u'用户名', default='', blank=True)
  99. password = models.CharField(max_length=128, verbose_name=u'密码')
  100. userEmail = models.EmailField(max_length=64, verbose_name=u'邮箱', default='', blank=True)
  101. # 实际的路径就是 MEDIA_ROOT/Image/filename,所以可用upload_to来指定文件存放的前缀路径
  102. userIconPath = ProcessedImageField(blank=True, upload_to='User/Images/', default='static/User/default.png',
  103. verbose_name=u'头像',
  104. # 图片将处理成85x85的尺寸
  105. processors=[ResizeToFill(85, 85)], )
  106. userIconUrl = models.URLField(blank=True, max_length=128, default=SERVER_DOMAIN
  107. + 'account/getAvatar/User/defaultUser.png')
  108. NickName = models.CharField(blank=True, max_length=64, default='', verbose_name=u'用户昵称')
  109. is_superuser = models.IntegerField(blank=True, default=0, verbose_name=u'用户类型')
  110. is_active = models.BooleanField(blank=True, default=False, verbose_name=u'用户活动状态')
  111. data_joined = models.DateTimeField(blank=True, verbose_name=u'加入时间', auto_now_add=True)
  112. last_login = models.DateTimeField(blank=True, verbose_name='u最后登录时间', auto_now=True)
  113. user_isValid = models.BooleanField(blank=True, default=False, verbose_name=u'用户有效性', help_text=u'是否是认证通过的用户')
  114. online = models.BooleanField(blank=True, default=False, verbose_name=u'用户登录状态')
  115. machine_code = models.CharField(blank=True, max_length=128, default='', verbose_name=u'机械码')
  116. language = models.CharField(blank=True, max_length=16, default='en', verbose_name=u'语言地区')
  117. # 手机注册增加字段
  118. phone = models.CharField(max_length=16, verbose_name=u'手机号', default='', blank=True)
  119. objects = UserManager()
  120. USERNAME_FIELD = 'userID' # 必须有一个唯一标识
  121. REQUIRED_FIELDS = ['is_superuser'] # 创建superuser时的必须字段
  122. class Meta:
  123. ordering = ('-data_joined',)
  124. verbose_name = u'用户信息'
  125. db_table = 'device_user'
  126. get_latest_by = 'last_login'
  127. def __str__(self):
  128. return self.username
  129. def get_role_id(self):
  130. rids = []
  131. roles = self.role.all()
  132. for role in roles:
  133. rids.append(role.rid)
  134. return rids
  135. def get_all_permission(self):
  136. roles = self.role.all()
  137. perms = self.permission.all()
  138. permslist = []
  139. for role in roles:
  140. if role.rid == 100:
  141. perms = Permissions.objects.all()
  142. for perm in perms:
  143. permslist.append(perm.permName)
  144. return permslist
  145. for perm in perms:
  146. permslist.append(perm.permName)
  147. permSet = set(permslist)
  148. for role in roles:
  149. permlist_tmp = []
  150. for perm in role.permission.all():
  151. permlist_tmp.append(perm.permName)
  152. permSet_tmp = set(permlist_tmp)
  153. permSet = permSet.union(permSet_tmp)
  154. permslist = list(permSet)
  155. permslist.sort()
  156. return permslist
  157. @property
  158. def is_staff(self):
  159. return self.is_superuser
  160. class Equipment_Info(models.Model):
  161. id = models.AutoField(primary_key=True, verbose_name=u'自增标记ID')
  162. devUid = models.CharField(default='', blank=True, max_length=32, verbose_name=u'设备ID')
  163. devNickName = models.CharField(blank=True, max_length=32, default='', verbose_name=u'设备昵称')
  164. Channel = models.IntegerField(default=1, blank=True, verbose_name=u'设备通道')
  165. eventType = models.IntegerField(default=0, blank=True, verbose_name=u'事件类型')
  166. status = models.BooleanField(blank=True, default=False, verbose_name=u'事件状态')
  167. alarm = models.CharField(blank=True, max_length=256, verbose_name=u'报警信息')
  168. eventTime = models.CharField(blank=True, default='', max_length=16, verbose_name=u'设备报警时间')
  169. receiveTime = models.CharField(blank=True, default='', max_length=16, verbose_name=u'接收到报警时间')
  170. userID = models.ForeignKey(Device_User, null=True, blank=True, to_field='userID', on_delete=models.CASCADE)
  171. is_st = models.SmallIntegerField(default=0, verbose_name='是否截图') # 0 否,1 是图,2,视频
  172. addTime = models.IntegerField(verbose_name='添加时间', default=0)
  173. def __str__(self):
  174. return self.id
  175. class Meta:
  176. db_table = 'equipment_info'
  177. verbose_name = u'设备信息推送表'
  178. verbose_name_plural = verbose_name
  179. ordering = ('-id',)
  180. # 存储通表
  181. class VodBucketModel(models.Model):
  182. id = models.AutoField(primary_key=True, verbose_name='主键')
  183. bucket = models.CharField(max_length=30, verbose_name='设备UID')
  184. storeDay = models.IntegerField(default=0, verbose_name='存储生命周期(天)')
  185. content = models.TextField(verbose_name='描述')
  186. endpoint = models.CharField(max_length=125, default='', verbose_name='存储节点')
  187. area = models.CharField(max_length=16, default='', verbose_name='区域')
  188. region = models.CharField(max_length=16, default='', verbose_name='regionID')
  189. addTime = models.IntegerField(verbose_name='添加时间', default=0)
  190. updTime = models.IntegerField(verbose_name='更新时间', default=0)
  191. def __str__(self):
  192. return self.id
  193. class Meta:
  194. db_table = 'vod_bucket'
  195. verbose_name = u'存储通信息'
  196. verbose_name_plural = verbose_name
  197. ordering = ('-id',)
  198. class VodHlsModel(models.Model):
  199. id = models.AutoField(primary_key=True, verbose_name='回放列表主键')
  200. uid = models.CharField(max_length=20, verbose_name='设备UID')
  201. channel = models.SmallIntegerField(default=0, verbose_name='通道')
  202. time = models.IntegerField(verbose_name='播放列表名字时间戳', default=0, db_index=True)
  203. endTime = models.IntegerField(verbose_name='删除时间', default=0)
  204. sec = models.IntegerField(verbose_name='秒数', default=0)
  205. bucket = models.ForeignKey(VodBucketModel, blank=True, to_field='id', on_delete=models.CASCADE, default=1,
  206. verbose_name='存储空间')
  207. def __str__(self):
  208. return self.id
  209. class Meta:
  210. db_table = 'vod_hls'
  211. verbose_name = u'云存回放信息表'
  212. verbose_name_plural = verbose_name
  213. ordering = ('-id',)
  214. class OssCrdModel(models.Model):
  215. id = models.AutoField(primary_key=True, verbose_name='主键')
  216. uid = models.CharField(max_length=20, verbose_name='设备UID')
  217. channel = models.SmallIntegerField(default=0, verbose_name='通道')
  218. data = models.TextField(verbose_name='设备sts')
  219. addTime = models.IntegerField(verbose_name='添加时间', default=0)
  220. bucket = models.ForeignKey(VodBucketModel, blank=True, to_field='id', on_delete=models.CASCADE, default=1,
  221. verbose_name='存储空间')
  222. def __str__(self):
  223. return self.id
  224. class Meta:
  225. db_table = 'osscrd'
  226. verbose_name = u'设备证书'
  227. verbose_name_plural = verbose_name
  228. ordering = ('-id',)
  229. class UID_Bucket(models.Model):
  230. id = models.AutoField(primary_key=True, verbose_name=u'自增标记ID')
  231. uid = models.CharField(max_length=20, verbose_name='设备UID')
  232. channel = models.SmallIntegerField(default=0, verbose_name='通道')
  233. bucket = models.ForeignKey(VodBucketModel, blank=True, to_field='id', on_delete=models.CASCADE, verbose_name='存储空间')
  234. status = models.SmallIntegerField(default=0, verbose_name='状态[0:开启,1:关闭]')
  235. endTime = models.IntegerField(verbose_name='结束时间', default=0)
  236. class Meta:
  237. db_table = 'vod_uid_bucket'
  238. verbose_name = u'设备关联套餐表'
  239. verbose_name_plural = verbose_name
  240. ordering = ('id',)
  241. class UID_Preview(models.Model):
  242. id = models.AutoField(primary_key=True, verbose_name=u'自增标记ID')
  243. uid = models.CharField(default='', max_length=20, db_index=True, verbose_name='设备UID')
  244. channel = models.SmallIntegerField(verbose_name=u'通道号', default=0)
  245. addTime = models.IntegerField(verbose_name='添加时间', default=0)
  246. updTime = models.IntegerField(verbose_name='更新时间', default=0)
  247. class Meta:
  248. db_table = 'uid_preview'
  249. verbose_name = '设备预览图'
  250. verbose_name_plural = verbose_name
  251. ordering = ('id',)
  252. # 设备推送重构
  253. # 设备配置表,新
  254. class UidSetModel(models.Model):
  255. id = models.AutoField(primary_key=True, verbose_name=u'自增标记ID')
  256. uid = models.CharField(max_length=20, db_index=True, unique=True, verbose_name='设备UID')
  257. channel = models.SmallIntegerField(default=0, verbose_name='通道数量') #
  258. detect_status = models.SmallIntegerField(default=0, verbose_name='推送开关') # 状态[0:关闭,1:开启,2:用户解绑]
  259. detect_interval = models.IntegerField(verbose_name='推送间隔', default=60) # 秒
  260. addTime = models.IntegerField(verbose_name='添加时间', default=0)
  261. updTime = models.IntegerField(verbose_name='更新时间', default=0)
  262. ucode = models.CharField(max_length=32, verbose_name='设备产品码', default='')
  263. version = models.CharField(max_length=32, verbose_name='设备版本', default='')
  264. p2p_region = models.CharField(max_length=16, verbose_name='设备区域', default='ALL') # ALL CN EU US
  265. cloud_vod = models.SmallIntegerField(default=2, verbose_name='云存开关') # 0,关闭,1开启,2,不支持
  266. tz = models.CharField(default='', max_length=16, verbose_name='设备时区') # +8
  267. video_code = models.SmallIntegerField(default=0, verbose_name='编码类型') # 0:264,1:265
  268. nickname = models.CharField(default='', max_length=64, verbose_name='设备昵称')
  269. ip = models.CharField(max_length=20, default='', verbose_name=u'设备ip')
  270. pwd = models.CharField(max_length=32, default='', verbose_name=u'设备密码')
  271. # 设备重置后第一次启动时间
  272. resetTime = models.IntegerField(default=0, verbose_name='设备重置时间')
  273. class Meta:
  274. db_table = 'uid_set'
  275. verbose_name = u'设备配置表'
  276. verbose_name_plural = verbose_name
  277. ordering = ('id',)
  278. # 设备关联用户推送
  279. class UidPushModel(models.Model):
  280. id = models.AutoField(primary_key=True, verbose_name='自增id')
  281. userID = models.ForeignKey(Device_User, to_field='userID', on_delete=models.CASCADE)
  282. uid_set = models.ForeignKey(UidSetModel, to_field='id', on_delete=models.CASCADE)
  283. appBundleId = models.CharField(blank=True, max_length=32, verbose_name=u'appID')
  284. app_type = models.IntegerField(default=0, verbose_name=u'app类型 1:ios,2:安卓')
  285. push_type = models.IntegerField(default=0, verbose_name=u'推送类型') # 0,apns 1,安卓gcm 2,极光
  286. token_val = models.CharField(default='', max_length=160, verbose_name=u'设备验证令牌')
  287. m_code = models.CharField(default='', max_length=64, verbose_name='手机唯一标识')
  288. addTime = models.IntegerField(verbose_name='添加时间', default=0)
  289. updTime = models.IntegerField(verbose_name='更新时间', default=0)
  290. lang = models.CharField(max_length=8, verbose_name='推送语言类型', default='en') # en英文 cn中文
  291. tz = models.CharField(max_length=8, verbose_name='utc时区', default='0')
  292. class Meta:
  293. db_table = 'uid_push'
  294. verbose_name = '设备绑定'
  295. verbose_name_plural = verbose_name
  296. ordering = ('-id',)