def main_menu_factory_empty_database(config, database_connection, screen):
user_dao = UserDao(database_connection)
def _inner(event_handler):
return create_main_menu(screen, event_handler, config,
database_connection)
return _inner
def validate_user_exist(self, user_name_input):
user_name_dao = UserDao()
user_name_value = user_name_dao.get_user_name()
for user_name in user_name_value:
if user_name_input != user_name:
return False
return True
def __init__(self):
self.token = ''
self.bot_user_name = ''
self.bot_name = ''
self.updater = Updater(self.token, use_context=True)
self.user_operations = UserOperaetions()
self.user_dao = UserDao()
self.dp = self.updater.dispatcher
self.updater.start_polling()
self.updater.idle()
def main_menu_factory(config, database_connection, screen):
user_dao = UserDao(database_connection)
user_dao.create(User("apina"))
user_dao.create(User("banaani"))
user_dao.create(User("cembalo"))
def _inner(event_handler):
return create_main_menu(screen, event_handler, config,
database_connection)
return _inner
def game_factory_factory(config, database_connection, screen):
user_dao = UserDao(database_connection)
user_dao.create(User("apina"))
user_dao.create(User("banaani"))
user_dao.create(User("cembalo"))
def _inner(game_length, event_handler):
config.game_length = game_length
return GameFactory(config,
user_dao,
event_handler,
screen,
n_players=2)
return _inner
def test_user_dao(self):
# Make sure we're not overwriting a database
if exists(f'{TestUserDao.db_name}.db'):
logging.fatal(
f"Database test file '{TestUserDao.db_name}.db' already exists. Aborting."
)
return
# Create the database and dao
db = Db(TestUserDao.db_name)
dao = UserDao()
# Test non-existence
self.assertFalse(db.does_table_exist(dao),
f"User table exists before creation")
# Test creation
try:
db.create_table(dao)
except Exception as e:
self.fail(f"Create user table failed with exception {e}")
self.assertTrue(db.does_table_exist(dao),
f"User table does not exist after creation")
# Test empty count
self.assertEqual(dao.get_count(db), 0,
f"User table contains rows when none entered")
# Test deletion
try:
db.delete_table(dao)
except Exception as e:
self.fail(f"Delete user table failed with exception {e}")
self.assertFalse(db.does_table_exist(dao),
f"User table exists after deletion")
# Clean up
db.destroy()
def instance():
if not hasattr(DbManager, "_instance"):
DbManager._instance = DbManager()
DbManager._db_conn = MysqlConnect()
DbManager._daos.update({
'episode':
EpisodeDao(DbManager._db_conn),
'user':
UserDao(DbManager._db_conn),
'keyword':
KeywordDao(DbManager._db_conn),
'keyword_episode':
KeywordEpisodeDao(DbManager._db_conn),
'subject':
SubjectDao(DbManager._db_conn),
'subject_episode':
SubjectEpisodeDao(DbManager._db_conn),
'page_episode':
PageEpisodeDao(DbManager._db_conn),
'ordered':
OrderedDao(DbManager._db_conn),
'page':
PageDao(DbManager._db_conn),
'cat_list':
CatListDao(DbManager._db_conn),
'cat_episode':
CatListEpisodeDao(DbManager._db_conn),
'cat_exclude':
CatExcludeDao(DbManager._db_conn),
'channel_exclude':
ChannelExcludeDao(DbManager._db_conn),
'category':
CategoryDao(DbManager._db_conn),
'tag':
TagDao(DbManager._db_conn),
})
return DbManager._instance
def __init__(self, drive_client, sheets_client):
self._service_dao = ServiceDao(sheets_client)
self._songs_dao = SongsDao(sheets_client, drive_client)
self._user_dao = UserDao()
self._recipient_dao = RecipientDao()
self._band_dao = BandDao(self._recipient_dao)