32 lines
1.1 KiB
Python
32 lines
1.1 KiB
Python
from cryptoai.agents.crypto_agent import CryptoAgent
|
|
import cryptoai.tasks.token_selector as token_selector
|
|
from cryptoai.agents.crypto_agent import CryptoAgent
|
|
import cryptoai.tasks.news as news
|
|
from cryptoai.utils.db_manager import SessionLocal
|
|
from cryptoai.models.user import User
|
|
from cryptoai.models.user_subscription import UserSubscriptionManager
|
|
from datetime import datetime, timedelta
|
|
import asyncio
|
|
import logging
|
|
import uuid
|
|
logger = logging.getLogger(__name__)
|
|
logger.setLevel(logging.DEBUG)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
|
|
session = SessionLocal()
|
|
users = session.query(User).all()
|
|
|
|
user_subscription_manager = UserSubscriptionManager(session)
|
|
|
|
for u in users:
|
|
# 如果没有用户订阅，则增加订阅信息
|
|
if not user_subscription_manager.get_subscription_by_user_id(u.id):
|
|
expire_time = datetime.now().replace(hour=23, minute=59, second=59) + timedelta(days=7)
|
|
|
|
random_order_id = str(uuid.uuid4())
|
|
|
|
user_subscription_manager.create_subscription(u.id,1, 1, random_order_id, expire_time)
|
|
logger.info(f"用户 {u.mail} 增加初始订阅信息")
|