002-users.py 799 B

123456789101112131415161718192021222324252627282930313233
  1. import pathlib
  2. current = pathlib.Path(__file__).parent
  3. root = current.parent
  4. import sys
  5. sys.path.append(str(root))
  6. import assets
  7. import sqlmodel
  8. builder = assets.user_builder()
  9. builder.nick = "test1"
  10. builder.password = "qwerty"
  11. test1 = builder.result
  12. connection = sqlmodel.create_engine("sqlite:///002-test.db")
  13. sqlmodel.SQLModel.metadata.create_all(connection)
  14. with assets.user_loader(connection) as loader:
  15. print("Register: " + str(loader.register(test1)))
  16. logged = loader.login("test1", "qwerty")
  17. print("Logged user apikey: " + str(logged.apikey))
  18. bad_psk = loader.login("test1", "qweee")
  19. print("Bad password user: " + str(bad_psk))
  20. logged.apikey = assets.apikey()
  21. loader.save(logged)
  22. print("Logged user apikey after change: " + str(logged.apikey))