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