get-data.py

#!/usr/bin/python3
# -*- coding: utf-8 -*-
# Author: Himura Kazuto <himura@tulafest.ru>

import os
import sqlite3
import sys

from lib.config import read_config
from lib.authenticator import Authenticator
from lib.fetcher import Fetcher
from lib.make_db import MakeDB

if __name__ == '__main__':
    config = read_config()
    event_name = config['event_name']
    c2_login = config['admin_cs2_name']
    c2_password = config['admin_cs2_password'] if 'admin_cs2_password' in config else None
    db_path = config['db_path']
    sql = config['sql_after_get'].strip() if 'sql_after_get' in config else None

    all_data = len(sys.argv) > 1 and sys.argv[1] == '-a'

    a = Authenticator(event_name, c2_login, c2_password)
    if not a.sign_in():
        exit()

    print()
    f = Fetcher(a.event_name, a.cookie)
    if not f.fetch_data():
        exit()

    if all_data:
        if not f.fetch_etickets():
            exit()
        if not f.fetch_details():
            exit()

    print('\nCreating ' + db_path + '...')
    MakeDB(db_path, f.data)

    if sql:
        from tabulate import tabulate

        print('\nConnecting to ' + db_path + ' again...')
        with sqlite3.connect(db_path, isolation_level=None) as db:
            c = db.cursor()
            c.execute('PRAGMA encoding = "UTF-8"')
            c.execute(sql)
            print(tabulate(c.fetchall(), headers=[description[0] for description in c.description], tablefmt='grid'))