python/utils/main.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

#!/usr/bin/env python3

import sys
import os
import urllib.request
import venv
import subprocess


def make_project_dir(directory):
    try:
        os.makedirs(directory)
    except FileExistsError:
        print("The directory you provided already exists, not doing anything")
        exit(1)
    return


def get_gitignore(directory):
    url = "https://raw.githubusercontent.com/github/gitignore/master/Python.gitignore"
    urllib.request.urlretrieve(url, os.path.join(directory, '.gitignore'))
    return


def make_venv(directory):
    venvdir = os.path.join(directory, 'venv')
    venv.create(env_dir=venvdir, with_pip=True)
    return venvdir


def prepare_venv(venv):
    whichpip = os.path.join(venv, 'bin', 'pip')
    output = subprocess.check_output(
        [whichpip, 'install', 'autopep8', 'pylint'])
    return


def main():
    directory = sys.argv[1]
    make_project_dir(directory)
    get_gitignore(directory)
    venv = make_venv(directory)
    prepare_venv(venv)
    return


if __name__ == '__main__':
    main()