92 lines
3.5 KiB
Python
92 lines
3.5 KiB
Python
from flask_sqlalchemy import SQLAlchemy
|
|
from datetime import datetime
|
|
from sqlalchemy import func
|
|
from werkzeug.security import generate_password_hash, check_password_hash
|
|
|
|
db = SQLAlchemy()
|
|
|
|
class User(db.Model):
|
|
"""用户模型"""
|
|
__tablename__ = 'users'
|
|
|
|
id = db.Column(db.Integer, primary_key=True)
|
|
username = db.Column(db.String(80), unique=True, nullable=False)
|
|
password_hash = db.Column(db.String(200), nullable=False)
|
|
created_at = db.Column(db.DateTime, default=datetime.utcnow)
|
|
|
|
def set_password(self, password):
|
|
"""设置密码（哈希加密）"""
|
|
self.password_hash = generate_password_hash(password)
|
|
|
|
def check_password(self, password):
|
|
"""验证密码"""
|
|
return check_password_hash(self.password_hash, password)
|
|
|
|
def to_dict(self):
|
|
return {
|
|
'id': self.id,
|
|
'username': self.username,
|
|
'created_at': self.created_at.isoformat() if self.created_at else None
|
|
}
|
|
|
|
class Task(db.Model):
|
|
"""任务模型"""
|
|
__tablename__ = 'tasks'
|
|
|
|
id = db.Column(db.Integer, primary_key=True)
|
|
title = db.Column(db.String(200), nullable=False)
|
|
description = db.Column(db.Text)
|
|
polished_description = db.Column(db.Text) # AI润色后的描述
|
|
status = db.Column(db.String(20), default='pending') # pending, in_progress, completed
|
|
created_at = db.Column(db.DateTime, default=datetime.utcnow)
|
|
updated_at = db.Column(db.DateTime, default=datetime.utcnow, onupdate=datetime.utcnow)
|
|
|
|
# 关联关系
|
|
time_records = db.relationship('TimeRecord', backref='task', lazy=True, cascade='all, delete-orphan')
|
|
|
|
def to_dict(self):
|
|
return {
|
|
'id': self.id,
|
|
'title': self.title,
|
|
'description': self.description,
|
|
'polished_description': self.polished_description,
|
|
'status': self.status,
|
|
'created_at': self.created_at.isoformat() if self.created_at else None,
|
|
'updated_at': self.updated_at.isoformat() if self.updated_at else None,
|
|
'total_time': self.get_total_time()
|
|
}
|
|
|
|
def get_total_time(self):
|
|
"""获取任务总时长（秒）"""
|
|
total_seconds = db.session.query(func.sum(TimeRecord.duration)).filter(
|
|
TimeRecord.task_id == self.id
|
|
).scalar() or 0
|
|
return int(total_seconds)
|
|
|
|
class TimeRecord(db.Model):
|
|
"""时间记录模型"""
|
|
__tablename__ = 'time_records'
|
|
|
|
id = db.Column(db.Integer, primary_key=True)
|
|
task_id = db.Column(db.Integer, db.ForeignKey('tasks.id'), nullable=False)
|
|
start_time = db.Column(db.DateTime, nullable=False)
|
|
end_time = db.Column(db.DateTime)
|
|
duration = db.Column(db.Integer) # 时长（秒）
|
|
created_at = db.Column(db.DateTime, default=datetime.utcnow)
|
|
|
|
def to_dict(self):
|
|
return {
|
|
'id': self.id,
|
|
'task_id': self.task_id,
|
|
'start_time': self.start_time.isoformat() if self.start_time else None,
|
|
'end_time': self.end_time.isoformat() if self.end_time else None,
|
|
'duration': self.duration,
|
|
'created_at': self.created_at.isoformat() if self.created_at else None
|
|
}
|
|
|
|
def calculate_duration(self):
|
|
"""计算时长"""
|
|
if self.start_time and self.end_time:
|
|
self.duration = int((self.end_time - self.start_time).total_seconds())
|
|
return self.duration
|