cmdpackage 0.1.3__py3-none-any.whl

src/defs/writeSetup.py

@@ -0,0 +1,124 @@
+#!/usr/bin/python
+# -*- coding: utf-8 -*-
+from ..templates.setupTemplates import (
+    setup_base_template, setup_line, gitignore_content, classifiers_line,
+    classifiers_template)
+from sys import version_info
+from .runSubProc import runSubProc
+from subprocess import Popen, PIPE
+from getpass import getuser
+import os
+
+def writeSetup() -> dict:
+    rtnDict = {}
+    fields = ['name', 'version', 'description', 'license', 'author']
+    setup_lines = ''
+
+    for field_name in fields:
+        default_value = default_values(field_name)
+        if field_name == 'description':
+            default_value = default_value.replace('name', rtnDict['name'])
+        input_msg = input_message(field_name, default_value)
+        input_value = get_input(input_msg, default=default_value)
+        rtnDict[field_name] = input_value
+        setup_lines += setup_line.substitute(
+            name=field_name, value=input_value
+        )
+
+    setup_content = setup_base_template.substitute(
+        setup_lines=setup_lines,
+        name=rtnDict['name'],
+        classifiers=gen_classifiers()
+    )
+
+    with open('setup.py', 'w') as setup_file:
+        write_content(setup_file, setup_content)
+
+    with_gitignore = get_input('commit a git repo [Y/n]?: ',
+                               default='y')
+    if with_gitignore.lower() == 'y':
+        with open('.gitignore', 'w') as gitignore_file:
+            write_content(gitignore_file, gitignore_content)
+        initGitRepo()
+    return rtnDict
+
+def input_message(field_name, default_value):
+    return u'{} ({}): '.format(field_name, default_value)
+
+
+def gen_classifiers():
+    mayor, minor = version_info[:2]
+    python = "Programming Language :: Python"
+    local = "Programming Language :: Python :: {}.{}".format(mayor, minor)
+    classifiers = [python, local]
+
+    classifiers_lines = ''
+    for cls in classifiers:
+        classifiers_lines += classifiers_line.substitute(classifier=cls)
+
+    return classifiers_template.substitute(classifiers=classifiers_lines)
+
+def initGitRepo():
+    rtnStr = runSubProc('ls .git')
+    if rtnStr.returncode != 0:
+        rtnStr = runSubProc(f'git init')
+    if rtnStr.returncode == 0:
+        rtnStr = runSubProc(f'git add .')
+    if rtnStr.returncode == 0:
+        rtnStr = runSubProc(f'git commit -m "inital commit"',noOutput=False)
+
+def get_username():
+    '''Get git config values.'''
+    username = ''
+
+    # use try-catch to prevent crashes if user doesn't install git
+    try:
+        # run git config --global <key> to get username
+        git_command = ['git', 'config', '--global', 'user.name']
+        p = Popen(git_command, stdout=PIPE, stderr=PIPE)
+        output, err = p.communicate()
+
+        # turn stdout into unicode and strip it
+        username = output.decode('utf-8').strip()
+
+        # if user doesn't set global git config name, then use getuser()
+        if not username:
+            username = getuser()
+    except OSError:
+        # if git command is not found, then use getuser()
+        username = getuser()
+
+    return username
+
+
+def default_values(field_name):
+    if field_name == 'name':
+        rtnStr = os.path.relpath('.', '..')
+        rtnStr = rtnStr.replace("-","_")
+        return rtnStr
+    if field_name == 'version':
+        return '0.1.0'
+    elif field_name == 'description':
+        return 'name pip package'
+    elif field_name == 'license':
+        return 'MIT'
+    elif field_name == 'author':
+        return get_username()
+
+
+def get_input(input_msg, default=None):
+    if version_info >= (3, 0):
+        input_value = input(input_msg)
+    else:
+        input_value = raw_input(input_msg.encode('utf8')).decode('utf8')
+
+    if input_value == '':
+        return default
+    return input_value
+
+
+def write_content(file, content):
+    if version_info >= (3, 0):
+        file.write(content)
+    else:
+        file.write(content.encode('utf8'))