diff --git a/.gitignore b/.gitignore index d532c87..c0a9379 100644 --- a/.gitignore +++ b/.gitignore @@ -5,6 +5,9 @@ venv/ # Logs logs/ +# Calendars +calendars/ + # Bot private edt edt.* @@ -12,3 +15,6 @@ token.ini # PyCharm .idea/ + +# Alembic +alembic* diff --git a/update.py b/update.py deleted file mode 100644 index cc39708..0000000 --- a/update.py +++ /dev/null @@ -1,9 +0,0 @@ -import shelve -from base import User - -with shelve.open("edt", writeback=True) as db: - for u in db: - nu = User(0, None) - for v in db[u].__dict__: - setattr(nu, v, db[u].__dict__[v]) - db[u] = nu diff --git a/update.sh b/update.sh new file mode 100755 index 0000000..b6b25de --- /dev/null +++ b/update.sh @@ -0,0 +1,13 @@ +#!/bin/bash +if [ -f edt.db ]; then + if [ ! -d alembic ]; then + alembic init alembic + sed -i '/sqlalchemy.url/s/= .*/= sqlite:\/\/\/edt.db/' alembic.ini + sed -i "/target_metadata = None/s/target_metadata.*/import os, sys\nsys.path.append(os.getcwd())\nfrom base import Base\ntarget_metadata = Base.metadata/" alembic/env.py + fi + + alembic revision --autogenerate -m "Auto upgrade" + alembic upgrade head +else + echo "No database !" +fi