Loading Pipfile +1 −1 Original line number Diff line number Diff line Loading @@ -4,7 +4,7 @@ verify_ssl = true name = "pypi" [packages] common = {editable = true, directory = "./src"} common-package = {editable = true, path = "./src", extras=["postgres"]} [dev-packages] pytest = "*" Loading Pipfile.lock +327 −76 File changed.Preview size limit exceeded, changes collapsed. Show changes src/common/auth.py +6 −1 Original line number Diff line number Diff line Loading @@ -174,6 +174,10 @@ def login( return {"msg": "Invalid uid or password"}, 401 def check_username_exists(args, kwargs): return True def register_user( username: str, password: str, Loading @@ -184,7 +188,7 @@ def register_user( work_sector: str, user_type: str, reason: str, db: UNDB, db: Database, table_name: str, username_column: str, password_column: str, Loading Loading @@ -235,6 +239,7 @@ def reset_password( table_name: str, username_column: str, salt_column: str, password_column: str, ) -> dict: # Retrieve user's salt from the database user_salt = db.query( Loading src/common/mixins/multiple.py +4 −4 Original line number Diff line number Diff line Loading @@ -10,9 +10,6 @@ from concurrent.futures import ThreadPoolExecutor from common.env import mock_env_vars from common.database import Database from common.logz import create_logger from common.mixins.postgres import PostgresMixin from common.mixins.mssql import MSSQLMixin from common.mixins.influx import InfluxMixin class MultiDatabase: Loading @@ -36,13 +33,16 @@ class MultiDatabase: @staticmethod def _get_mixin_for_type( db_type: str, ) -> Optional[Type[Union[PostgresMixin, MSSQLMixin, InfluxMixin]]]: ) -> Optional[Type[Union['PostgresMixin', 'MSSQLMixin', 'InfluxMixin']]]: db_type = db_type.lower() if db_type.startswith("postgres") or db_type.startswith("pg"): from common.mixins.postgres import PostgresMixin return PostgresMixin if db_type.startswith("mssql"): from common.mixins.mssql import MSSQLMixin return MSSQLMixin if db_type.startswith("influx"): from common.mixins.influx import InfluxMixin return InfluxMixin # add more database Mixins here as they become available return None Loading src/pyproject.toml +1 −0 Original line number Diff line number Diff line Loading @@ -40,6 +40,7 @@ postgres = ["psycopg2-binary==2.9.9"] mssql = ["pymssql==2.2.11"] influx = ["influxdb==5.3.1"] auth = [ "lxml", "xmlsec==1.3.13", "onelogin==2.0.3", "python3_saml==1.14.0", Loading Loading
Pipfile +1 −1 Original line number Diff line number Diff line Loading @@ -4,7 +4,7 @@ verify_ssl = true name = "pypi" [packages] common = {editable = true, directory = "./src"} common-package = {editable = true, path = "./src", extras=["postgres"]} [dev-packages] pytest = "*" Loading
src/common/auth.py +6 −1 Original line number Diff line number Diff line Loading @@ -174,6 +174,10 @@ def login( return {"msg": "Invalid uid or password"}, 401 def check_username_exists(args, kwargs): return True def register_user( username: str, password: str, Loading @@ -184,7 +188,7 @@ def register_user( work_sector: str, user_type: str, reason: str, db: UNDB, db: Database, table_name: str, username_column: str, password_column: str, Loading Loading @@ -235,6 +239,7 @@ def reset_password( table_name: str, username_column: str, salt_column: str, password_column: str, ) -> dict: # Retrieve user's salt from the database user_salt = db.query( Loading
src/common/mixins/multiple.py +4 −4 Original line number Diff line number Diff line Loading @@ -10,9 +10,6 @@ from concurrent.futures import ThreadPoolExecutor from common.env import mock_env_vars from common.database import Database from common.logz import create_logger from common.mixins.postgres import PostgresMixin from common.mixins.mssql import MSSQLMixin from common.mixins.influx import InfluxMixin class MultiDatabase: Loading @@ -36,13 +33,16 @@ class MultiDatabase: @staticmethod def _get_mixin_for_type( db_type: str, ) -> Optional[Type[Union[PostgresMixin, MSSQLMixin, InfluxMixin]]]: ) -> Optional[Type[Union['PostgresMixin', 'MSSQLMixin', 'InfluxMixin']]]: db_type = db_type.lower() if db_type.startswith("postgres") or db_type.startswith("pg"): from common.mixins.postgres import PostgresMixin return PostgresMixin if db_type.startswith("mssql"): from common.mixins.mssql import MSSQLMixin return MSSQLMixin if db_type.startswith("influx"): from common.mixins.influx import InfluxMixin return InfluxMixin # add more database Mixins here as they become available return None Loading
src/pyproject.toml +1 −0 Original line number Diff line number Diff line Loading @@ -40,6 +40,7 @@ postgres = ["psycopg2-binary==2.9.9"] mssql = ["pymssql==2.2.11"] influx = ["influxdb==5.3.1"] auth = [ "lxml", "xmlsec==1.3.13", "onelogin==2.0.3", "python3_saml==1.14.0", Loading