database.py

import sqlite3
from datetime import datetime

DB_NAME = 'database.db'

def init_db():
    """Инициализация базы данных и создание таблиц"""
    conn = sqlite3.connect(DB_NAME)
    cursor = conn.cursor()

    # Таблица пользователей (сотрудников)
    cursor.execute('''
        CREATE TABLE IF NOT EXISTS users (
            id INTEGER PRIMARY KEY AUTOINCREMENT,
            chat_id INTEGER UNIQUE NOT NULL,
            username TEXT NOT NULL,
            full_name TEXT,
            is_admin INTEGER DEFAULT 0,
            created_at TEXT DEFAULT CURRENT_TIMESTAMP
        )
    ''')

    # Таблица новостей
    cursor.execute('''
        CREATE TABLE IF NOT EXISTS news (
            id INTEGER PRIMARY KEY AUTOINCREMENT,
            text TEXT NOT NULL,
            created_at TEXT DEFAULT CURRENT_TIMESTAMP
        )
    ''')

    conn.commit()
    conn.close()
    print("✅ База данных инициализирована")

def seed_test_data():
    """Заполнение базы тестовыми данными для демонстрации"""
    conn = sqlite3.connect(DB_NAME)
    cursor = conn.cursor()

    # Проверяем, есть ли уже данные
    cursor.execute('SELECT COUNT(*) FROM users')
    if cursor.fetchone()[0] > 0:
        print("⚠️  Тестовые данные уже существуют")
        conn.close()
        return

    # Добавляем тестовых пользователей
    # ВАЖНО: Замените эти chat_id на реальные для тестирования!
    # Чтобы получить свой chat_id, напишите боту @userinfobot в Telegram
    test_users = [
        (298752282, 'denisyuce', 'Denis Yuce', 1),  # Администратор (HR)
        (114863002, 'mark', 'Mark Ivanov', 0),  # Марк - администратор (HR)
    ]

    cursor.executemany('INSERT INTO users (chat_id, username, full_name, is_admin) VALUES (?, ?, ?, ?)', test_users)

    # Добавляем тестовые новости
    test_news = [
        'Добро пожаловать в систему HR-уведомлений!',
        'С понедельника меняется график работы: 9:00-18:00',
        'Напоминаем о корпоративе в пятницу в 19:00',
        'Новые правила оформления отпусков доступны в общей папке',
    ]

    for news_text in test_news:
        cursor.execute('INSERT INTO news (text) VALUES (?)', (news_text,))

    conn.commit()
    conn.close()
    print("✅ Тестовые данные добавлены")

def get_all_users():
    """Получить всех пользователей"""
    conn = sqlite3.connect(DB_NAME)
    cursor = conn.cursor()
    cursor.execute('SELECT chat_id, username FROM users')
    users = cursor.fetchall()
    conn.close()
    return users

def get_all_users_full():
    """Получить всех пользователей с полной информацией"""
    conn = sqlite3.connect(DB_NAME)
    cursor = conn.cursor()
    cursor.execute('SELECT username, full_name, is_admin, created_at FROM users ORDER BY is_admin DESC, full_name')
    users = cursor.fetchall()
    conn.close()
    return users

def get_user_by_username(username):
    """Найти пользователя по username"""
    conn = sqlite3.connect(DB_NAME)
    cursor = conn.cursor()
    cursor.execute('SELECT chat_id, username FROM users WHERE username = ?', (username,))
    user = cursor.fetchone()
    conn.close()
    return user

def is_admin(chat_id):
    """Проверить, является ли пользователь администратором"""
    conn = sqlite3.connect(DB_NAME)
    cursor = conn.cursor()
    cursor.execute('SELECT is_admin FROM users WHERE chat_id = ?', (chat_id,))
    result = cursor.fetchone()
    conn.close()
    return result and result[0] == 1

def add_news(text):
    """Добавить новость"""
    conn = sqlite3.connect(DB_NAME)
    cursor = conn.cursor()
    cursor.execute('INSERT INTO news (text) VALUES (?)', (text,))
    conn.commit()
    conn.close()

def get_latest_news(limit=5):
    """Получить последние новости"""
    conn = sqlite3.connect(DB_NAME)
    cursor = conn.cursor()
    cursor.execute('SELECT text, created_at FROM news ORDER BY id DESC LIMIT ?', (limit,))
    news = cursor.fetchall()
    conn.close()
    return news

def add_user(chat_id, username):
    """Добавить нового пользователя"""
    conn = sqlite3.connect(DB_NAME)
    cursor = conn.cursor()
    try:
        cursor.execute('INSERT INTO users (chat_id, username) VALUES (?, ?)', (chat_id, username))
        conn.commit()
        conn.close()
        return True
    except sqlite3.IntegrityError:
        # Пользователь уже существует
        conn.close()
        return False

if __name__ == '__main__':
    # Если запустить этот файл напрямую, инициализируется база с тестовыми данными
    init_db()
    seed_test_data()
    print("\n📊 Статистика базы данных:")
    print(f"Пользователей: {len(get_all_users())}")
    print(f"Новостей: {len(get_latest_news(100))}")