def get(self, token):
"""Get token from the storage.
:param token: token as string.
"""
args = (token, )
row = self.db_manager.execute_sql_and_fetchone(SQL_TOKEN_GET, args)
if row:
token_object = convert_db_row_to_dict(row, TOKEN_MODEL_FIELDS)
else:
token_object = {}
return token_object
def get(self, token):
"""
Get token from the storage.
:param token: token as string.
"""
args = (token, )
row = self.db_manager.execute_sql_and_fetchone(SQL_TOKEN_GET, args)
if row:
token_object = convert_db_row_to_dict(row, TOKEN_MODEL_FIELDS)
else:
token_object = {}
return token_object
def get_token_by_username(self, token_name, username):
"""
Get token from the storage by token_name and username.
:param token: token as string.
"""
args = (token_name, username)
row = self.db_manager.execute_sql_and_fetchone(
SQL_TOKEN_GET_BY_TOKEN_USERNAME, args)
if row:
token_object = convert_db_row_to_dict(row, TOKEN_MODEL_FIELDS)
else:
token_object = {}
return token_object
def get_token_by_username(self, token_name, username):
"""Get token from the storage by token_name and username.
:param token: token as string.
"""
args = (token_name, username)
row = self.db_manager.execute_sql_and_fetchone(
SQL_TOKEN_GET_BY_TOKEN_USERNAME, args
)
if row:
token_object = convert_db_row_to_dict(row, TOKEN_MODEL_FIELDS)
else:
token_object = {}
return token_object
def convert_user_raw_data_to_dict(user_raw_data, with_id=False):
"""Convert database row to the dictionary object.
:param user_raw_data: database row.
:param with_id: boolean flag, which means necessity to append to the result
dictionary primary key of database row or not.
"""
if with_id:
fields_tuple = USER_MODEL_FIELDS
else:
fields_tuple = USER_MODEL_FIELDS_WITHOUT_PK
bool_fields = ('is_superuser', 'is_staff', 'is_user', 'is_active')
user_data = convert_db_row_to_dict(user_raw_data, fields_tuple)
for field in bool_fields:
user_data[field] = bool(user_data[field])
return user_data
def convert_user_raw_data_to_dict(user_raw_data, with_id=False):
"""
Convert database row to the dictionary object.
:param user_raw_data: database row.
:param with_id: boolean flag, which means necessity to append to the result
dictionary primary key of database row or not.
"""
if with_id:
fields_tuple = USER_MODEL_FIELDS
else:
fields_tuple = USER_MODEL_FIELDS_WITHOUT_PK
bool_fields = ('is_superuser', 'is_staff', 'is_user', 'is_active')
user_data = convert_db_row_to_dict(user_raw_data, fields_tuple)
for field in bool_fields:
user_data[field] = bool(user_data[field])
return user_data
def test_convert_db_row_to_dict_with_unfilled_data(row, mapped_fields):
data = convert_db_row_to_dict(row, mapped_fields)
assert set([]) == set(data.values())
assert set([]) == set(data.keys())
def test_convert_db_row_to_dict(row, mapped_fields):
data = convert_db_row_to_dict(row, mapped_fields)
assert set(row) == set(data.values())
assert set(mapped_fields) == set(data.keys())
