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} 增加初始订阅信息")