from django.db import models
from django.contrib.auth.models import User

# 文集模型
class Project(models.Model):
 name = models.CharField(verbose_name="文档名称",max_length=50)
 intro = models.TextField(verbose_name="介绍")
 # 文集权限说明:0表示公开,1表示私密,2表示指定用户可见,3表示访问码可见 默认公开
 role = models.IntegerField(choices=((0,0),(1,1),(2,2),(3,3)), default=0,verbose_name="文集权限")
 role_value = models.TextField(verbose_name="文集权限值",blank=True,null=True)
 create_user = models.ForeignKey(User,on_delete=models.CASCADE)
 create_time = models.DateTimeField(auto_now_add=True)
 modify_time = models.DateTimeField(auto_now=True)

 def __str__(self):
 return self.name

 class Meta:
 verbose_name = '文集'
 verbose_name_plural = verbose_name

 def get_absolute_url(self):
 from django.urls import reverse
 return reverse("pro_index",
 kwargs={
 "pro_id":self.pk}
 )

# 文集协作模型
class ProjectCollaborator(models.Model):
 project = models.ForeignKey(Project,on_delete=models.CASCADE) # 文集
 user = models.ForeignKey(User,on_delete=models.CASCADE) # 用户
 # 用户的协作模式:0表示可新建文档,可修改、删除自己新建的文档,1表示可新建文档,可删除自己创建的文档、可修改所有文档
 role = models.IntegerField(choices=((0,0),(1,1)),default=0,verbose_name='协作模式')
 create_time = models.DateTimeField(auto_now=True,verbose_name='添加时间')
 modify_time = models.DateTimeField(auto_now_add=True,verbose_name='修改时间')

 def __str__(self):
 return self.project

 class Meta:
 verbose_name = '文集协作'
 verbose_name_plural = verbose_name

# 文档模型
class Doc(models.Model):
 name = models.CharField(verbose_name="文档标题",max_length=50)
 pre_content = models.TextField(verbose_name="编辑内容",null=True,blank=True)
 content = models.TextField(verbose_name="文档内容",null=True,blank=True)
 parent_doc = models.IntegerField(default=0,verbose_name="上级文档")
 top_doc = models.IntegerField(default=0,verbose_name="所属项目")
 sort = models.IntegerField(verbose_name='排序',default=99)
 create_user = models.ForeignKey(User,on_delete=models.CASCADE)
 create_time = models.DateTimeField(auto_now_add=True)
 modify_time = models.DateTimeField(auto_now=True)
 # 文档状态说明:0表示草稿状态,1表示发布状态,2表示删除状态
 status = models.IntegerField(choices=((0,0),(1,1)),default=1,verbose_name='文档状态')

 def __str__(self):
 return self.name

 class Meta:
 verbose_name = '文档'
 verbose_name_plural = verbose_name
 indexes = [
 models.Index(fields=['top_doc','parent_doc','status']),
 models.Index(fields=['sort']),
 ]
 # ordering = ['-create_time','sort']

 def get_absolute_url(self):
 from django.urls import reverse
 return reverse("doc",
 kwargs={
 "pro_id": str(self.top_doc),
 "doc_id":self.pk}
 )

# 文档历史模型
class DocHistory(models.Model):
 doc = models.ForeignKey(Doc,on_delete=models.CASCADE)
 pre_content = models.TextField(verbose_name='文档历史编辑内容',null=True,blank=True)
 create_user = models.ForeignKey(User,on_delete=models.SET_NULL,null=True)
 create_time = models.DateTimeField(auto_now=True)

 def __str__(self):
 return self.doc

 class Meta:
 verbose_name = '文档历史'
 verbose_name_plural = verbose_name

# 文档模板模型
class DocTemp(models.Model):
 name = models.CharField(verbose_name="模板名称",max_length=50)
 content = models.TextField(verbose_name="文档模板")
 create_user = models.ForeignKey(User,on_delete=models.CASCADE)
 create_time = models.DateTimeField(auto_now_add=True)
 modify_time = models.DateTimeField(auto_now=True)

 def __str__(self):
 return self.name

 class Meta:
 verbose_name = '文档模板'
 verbose_name_plural = verbose_name

# 文集导出模型
class ProjectReport(models.Model):
 project = models.OneToOneField(Project,unique=True,on_delete=models.CASCADE)
 # 允许导出,默认为0-允许,1-不允许
 allow_epub = models.IntegerField(default=0,verbose_name="前台导出EPUB")
 allow_pdf = models.IntegerField(default=0, verbose_name="前台导出PDF")

 def __str__(self):
 return self.project.name

 class Meta:
 verbose_name = '文集导出'
 verbose_name_plural = verbose_name


# 文集导出文集模型
class ProjectReportFile(models.Model):
 project = models.ForeignKey(Project, on_delete=models.CASCADE) # 外键关联文集
 file_type = models.CharField(choices=(('epub', 'epub'), ('pdf', 'pdf'), ('docx', 'docx')), verbose_name='文件类型',max_length=10)
 file_name = models.CharField(max_length=100, verbose_name='文件名称')
 file_path = models.CharField(max_length=250, verbose_name='文件路径')
 create_time = models.DateTimeField(auto_now_add=True)

 def __str__(self):
 return self.file_name

 class Meta:
 verbose_name = '附件管理'
 verbose_name_plural = verbose_name

# 图片分组模型
class ImageGroup(models.Model):
 user = models.ForeignKey(User,on_delete=models.CASCADE)
 group_name = models.CharField(verbose_name="图片分组",max_length=50,default="默认分组")

 def __str__(self):
 return self.group_name

 class Meta:
 verbose_name = '图片分组'
 verbose_name_plural = verbose_name

# 图片模型
class Image(models.Model):
 user = models.ForeignKey(User,on_delete=models.CASCADE)
 file_path = models.CharField(verbose_name="图片路径",max_length=250)
 file_name = models.CharField(verbose_name="图片名称",max_length=250,null=True,blank=True)
 group = models.ForeignKey(ImageGroup,on_delete=models.SET_NULL,null=True,verbose_name="图片分组")
 remark = models.CharField(verbose_name="图片备注",null=True,blank=True,max_length=250,default="图片描述")
 create_time = models.DateTimeField(verbose_name='创建时间',auto_now_add=True)
 modify_time = models.DateTimeField(verbose_name='修改时间',auto_now=True)

 class Meta:
 verbose_name = '素材图片'
 verbose_name_plural = verbose_name


# 附件模型
class Attachment(models.Model):
 file_name = models.CharField(max_length=200,verbose_name="附件名",default='mrdoc_附件.zip')
 file_size = models.CharField(max_length=100,verbose_name="附件大小",blank=True,null=True)
 file_path = models.FileField(upload_to='attachment/%Y/%m/',verbose_name='附件')
 user = models.ForeignKey(User,on_delete=models.CASCADE,)
 create_time = models.DateTimeField(auto_now_add=True)

 def __str__(self):
 return self.file_name

 class Meta:
 verbose_name = '附件管理'
 verbose_name_plural = verbose_name