File size: 6,896 Bytes
e657e99 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 | from django.db import models
from django.contrib.auth.models import User
import uuid
import os
import json
def get_file_path(instance, filename):
"""为上传的文件生成唯一的文件路径"""
ext = filename.split('.')[-1]
filename = f"{uuid.uuid4()}.{ext}"
return os.path.join(f'users/{instance.project.user.id}/projects/{instance.project.id}/data/', filename)
def get_export_file_path(instance, filename):
"""为导出的文件生成唯一的文件路径"""
ext = filename.split('.')[-1]
filename = f"{uuid.uuid4()}.{ext}"
return os.path.join(f'users/{instance.project.user.id}/projects/{instance.project.id}/exports/', filename)
# class ThirdPartyUserMapping(models.Model):
# """第三方用户映射模型"""
# user = models.OneToOneField(User, on_delete=models.CASCADE, related_name='third_party_mapping')
# third_party_user_id = models.IntegerField('第三方用户ID', unique=True)
# third_party_username = models.CharField('第三方用户名', max_length=255)
# created_at = models.DateTimeField('创建时间', auto_now_add=True)
# updated_at = models.DateTimeField('更新时间', auto_now=True)
# class Meta:
# verbose_name = '第三方用户映射'
# verbose_name_plural = '第三方用户映射'
# def __str__(self):
# return f"{self.third_party_username} ({self.third_party_user_id}) -> {self.user.username}"
# class TokenStorage(models.Model):
# """Token存储模型"""
# token = models.CharField('Token', max_length=512, unique=True)
# user = models.ForeignKey(User, on_delete=models.CASCADE, related_name='tokens')
# third_party_user_id = models.IntegerField('第三方用户ID')
# third_party_username = models.CharField('第三方用户名', max_length=255)
# created_at = models.DateTimeField('创建时间', auto_now_add=True)
# last_used = models.DateTimeField('最后使用时间', auto_now=True)
# expires_at = models.DateTimeField('过期时间', null=True, blank=True)
# class Meta:
# verbose_name = 'Token存储'
# verbose_name_plural = 'Token存储'
# indexes = [
# models.Index(fields=['token']),
# models.Index(fields=['third_party_user_id']),
# ]
# def __str__(self):
# return f"Token for {self.third_party_username} ({self.third_party_user_id})"
class Project(models.Model):
"""项目模型"""
STATUS_CHOICES = (
('initialized', '已初始化'),
('data_loaded', '数据已加载'),
('preprocessing', '数据预处理中'),
('preprocessing_completed', '数据预处理完成'),
('analysis', '分析中'),
('completed', '已完成'),
('error', '出错'),
)
id = models.UUIDField(primary_key=True, default=uuid.uuid4, editable=False)
name = models.CharField('项目名称', max_length=200)
description = models.TextField('项目描述', blank=True)
user = models.ForeignKey(User, on_delete=models.CASCADE, related_name='projects')
created_at = models.DateTimeField('创建时间', auto_now_add=True)
last_modified = models.DateTimeField('最后修改时间', auto_now=True)
status = models.CharField('项目状态', max_length=50, choices=STATUS_CHOICES, default='initialized')
active_data_file = models.ForeignKey('DataFile', on_delete=models.SET_NULL, related_name='active_in_projects', null=True, blank=True, verbose_name='当前活动数据文件')
class Meta:
verbose_name = '项目'
verbose_name_plural = '项目'
def __str__(self):
return self.name
class DataFile(models.Model):
"""数据文件模型"""
project = models.ForeignKey(Project, on_delete=models.CASCADE, related_name='data_files')
file = models.FileField('文件', upload_to=get_file_path)
filename = models.CharField('文件名', max_length=255)
uploaded_at = models.DateTimeField('上传时间', auto_now_add=True)
wavenumber_min = models.FloatField('最小波数', null=True, blank=True)
wavenumber_max = models.FloatField('最大波数', null=True, blank=True)
group_index = models.IntegerField('分组索引', default=2, null=True, blank=True)
class Meta:
verbose_name = '数据文件'
verbose_name_plural = '数据文件'
def __str__(self):
return self.filename
class PreprocessingTask(models.Model):
"""预处理任务模型"""
project = models.ForeignKey(Project, on_delete=models.CASCADE, related_name='preprocessing_tasks')
created_at = models.DateTimeField('创建时间', auto_now_add=True)
completed_at = models.DateTimeField('完成时间', null=True, blank=True)
parameters = models.JSONField('参数', default=dict)
status = models.CharField('状态', max_length=50, default='pending')
result = models.JSONField('结果', null=True, blank=True)
class Meta:
verbose_name = '预处理任务'
verbose_name_plural = '预处理任务'
def __str__(self):
return f"{self.project.name} - 预处理任务 {self.id}"
class AnalysisTask(models.Model):
"""分析任务模型"""
ANALYSIS_TYPE_CHOICES = (
('meta_free', '无元数据分析'),
('meta_based', '基于元数据分析'),
)
project = models.ForeignKey(Project, on_delete=models.CASCADE, related_name='analysis_tasks')
analysis_type = models.CharField('分析类型', max_length=20, choices=ANALYSIS_TYPE_CHOICES)
created_at = models.DateTimeField('创建时间', auto_now_add=True)
completed_at = models.DateTimeField('完成时间', null=True, blank=True)
parameters = models.JSONField('参数', default=dict)
status = models.CharField('状态', max_length=50, default='pending')
result = models.JSONField('结果', null=True, blank=True)
class Meta:
verbose_name = '分析任务'
verbose_name_plural = '分析任务'
def __str__(self):
return f"{self.project.name} - {self.get_analysis_type_display()} 任务 {self.id}"
class ExportTask(models.Model):
"""导出任务模型"""
project = models.ForeignKey(Project, on_delete=models.CASCADE, related_name='export_tasks')
created_at = models.DateTimeField('创建时间', auto_now_add=True)
completed_at = models.DateTimeField('完成时间', null=True, blank=True)
format = models.CharField('格式', max_length=10)
components = models.JSONField('组件', default=list)
status = models.CharField('状态', max_length=50, default='pending')
file = models.FileField('导出文件', upload_to=get_export_file_path, null=True, blank=True)
class Meta:
verbose_name = '导出任务'
verbose_name_plural = '导出任务'
def __str__(self):
return f"{self.project.name} - 导出任务 {self.id}"
|