def create_database(cls):
""" Allows user to create database with a specified name """
name = raw_input(
"Specify database name (default: music.db) - type 'exit' to abort): "
) or "music.db"
if name == 'exit':
return DatabaseMenu
State.ApplicationState.album_manager = Manager.AlbumManager(
os.path.dirname(sys.argv[0]) + "/" + name)
return MainMenu
def load_database(cls):
""" Allows user to load database with a specified name """
files = []
for f in os.listdir(os.path.dirname(sys.argv[0])):
if f.endswith(".db"):
files.append(f)
if not files:
print "No database files found."
raw_input("\nPress ENTER to go back to the previous menu... ")
return DatabaseMenu
else:
print "Databases found: "
for f in files:
print(f)
print
name = raw_input(
"Specify database name (default: music.db - type 'exit' to abort): "
) or "music.db"
if name == 'exit':
return DatabaseMenu
elif not os.path.exists(os.path.dirname(sys.argv[0]) + "/" + name):
print "The specified file does not exist.\n"
return None
State.ApplicationState.album_manager = Manager.AlbumManager(
os.path.dirname(sys.argv[0]) + "/" + name)
return MainMenu