gs_pypi/gs_pypi_generate_db.py


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60

#!/usr/bin/env python
# -*- coding: utf-8 -*-

"""
    pypi_db.py
    ~~~~~~~~~~
    
    PyPI database generation
    
    :copyright: (c) 2013 by Jauhien Piatlicki
    :license: GPL-2, see LICENSE for more details.
"""

import os
import sys

from g_sorcery.compatibility import TemporaryDirectory
from g_sorcery.exceptions import FileJSONError
from g_sorcery.fileutils import copy_all, FileJSON
from g_sorcery.logger import Logger

from .pypi_db import PypiDBGenerator

def main():

    if len(sys.argv) < 2:
        print("Usage: ")
        print("gs-pypi-generate-db db_name")
        return -1
    
    logger = Logger()
    cfg_path = None
    for path in '.', '~', '/etc/g-sorcery':
        current = os.path.join(path, "gs-pypi.json")
        if (os.path.isfile(current)):
            cfg_path = path
            break
    if not cfg_path:
        logger.error('no config file for gs-pypi backend\n')
        return -1
    cfg_f = FileJSON(cfg_path, "gs-pypi.json", ['package'])
    try:
        config = cfg_f.read()
    except FileJSONError as e:
        logger.error('error loading config file for gs-pypi: ' + str(e) + '\n')
        return -1

    generator = PypiDBGenerator()
    db_name = sys.argv[1]
    temp_dir = TemporaryDirectory()
    pkg_db = generator(temp_dir.name, "pypi", config=config["repositories"]["pypi"], common_config=config["common_config"])
    if os.path.exists(db_name):
        os.system('rm -rf ' + db_name + '/*')
    else:
        os.mkdir(db_name)
    copy_all(os.path.join(temp_dir.name, "pypi/db"), db_name)
    os.system('tar cvzf ' +  db_name + '.tar.gz ' + db_name)

if __name__ == "__main__":
    sys.exit(main())