users.py 840 B

123456789101112131415161718192021222324252627282930
  1. import json
  2. import os
  3. __dir__ = os.path.dirname(__file__)
  4. with open(f"{__dir__}/users.csv") as f:
  5. data = f.readlines()
  6. script = "DECLARE @inserted_row_id INT;\n"
  7. for index, row in enumerate(data):
  8. if index == 0:
  9. continue
  10. parts = row.split(";")
  11. name, surname = parts[1].split(" ", 1)
  12. login = parts[2]
  13. password = parts[3]
  14. usertype = parts[7]
  15. script += f"INSERT INTO users (role,login,password,surname,name) VALUES ({usertype}, '{login}', '{password}', '{surname}', '{name}')\n"
  16. script += "SELECT @inserted_row_id = SCOPE_IDENTITY();\n"
  17. services = json.loads(parts[6][1:-1])
  18. for item in services:
  19. script += f"INSERT INTO employee_services (service_id, user_id) VALUES ((SELECT id FROM services WHERE code={item['code']}), @inserted_row_id);\n"
  20. print(script)