dbupdate.py

#! /usr/bin/python

import pymysql.cursors
import argparse

from configobj import ConfigObj

from lib.dbcheck import check_db_version


def migrate_db():
    version = check_db_version(cursor)
    print("Old Version {}".format(version))

    if version < 1:
        cursor.execute("alter table `user` add column lat double default 0")
        cursor.execute("alter table `user` add column lon double default 0")
        cursor.execute("alter table `user` add column dist double default 0")
        version = 1
    if version < 2:
        cursor.execute("alter table `userassign` add column level int default 0")
        version = 2
    if version < 3:
        cursor.execute('create table userallow (`chatid` varchar(45) not null, primary key (`chatid`)) engine=InnoDB '
                       'default charset=utf8mb4;')
        version = 3

    cursor.execute("update dbversion set version = '%s'" % version)

    version = check_db_version(cursor)
    print("New Version {}".format(version))
    print("Migration complete")


############# MAIN ################

##################
# parsing arguments
parser = argparse.ArgumentParser()
parser.add_argument("-c", dest='configfile', help="configfile", type=str, default="config.ini")
args = parser.parse_args()

# read inifile
try:
    config = ConfigObj(args.configfile)
    token = config.get('token')
    db = config['dbname']
    dbhost = config['dbhost']
    dbport = config.get('dbport', '3306')
    dbuser = config['dbuser']
    dbpassword = config['dbpassword']
except:
    print("Inifile not given")
    quit()

# connect to database
#
try:
    connection = pymysql.connect(host=dbhost,
                                 user=dbuser,
                                 password=dbpassword,
                                 db=db,
                                 port=int(dbport),
                                 charset='utf8mb4',
                                 autocommit='True')
    cursor = connection.cursor()
except:
    print("can not connect to database")
    quit()

migrate_db()