Example #1
0
def update_output():
    # Init
    db_utils.create_db()

    #db_utils.get_tasks_info()
    json_every_task()
    html_every_task()

    # Render to image http://stackoverflow.com/questions/2192799/html-to-image-in-javascript-or-python
    subprocess.check_call(['phantomjs', 'generate_image.js', 'html/'])
Example #2
0
def before_all(context):
    """Set env parameters."""
    os.environ['LINES'] = "100"
    os.environ['COLUMNS'] = "100"
    os.environ['EDITOR'] = 'ex'

    context.package_root = os.path.abspath(
        os.path.dirname(os.path.dirname(os.path.dirname(__file__))))

    os.environ["COVERAGE_PROCESS_START"] = os.path.join(context.package_root,
                                                        '.coveragerc')

    context.exit_sent = False

    vi = '_'.join([str(x) for x in sys.version_info[:3]])
    db_name = context.config.userdata.get(
        'my_test_db', None) or "mycli_behave_tests"
    db_name_full = '{0}_{1}'.format(db_name, vi)

    # Store get params from config/environment variables
    context.conf = {
        'host': context.config.userdata.get(
            'my_test_host',
            os.getenv('PYTEST_HOST', 'localhost')
        ),
        'user': context.config.userdata.get(
            'my_test_user',
            os.getenv('PYTEST_USER', 'root')
        ),
        'pass': context.config.userdata.get(
            'my_test_pass',
            os.getenv('PYTEST_PASSWORD', None)
        ),
        'cli_command': context.config.userdata.get(
            'my_cli_command', None) or
        sys.executable + ' -c "import coverage ; coverage.process_startup(); import mycli.main; mycli.main.cli()"',
        'dbname': db_name,
        'dbname_tmp': db_name_full + '_tmp',
        'vi': vi,
        'pager_boundary': '---boundary---',
    }

    _, my_cnf = mkstemp()
    with open(my_cnf, 'w') as f:
        f.write(
            '[client]\n'
            'pager={0} {1} {2}\n'.format(
                sys.executable, os.path.join(context.package_root,
                                             'test/features/wrappager.py'),
                context.conf['pager_boundary'])
        )
    context.conf['defaults-file'] = my_cnf

    context.cn = dbutils.create_db(context.conf['host'], context.conf['user'],
                                   context.conf['pass'],
                                   context.conf['dbname'])

    context.fixture_data = fixutils.read_fixture_files()
Example #3
0
def before_all(context):
    """
    Set env parameters.
    """
    os.environ['LINES'] = "100"
    os.environ['COLUMNS'] = "100"
    os.environ['PAGER'] = 'cat'
    os.environ['EDITOR'] = 'nano'

    context.exit_sent = False

    vi = '_'.join([str(x) for x in sys.version_info[:3]])
    db_name = context.config.userdata.get('pg_test_db', None)
    db_name_full = '{0}_{1}'.format(db_name, vi)

    # Store get params from config.
    context.conf = {
        'host': context.config.userdata.get('pg_test_host', 'localhost'),
        'user': context.config.userdata.get('pg_test_user', 'postgres'),
        'pass': context.config.userdata.get('pg_test_pass', None),
        'dbname': db_name_full,
        'dbname_tmp': db_name_full + '_tmp',
        'vi': vi
    }

    # Store old env vars.
    context.pgenv = {
        'PGDATABASE': os.environ.get('PGDATABASE', None),
        'PGUSER': os.environ.get('PGUSER', None),
        'PGHOST': os.environ.get('PGHOST', None),
        'PGPASS': os.environ.get('PGPASS', None),
    }

    # Set new env vars.
    os.environ['PGDATABASE'] = context.conf['dbname']
    os.environ['PGUSER'] = context.conf['user']
    os.environ['PGHOST'] = context.conf['host']

    if context.conf['pass']:
        os.environ['PGPASS'] = context.conf['pass']
    else:
        if 'PGPASS' in os.environ:
            del os.environ['PGPASS']
        if 'PGHOST' in os.environ:
            del os.environ['PGHOST']

    context.cn = dbutils.create_db(context.conf['host'], context.conf['user'],
                                   context.conf['pass'],
                                   context.conf['dbname'])

    context.fixture_data = fixutils.read_fixture_files()
Example #4
0
def before_all(context):
    """
    Set env parameters.
    """
    os.environ["LINES"] = "100"
    os.environ["COLUMNS"] = "100"
    os.environ["PAGER"] = "cat"
    os.environ["EDITOR"] = "nano"

    context.exit_sent = False

    vi = "_".join([str(x) for x in sys.version_info[:3]])
    db_name = context.config.userdata.get("pg_test_db", None)
    db_name_full = "{0}_{1}".format(db_name, vi)

    # Store get params from config.
    context.conf = {
        "host": context.config.userdata.get("pg_test_host", "localhost"),
        "user": context.config.userdata.get("pg_test_user", "postgres"),
        "pass": context.config.userdata.get("pg_test_pass", None),
        "dbname": db_name_full,
        "dbname_tmp": db_name_full + "_tmp",
        "vi": vi,
    }

    # Store old env vars.
    context.pgenv = {
        "PGDATABASE": os.environ.get("PGDATABASE", None),
        "PGUSER": os.environ.get("PGUSER", None),
        "PGHOST": os.environ.get("PGHOST", None),
        "PGPASS": os.environ.get("PGPASS", None),
    }

    # Set new env vars.
    os.environ["PGDATABASE"] = context.conf["dbname"]
    os.environ["PGUSER"] = context.conf["user"]
    os.environ["PGHOST"] = context.conf["host"]

    if context.conf["pass"]:
        os.environ["PGPASS"] = context.conf["pass"]
    else:
        if "PGPASS" in os.environ:
            del os.environ["PGPASS"]
        if "PGHOST" in os.environ:
            del os.environ["PGHOST"]

    context.cn = dbutils.create_db(
        context.conf["host"], context.conf["user"], context.conf["pass"], context.conf["dbname"]
    )

    context.fixture_data = fixutils.read_fixture_files()
Example #5
0
def before_all(context):
    """
    Set env parameters.
    """
    os.environ['LINES'] = "100"
    os.environ['COLUMNS'] = "100"
    os.environ['PAGER'] = 'cat'
    os.environ['EDITOR'] = 'nano'

    context.exit_sent = False

    vi = '_'.join([str(x) for x in sys.version_info[:3]])
    db_name = context.config.userdata.get('pg_test_db', None)
    db_name_full = '{0}_{1}'.format(db_name, vi)

    # Store get params from config.
    context.conf = {
        'host': context.config.userdata.get('pg_test_host', 'localhost'),
        'user': context.config.userdata.get('pg_test_user', 'postgres'),
        'pass': context.config.userdata.get('pg_test_pass', None),
        'dbname': db_name_full,
        'dbname_tmp': db_name_full + '_tmp',
        'vi': vi
    }

    # Store old env vars.
    context.pgenv = {
        'PGDATABASE': os.environ.get('PGDATABASE', None),
        'PGUSER': os.environ.get('PGUSER', None),
        'PGHOST': os.environ.get('PGHOST', None),
        'PGPASS': os.environ.get('PGPASS', None),
    }

    # Set new env vars.
    os.environ['PGDATABASE'] = context.conf['dbname']
    os.environ['PGUSER'] = context.conf['user']
    os.environ['PGHOST'] = context.conf['host']

    if context.conf['pass']:
        os.environ['PGPASS'] = context.conf['pass']
    else:
        if 'PGPASS' in os.environ:
            del os.environ['PGPASS']
        if 'PGHOST' in os.environ:
            del os.environ['PGHOST']

    context.cn = dbutils.create_db(context.conf['host'], context.conf['user'],
                                   context.conf['pass'],
                                   context.conf['dbname'])

    context.fixture_data = fixutils.read_fixture_files()
Example #6
0
def before_all(context):
    """Set env parameters."""
    os.environ['LINES'] = "100"
    os.environ['COLUMNS'] = "100"
    os.environ['EDITOR'] = 'ex'
    os.environ["COVERAGE_PROCESS_START"] = os.getcwd() + "/../.coveragerc"

    context.exit_sent = False

    vi = '_'.join([str(x) for x in sys.version_info[:3]])
    db_name = context.config.userdata.get('my_test_db',
                                          None) or "mycli_behave_tests"
    db_name_full = '{0}_{1}'.format(db_name, vi)

    # Store get params from config/environment variables
    context.conf = {
        'host':
        context.config.userdata.get('my_test_host',
                                    os.getenv('PYTEST_HOST', 'localhost')),
        'user':
        context.config.userdata.get('my_test_user',
                                    os.getenv('PYTEST_USER', 'root')),
        'pass':
        context.config.userdata.get('my_test_pass',
                                    os.getenv('PYTEST_PASSWORD', None)),
        'cli_command':
        context.config.userdata.get('my_cli_command', None) or sys.executable +
        ' -c "import coverage ; coverage.process_startup(); import mycli.main; mycli.main.cli()"',
        'dbname':
        db_name,
        'dbname_tmp':
        db_name_full + '_tmp',
        'vi':
        vi,
        'pager_boundary':
        '---boundary---',
    }
    os.environ['PAGER'] = "{0} {1} {2}".format(sys.executable,
                                               "test/features/wrappager.py",
                                               context.conf['pager_boundary'])

    context.cn = dbutils.create_db(context.conf['host'], context.conf['user'],
                                   context.conf['pass'],
                                   context.conf['dbname'])

    context.fixture_data = fixutils.read_fixture_files()
Example #7
0
def before_all(context):
    """
    Set env parameters.
    """
    os.environ['LINES'] = "100"
    os.environ['COLUMNS'] = "100"
    os.environ['PAGER'] = 'cat'

    context.exit_sent = False

    # Store get params from config.
    context.conf = {
        'host': context.config.userdata.get('pg_test_host', 'localhost'),
        'user': context.config.userdata.get('pg_test_user', 'root'),
        'pass': context.config.userdata.get('pg_test_pass', None),
        'dbname': context.config.userdata.get('pg_test_db', None),
    }

    # Store old env vars.
    context.pgenv = {
        'PGDATABASE': os.environ.get('PGDATABASE', None),
        'PGUSER': os.environ.get('PGUSER', None),
        'PGHOST': os.environ.get('PGHOST', None),
        'PGPASS': os.environ.get('PGPASS', None),
    }

    # Set new env vars.
    os.environ['PGDATABASE'] = context.conf['dbname']
    os.environ['PGUSER'] = context.conf['user']
    os.environ['PGHOST'] = context.conf['host']
    if context.conf['pass']:
        os.environ['PGPASS'] = context.conf['pass']
    elif 'PGPASS' in os.environ:
        del os.environ['PGPASS']

    context.cn = dbutils.create_db(context.conf['host'], context.conf['user'],
                                   context.conf['pass'],
                                   context.conf['dbname'])

    context.fixture_data = fixutils.read_fixture_files()
Example #8
0
#!/home/common/scripts/python/bin/python3

import sqlite3
import db_utils as db


class scheduleDB
    """Create schedule database object in memory."""

    def __init__(self,schedule)
        self.db = db.create_db(schedule)

Example #9
0
def before_all(context):
    """Set env parameters."""
    env_old = copy.deepcopy(dict(os.environ))
    os.environ['LINES'] = "100"
    os.environ['COLUMNS'] = "100"
    os.environ['PAGER'] = 'cat'
    os.environ['EDITOR'] = 'ex'
    os.environ['VISUAL'] = 'ex'

    context.package_root = os.path.abspath(
        os.path.dirname(os.path.dirname(os.path.dirname(__file__))))
    fixture_dir = os.path.join(
        context.package_root, 'tests/features/fixture_data')

    print('package root:', context.package_root)
    print('fixture dir:', fixture_dir)

    os.environ["COVERAGE_PROCESS_START"] = os.path.join(context.package_root,
                                                        '.coveragerc')

    context.exit_sent = False

    vi = '_'.join([str(x) for x in sys.version_info[:3]])
    db_name = context.config.userdata.get('pg_test_db', 'pgcli_behave_tests')
    db_name_full = '{0}_{1}'.format(db_name, vi)

    # Store get params from config.
    context.conf = {
        'host': context.config.userdata.get(
            'pg_test_host',
            os.getenv('PGHOST', 'localhost')
        ),
        'user': context.config.userdata.get(
            'pg_test_user',
            os.getenv('PGUSER', 'postgres')
        ),
        'pass': context.config.userdata.get(
            'pg_test_pass',
            os.getenv('PGPASSWORD', None)
        ),
        'port': context.config.userdata.get(
            'pg_test_port',
            os.getenv('PGPORT', '5432')
        ),
        'cli_command': (
            context.config.userdata.get('pg_cli_command', None) or
            '{python} -c "{startup}"'.format(
                python=sys.executable,
                startup='; '.join([
                    "import coverage",
                    "coverage.process_startup()",
                    "import pgcli.main",
                    "pgcli.main.cli()"]))),
        'dbname': db_name_full,
        'dbname_tmp': db_name_full + '_tmp',
        'vi': vi,
        'pager_boundary': '---boundary---',
    }
    os.environ['PAGER'] = "{0} {1} {2}".format(
        sys.executable,
        os.path.join(context.package_root, "tests/features/wrappager.py"),
        context.conf['pager_boundary'])

    # Store old env vars.
    context.pgenv = {
        'PGDATABASE': os.environ.get('PGDATABASE', None),
        'PGUSER': os.environ.get('PGUSER', None),
        'PGHOST': os.environ.get('PGHOST', None),
        'PGPASSWORD': os.environ.get('PGPASSWORD', None),
        'PGPORT': os.environ.get('PGPORT', None),
        'XDG_CONFIG_HOME': os.environ.get('XDG_CONFIG_HOME', None),
    }

    # Set new env vars.
    os.environ['PGDATABASE'] = context.conf['dbname']
    os.environ['PGUSER'] = context.conf['user']
    os.environ['PGHOST'] = context.conf['host']
    os.environ['PGPORT'] = context.conf['port']

    if context.conf['pass']:
        os.environ['PGPASSWORD'] = context.conf['pass']
    else:
        if 'PGPASSWORD' in os.environ:
            del os.environ['PGPASSWORD']

    context.cn = dbutils.create_db(context.conf['host'], context.conf['user'],
                                   context.conf['pass'], context.conf['dbname'],
                                   context.conf['port'])

    context.fixture_data = fixutils.read_fixture_files()

    # use temporary directory as config home
    context.env_config_home = tempfile.mkdtemp(prefix='pgcli_home_')
    os.environ['XDG_CONFIG_HOME'] = context.env_config_home
    show_env_changes(env_old, dict(os.environ))
Example #10
0
def before_all(context):
    """Set env parameters."""
    env_old = copy.deepcopy(dict(os.environ))
    os.environ["LINES"] = "100"
    os.environ["COLUMNS"] = "100"
    os.environ["PAGER"] = "cat"
    os.environ["EDITOR"] = "ex"
    os.environ["VISUAL"] = "ex"

    context.package_root = os.path.abspath(
        os.path.dirname(os.path.dirname(os.path.dirname(__file__))))
    fixture_dir = os.path.join(context.package_root,
                               "tests/features/fixture_data")

    print("package root:", context.package_root)
    print("fixture dir:", fixture_dir)

    os.environ["COVERAGE_PROCESS_START"] = os.path.join(
        context.package_root, ".coveragerc")

    context.exit_sent = False

    vi = "_".join([str(x) for x in sys.version_info[:3]])
    db_name = context.config.userdata.get("pg_test_db", "pgcli_behave_tests")
    db_name_full = "{0}_{1}".format(db_name, vi)

    # Store get params from config.
    context.conf = {
        "host":
        context.config.userdata.get("pg_test_host",
                                    os.getenv("PGHOST", "localhost")),
        "user":
        context.config.userdata.get("pg_test_user",
                                    os.getenv("PGUSER", "postgres")),
        "pass":
        context.config.userdata.get("pg_test_pass",
                                    os.getenv("PGPASSWORD", None)),
        "port":
        context.config.userdata.get("pg_test_port",
                                    os.getenv("PGPORT", "5432")),
        "cli_command": (context.config.userdata.get("pg_cli_command", None)
                        or '{python} -c "{startup}"'.format(
                            python=sys.executable,
                            startup="; ".join([
                                "import coverage",
                                "coverage.process_startup()",
                                "import pgcli.main",
                                "pgcli.main.cli()",
                            ]),
                        )),
        "dbname":
        db_name_full,
        "dbname_tmp":
        db_name_full + "_tmp",
        "vi":
        vi,
        "pager_boundary":
        "---boundary---",
    }
    os.environ["PAGER"] = "{0} {1} {2}".format(
        sys.executable,
        os.path.join(context.package_root, "tests/features/wrappager.py"),
        context.conf["pager_boundary"],
    )

    # Store old env vars.
    context.pgenv = {
        "PGDATABASE": os.environ.get("PGDATABASE", None),
        "PGUSER": os.environ.get("PGUSER", None),
        "PGHOST": os.environ.get("PGHOST", None),
        "PGPASSWORD": os.environ.get("PGPASSWORD", None),
        "PGPORT": os.environ.get("PGPORT", None),
        "XDG_CONFIG_HOME": os.environ.get("XDG_CONFIG_HOME", None),
        "PGSERVICEFILE": os.environ.get("PGSERVICEFILE", None),
    }

    # Set new env vars.
    os.environ["PGDATABASE"] = context.conf["dbname"]
    os.environ["PGUSER"] = context.conf["user"]
    os.environ["PGHOST"] = context.conf["host"]
    os.environ["PGPORT"] = context.conf["port"]
    os.environ["PGSERVICEFILE"] = os.path.join(fixture_dir,
                                               "mock_pg_service.conf")

    if context.conf["pass"]:
        os.environ["PGPASSWORD"] = context.conf["pass"]
    else:
        if "PGPASSWORD" in os.environ:
            del os.environ["PGPASSWORD"]

    context.cn = dbutils.create_db(
        context.conf["host"],
        context.conf["user"],
        context.conf["pass"],
        context.conf["dbname"],
        context.conf["port"],
    )

    context.fixture_data = fixutils.read_fixture_files()

    # use temporary directory as config home
    context.env_config_home = tempfile.mkdtemp(prefix="pgcli_home_")
    os.environ["XDG_CONFIG_HOME"] = context.env_config_home
    show_env_changes(env_old, dict(os.environ))
Example #11
0
def before_all(context):
    """Set env parameters."""
    os.environ['LINES'] = "100"
    os.environ['COLUMNS'] = "100"
    os.environ['EDITOR'] = 'ex'
    os.environ['LC_ALL'] = 'en_US.utf8'

    test_dir = os.path.abspath(os.path.dirname(os.path.dirname(__file__)))
    login_path_file = os.path.join(test_dir, 'mylogin.cnf')
    os.environ['MYSQL_TEST_LOGIN_FILE'] = login_path_file

    context.package_root = os.path.abspath(
        os.path.dirname(os.path.dirname(os.path.dirname(__file__))))

    os.environ["COVERAGE_PROCESS_START"] = os.path.join(context.package_root,
                                                        '.coveragerc')

    context.exit_sent = False

    vi = '_'.join([str(x) for x in sys.version_info[:3]])
    db_name = context.config.userdata.get(
        'my_test_db', None) or "mycli_behave_tests"
    db_name_full = '{0}_{1}'.format(db_name, vi)

    # Store get params from config/environment variables
    context.conf = {
        'host': context.config.userdata.get(
            'my_test_host',
            os.getenv('PYTEST_HOST', 'localhost')
        ),
        'user': context.config.userdata.get(
            'my_test_user',
            os.getenv('PYTEST_USER', 'root')
        ),
        'pass': context.config.userdata.get(
            'my_test_pass',
            os.getenv('PYTEST_PASSWORD', None)
        ),
        'cli_command': context.config.userdata.get(
            'my_cli_command', None) or
        sys.executable + ' -c "import coverage ; coverage.process_startup(); import mycli.main; mycli.main.cli()"',
        'dbname': db_name,
        'dbname_tmp': db_name_full + '_tmp',
        'vi': vi,
        'pager_boundary': '---boundary---',
    }

    _, my_cnf = mkstemp()
    with open(my_cnf, 'w') as f:
        f.write(
            '[client]\n'
            'pager={0} {1} {2}\n'.format(
                sys.executable, os.path.join(context.package_root,
                                             'test/features/wrappager.py'),
                context.conf['pager_boundary'])
        )
    context.conf['defaults-file'] = my_cnf
    context.conf['myclirc'] = os.path.join(context.package_root, 'test',
                                           'myclirc')

    context.cn = dbutils.create_db(context.conf['host'], context.conf['user'],
                                   context.conf['pass'],
                                   context.conf['dbname'])

    context.fixture_data = fixutils.read_fixture_files()
Example #12
0
import os
import sys
import requests

from db_utils import select_all_tasks, create_db, change_settings
from helpers import check_settings, get_index

if not os.path.isfile('settings.db'):
    create_db('settings.db')
    settings = select_all_tasks('settings.db')
    check_settings(settings['language'])

settings = select_all_tasks('settings.db')

is_used = True
token = settings['API_KEY']
url = settings['default_url']
headers = {
    'Content-Type': 'application/json',
    'Accept': 'application/json',
    'Authorization': f"Token {token}",
}

language = settings["language"]
while is_used:
    print('Введите желаемый адрес либо "0", если хотите изменить настройки: \n')
    inp = input()
    if inp == '0':
        language = change_settings(language)
    else:
        address = f'{{"query": "{inp}", "language": "{language}"}}'.encode('utf-8')
Example #13
0
def init_db() -> None:
    create_db()
    create_user_table()
    create_meetings_table()
Example #14
0
        # Close archive
        myfile.close()

        

# Find ZIPs and RARs in input directory
for f in os.listdir(INPUT_PATH):
    if fnmatch.fnmatch(f, '*.[Zz][Ii][Pp]'):
        zips.append(f)

    if fnmatch.fnmatch(f, '*.[Cc][Bb][Zz]'):
        zips.append(f)

    if fnmatch.fnmatch(f, '*.[Rr][Aa][Rr]'):
        rars.append(f)

    if fnmatch.fnmatch(f, '*.[Cc][Bb][Rr]'):
        rars.append(f)

# Check for existence of SQLite3 database, creating if necessary
if not os.path.exists(DB_FILE):
    print "SQLite database not found, creating file " + DB_FILE + "...",
    create_db()
    print "done."

# Scan and process ZIP files
scan_archive_file(zips, 'zip')

# Scan and process RAR files
scan_archive_file(rars, 'rar')
Example #15
0
        # Validate limit
        try:
            limit = int(limit)
            print limit
        except:
            print "Limit is not a number"
            limit = 1
            print "Limit set to default " + str(limit)

        if id is None and task is None:
            # Undo tasks (limited) with the newest date
            todo = 1
        elif id is not None:
            # Undo registry with specified id
            todo = 1
        elif task is not None:
            # Undo tasks (limited) with specified name with the newest date
            todo = 1

    # clear command
    elif  arguments['clear'] == True:
        try:
            db_utils.delete_db()
            db_utils.create_db()
        except:
            print "Failed cleaning db"

    # output command
    elif  arguments['output'] == True:
        todo = 1