42 lines
1.3 KiB
Python
42 lines
1.3 KiB
Python
from pokedex.db import metadata
|
|
|
|
### Helper functions for oracle
|
|
def rewrite_long_table_names():
|
|
"""Modifies the table names to disenvowel long table names.
|
|
Takes the metadata from memory or uses the imported one.
|
|
|
|
Returns a dictionary matching short names -> long names.
|
|
"""
|
|
|
|
# Load table names from metadata
|
|
t_names = metadata.tables.keys()
|
|
|
|
table_names = list(t_names)
|
|
table_objs = [metadata.tables[name] for name in table_names]
|
|
|
|
# Prepare a dictionary to match old<->new names
|
|
dictionary = {}
|
|
|
|
# Shorten table names, Oracle limits table and column names to 30 chars
|
|
for table in table_objs:
|
|
table._orginal_name = table.name[:]
|
|
dictionary[table.name]=table._orginal_name
|
|
if len(table._orginal_name) > 30:
|
|
for letter in ['a', 'e', 'i', 'o', 'u', 'y']:
|
|
table.name=table.name.replace(letter,'')
|
|
dictionary[table.name]=table._orginal_name
|
|
return dictionary
|
|
|
|
|
|
def restore_long_table_names(metadata,dictionary):
|
|
"""Modifies the table names to restore the long-naming.
|
|
|
|
`metadata`
|
|
The metadata to restore.
|
|
|
|
`dictionary`
|
|
The dictionary matching short name -> long name.
|
|
"""
|
|
for table in metadata.tables.values():
|
|
table.name = dictionary[table.name]
|