hoelee/tubesync
1
0
Fork 0
mirror of https://github.com/meeb/tubesync.git synced 2025-06-23 21:46:44 +00:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity

Update fix-mariadb.py

Browse Source
This commit is contained in:
tcely 2025-05-04 01:20:39 -04:00 committed by GitHub
parent 919ba49e28
commit 960c07a846
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 14 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
tubesync/sync/management/commands/fix-mariadb.py
View File

@ -11,7 +11,12 @@ def _(arg_str):
def SQLTable(arg_table): def SQLTable(arg_table):
if not isinstance(arg_table, str): if not isinstance(arg_table, str):
raise TypeError(type(arg_table)) raise TypeError(type(arg_table))
if not arg_table.startswith('sync_'): tables = db.connection.introspection.table_names(include_views=False)
valid_table_name = (
arg_table.startswith('sync_') and
arg_table in tables
)
if not valid_table_name:
raise ValueError(_('Invalid table name')) raise ValueError(_('Invalid table name'))
return str(arg_table) return str(arg_table)
@ -43,20 +48,27 @@ class Command(BaseCommand):
_('An invalid database vendor is configured') _('An invalid database vendor is configured')
+ f': {db.connection.vendor}' + f': {db.connection.vendor}'
) )
db_is_mariadb = ( db_is_mariadb = (
hasattr(db.connection, 'mysql_is_mariadb') and hasattr(db.connection, 'mysql_is_mariadb') and
db.connection.is_usable() and
db.connection.mysql_is_mariadb() db.connection.mysql_is_mariadb()
) )
if not db_is_mariadb: if not db_is_mariadb:
raise CommandError(_('Not conbected to a MariaDB database server.')) raise CommandError(_('Not conbected to a MariaDB database server.'))
display_name = db.connection.display_name
table_names = options.get('delete_table') table_names = options.get('delete_table')
log.info('Start') log.info('Start')
if options['uuid_columns']: if options['uuid_columns']:
if 'uuid' != db.connection.data_types.get('UUIDField', ''):
raise CommandError(_(
f'The {display_name} database server does not support UUID columns.'
))
self.stdout.write('Time to update the columns!') self.stdout.write('Time to update the columns!')
self.stdout.write('Tables to drop:') self.stdout.write('Tables to delete:')
pp( table_names ) pp( table_names )
# All done # All done