Trivial migrations implementation

This commit is contained in:
SoniEx2 2019-05-03 20:19:24 -03:00
parent 77f5a068f4
commit 58f3175668
1 changed files with 40 additions and 9 deletions

View File

@ -66,6 +66,10 @@ TEMPLATE = """<!DOCTYPE html>
</html>
"""
MIGRATIONS = {
"test": ("-- apply", "-- revert", "does nothing")
}
try:
data_home = os.environ['XDG_DATA_HOME']
except KeyError:
@ -170,6 +174,7 @@ def remove(url):
conn.commit()
conn.close()
def migrations():
@ganarchy.group()
def migrations():
"""Modifies the DB to work with a newer/older version.
@ -177,12 +182,38 @@ def migrations():
WARNING: THIS COMMAND CAN BE EXTREMELY DESTRUCTIVE!"""
@migrations.command()
def apply():
""" NYI """
@click.argument('migration')
def apply(migration):
"""Applies the migration with the given name."""
conn = sqlite3.connect(data_home + "/ganarchy.db")
c = conn.cursor()
click.echo(MIGRATIONS[migration][0])
c.execute(MIGRATIONS[migration][0])
conn.commit()
conn.close()
@click.argument('migration')
@migrations.command()
def revert():
""" NYI """
def revert(migration):
"""Reverts the migration with the given name."""
conn = sqlite3.connect(data_home + "/ganarchy.db")
c = conn.cursor()
click.echo(MIGRATIONS[migration][1])
c.execute(MIGRATIONS[migration][1])
conn.commit()
conn.close()
@click.argument('migration', required=False)
@migrations.command()
def info(migration):
"""Shows information about the migration with the given name."""
if not migration:
# TODO could be improved
click.echo(MIGRATIONS.keys())
else:
click.echo(MIGRATIONS[migration][2])
migrations()
@ganarchy.command()
def cron_target():