From fa752cff978024bcfc3a114eaeadfd0b0f27e583 Mon Sep 17 00:00:00 2001 From: MoonTestUse1 Date: Sat, 28 Dec 2024 07:45:57 +0600 Subject: [PATCH] =?UTF-8?q?=D0=A1=D0=BE=D0=B7=D0=B4=D0=B0=D0=BD=D0=B8?= =?UTF-8?q?=D1=8F=20=D1=81=D0=BA=D1=80=D0=B8=D0=BF=D1=82=D0=B0,=20=D1=8E?= =?UTF-8?q?=D0=B7=D0=B5=D1=80=D0=B0?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- backend/scripts.py/create_test_user.py | 35 ++++++++++++++++++++++++++ 1 file changed, 35 insertions(+) create mode 100644 backend/scripts.py/create_test_user.py diff --git a/backend/scripts.py/create_test_user.py b/backend/scripts.py/create_test_user.py new file mode 100644 index 0000000..27e5ae2 --- /dev/null +++ b/backend/scripts.py/create_test_user.py @@ -0,0 +1,35 @@ +"""Script to create a test user in the database""" +from sqlalchemy.orm import Session +from app.database import SessionLocal +from app.crud import employees +from app.models.employee import EmployeeCreate + +def create_test_employee(): + db = SessionLocal() + try: + # Create test employee data + test_employee = EmployeeCreate( + first_name="Иван", + last_name="Иванов", + department="general", + office="101", + password="test123" + ) + + # Check if employee already exists + existing_employee = employees.get_employee_by_lastname(db, test_employee.last_name) + if existing_employee: + print(f"Employee {test_employee.last_name} already exists") + return + + # Create new employee + db_employee = employees.create_employee(db, test_employee) + print(f"Created test employee: {db_employee.last_name} (ID: {db_employee.id})") + + except Exception as e: + print(f"Error creating test employee: {e}") + finally: + db.close() + +if __name__ == "__main__": + create_test_employee() \ No newline at end of file