54 lines
1.9 KiB
Python
54 lines
1.9 KiB
Python
from django.db import models
|
||
|
||
class Dataset(models.Model):
|
||
# 自增主键字段，Django 默认为每个模型提供一个 id 字段
|
||
# 但是如果你需要自定义主键字段，可以用 `primary_key=True` 显式声明
|
||
id = models.AutoField(primary_key=True) # 自增主键
|
||
|
||
# 数据集名称，最长255个字符，不能为空
|
||
name = models.CharField(max_length=255)
|
||
|
||
# 创建该数据集的用户，最长100个字符，不能为空
|
||
user = models.CharField(max_length=100)
|
||
|
||
# 任务类型，最长50个字符，不能为空
|
||
task_type = models.CharField(max_length=16)
|
||
|
||
# 数据集大小，使用BigIntegerField来表示较大的整数
|
||
size = models.CharField(max_length=64)
|
||
|
||
# 数据集的总图片数
|
||
number = models.IntegerField()
|
||
|
||
#数据集的描述
|
||
description = models.TextField(default="")
|
||
|
||
# 创建时间，默认为当前时间戳
|
||
create_time = models.DateTimeField(auto_now_add=True) # 自动设置为当前时间（只在创建时）
|
||
|
||
# 数据集的类别，使用 JSONField 存储类别信息
|
||
categories = models.JSONField(default=list) # 默认值为一个空列表
|
||
|
||
# 该数据集是否上传至minio的一个标识
|
||
is_upload = models.BooleanField(default=False) # 表示数据集是否已上传，默认为 False
|
||
|
||
def update_is_upload(self, upload_status):
|
||
"""更新数据集的上传状态"""
|
||
self.is_upload = upload_status
|
||
self.save() # 保存更改到数据库
|
||
|
||
def __str__(self):
|
||
return self.name # 返回数据集名称作为对象的字符串表示
|
||
|
||
|
||
class Project(models.Model):
|
||
id = models.AutoField(primary_key=True)
|
||
name = models.CharField(max_length=255)
|
||
user = models.CharField(max_length=100)
|
||
task_type = models.CharField(max_length=16)
|
||
description = models.TextField(default="")
|
||
create_time = models.DateTimeField(auto_now_add=True)
|
||
|
||
def __str__(self):
|
||
return self.name
|