/
Frozin
/
DIS-4
Обзор
Документация
Войти
/
Frozin
/
DIS-4
Код
Запросы
0
Задачи
Вики
Пакеты
0
Релизы
0
CI/CD
Аналитика
Безопасность
master
reset_database.py
45 строк
2 KB
Frozin-p
Fixed database operations. Added files for testing
16 дек 2025, 16:45
16 дек 2025, 16:45
109ec6f
Код
Авторство
О чём код?
import psycopg2 from common.config import config def reset_database(): """Очистка базы данных от тестовых данных""" try: connection = psycopg2.connect( host=config.DB_HOST, port=config.DB_PORT, database=config.DB_NAME, user=config.DB_USER, password=config.DB_PASSWORD ) connection.autocommit = True cursor = connection.cursor() print("Очистка базы данных...") # Удаляем всех пользователей кроме, возможно, первого тестового cursor.execute("DELETE FROM users WHERE id > 1") print(f"Удалено пользователей: {cursor.rowcount}") # Очищаем таблицу processed_requests cursor.execute("DELETE FROM processed_requests") print(f"Очищено записей processed_requests: {cursor.rowcount}") # Сбрасываем sequence для users.id cursor.execute("SELECT setval('users_id_seq', (SELECT COALESCE(MAX(id), 1) FROM users))") # Сбрасываем sequence для api_keys.id cursor.execute("SELECT setval('api_keys_id_seq', (SELECT COALESCE(MAX(id), 1) FROM api_keys))") print("База данных очищена") cursor.close() connection.close() except Exception as e: print(f"Ошибка при очистке базы данных: {e}") if __name__ == "__main__": reset_database()