23 lines
643 B
Python
23 lines
643 B
Python
from cryptoai.models.user import User
|
|
from cryptoai.utils.db_manager import get_db_context, SessionLocal
|
|
from sqlalchemy.orm import Session
|
|
import logging
|
|
|
|
logger = logging.getLogger(__name__)
|
|
logger.setLevel(logging.DEBUG)
|
|
|
|
def task_run():
|
|
try:
|
|
session = SessionLocal()
|
|
users = session.query(User).filter(User.points < 1).all()
|
|
for user in users:
|
|
user.points = 1
|
|
session.commit()
|
|
logger.info(f"用户 {user.mail} 积分复位成功")
|
|
except Exception as e:
|
|
logger.error(f"用户积分复位失败: {e}")
|
|
finally:
|
|
if session:
|
|
session.close()
|
|
|