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())