models.py

from datetime import datetime
import os
from flask_sqlalchemy import SQLAlchemy
import logging

logger = logging.getLogger(__name__)
db = SQLAlchemy()

def init_db(app):
    """Initialize the database with the application"""
    try:
        logger.info("Initializing database...")
        db.init_app(app)
        
        with app.app_context():
            # Создаем директорию instance если её нет
            os.makedirs('instance', exist_ok=True)
            # Создаем все таблицы
            db.create_all()
            logger.info("Database tables created successfully")
            
    except Exception as e:
        logger.error(f"Failed to initialize database: {str(e)}", exc_info=True)
        try:
            if db.session:
                db.session.rollback()
        except:
            pass
        raise

class Command(db.Model):
    """Model for storing voice commands and their results"""
    __tablename__ = 'commands'
    
    id = db.Column(db.Integer, primary_key=True)
    text = db.Column(db.String(500), nullable=False)
    command_type = db.Column(db.String(50))
    status = db.Column(db.String(20))
    result = db.Column(db.Text)
    created_at = db.Column(db.DateTime, default=datetime.utcnow)

    def __repr__(self):
        return f'<Command {self.id}: {self.command_type}>'

class Task(db.Model):
    """Model for storing tasks created by voice commands"""
    __tablename__ = 'tasks'
    
    id = db.Column(db.Integer, primary_key=True)
    title = db.Column(db.String(200), nullable=False)
    description = db.Column(db.Text)
    status = db.Column(db.String(20), default='pending')
    category = db.Column(db.String(50))
    priority = db.Column(db.String(20), default='normal')
    due_date = db.Column(db.DateTime)
    created_at = db.Column(db.DateTime, default=datetime.utcnow)

    def __repr__(self):
        return f'<Task {self.id}: {self.title}>'

class BusinessEntity(db.Model):
    """Model for storing business entities like clients, suppliers, etc."""
    __tablename__ = 'business_entities'
    
    id = db.Column(db.Integer, primary_key=True)
    name = db.Column(db.String(200), nullable=False)
    entity_type = db.Column(db.String(50), nullable=False)  # client, supplier, etc.
    description = db.Column(db.Text)
    contact_info = db.Column(db.Text)
    status = db.Column(db.String(20), default='active')
    created_at = db.Column(db.DateTime, default=datetime.utcnow)

    def __repr__(self):
        return f'<BusinessEntity {self.id}: {self.name} ({self.entity_type})>'