diff options
Diffstat (limited to '')
-rwxr-xr-x | giti.py | 184 |
1 files changed, 126 insertions, 58 deletions
@@ -1,68 +1,136 @@ -#!/usr/bin/python +#!/usr/bin/env python3 # coding=utf-8 import os import sys -import urllib2 +import requests __author__ = 'Victor Häggqvist' -__version__ = 0.1 - - -def gitiglobal(type): - """ - Fetch gitignore from the global directory - """ - print "Fetching .gitignore for", type,"in Global" - try: - gifile = urllib2.urlopen("https://raw.githubusercontent.com/github/gitignore/master/Global/"+type+".gitignore").read() - store(gifile) - except: - print "Not found in global either" - - -def store(file): - """ - Store content in file - """ - if os.path.isfile(".gitignore") == True: +__version__ = '0.2' + + +def repo_url(kind: str, global_dir=False) -> str: + if global_dir: + return 'https://raw.githubusercontent.com/github/gitignore/master/Global/%s.gitignore' % kind + else: + return 'https://raw.githubusercontent.com/github/gitignore/master/%s.gitignore' % kind + + +def get_ignore_file(kind: str, global_dir=False, path=False) -> str: + """ + Download the file + """ + if not path: + url = repo_url(kind, global_dir) + else: + if kind[0] == '/': + kind = kind[1:] + url = 'https://raw.githubusercontent.com/github/gitignore/master/%s' % kind + + r = requests.get(url) + if r.status_code != 200: + return None + return r.text + + +def search_for_file(kind: str) -> str: + """ + Search for file in https://github.com/github/gitignore + """ + headers = { + 'User-Agent': 'giti.py/0.0.1', + 'Accept': 'application/vnd.github.v3+json' + } + r = requests.get('https://api.github.com/search/code?q=repo:github/gitignore %s' % kind, headers=headers) + cont = r.json() + + print('One of these might be good:') + for i in range(0, len(cont['items'])): + name = cont['items'][i]['name'].split('.')[0] + print('[%s] %s' % (i, name)) + try: - merge = input("Do you want to merge with existing .gitignore [Y/n]:") - except: - merge = "y" - existingFile = open(".gitignore").read() - else: - merge = "y" - existingFile = "" - - if merge.lower() == "y": - gitignore = existingFile + file - f = open('.gitignore', 'w') - f.write(gitignore) - print ".gitignore baked :)" - else: - print "Did nothing, your .gitignore lives like before" - - - -def giti(type): - """ - Fetch gitignore from directory - """ - type = type[0].upper() + type[1:] # make first upper - print "Fetching .gitignore for", type - - try: - gifile = urllib2.urlopen("https://raw.githubusercontent.com/github/gitignore/master/"+type+".gitignore").read() - store(gifile) - except: - print "not found in master" - gitiglobal(type) + choice = int(input('Enter index: ')) + except TypeError: + return None + except KeyboardInterrupt: + return None + + if -1 < choice < len(cont['items']): + return cont['items'][choice]['path'] + else: + return None + + +def save_file(file: str): + """ + Store content in file + """ + merge = 'y' + replace = 'n' + if os.path.isfile('.gitignore'): + try: + merge = input('Do you want to merge with existing .gitignore [Y/n]: ') + merge = 'y' if merge == '' else 'n' + except KeyError: + merge = 'y' + + if merge.lower() == "n": + try: + replace = input('Do you want to replace existing .gitignore [y/N]: ') + except KeyError: + replace = 'n' + + if merge.lower() == 'y': + with open('.gitignore', 'a') as f: + f.write(file) + print('.gitignore baked :)') + + elif replace.lower() == 'y': + with open('.gitignore', 'w') as f: + f.write(file) + print('.gitignore replaced') + + else: + print('Did nothing, your .gitignore lives like before') + + +def giti(kind: str): + kind = kind[0].upper() + kind[1:] + print('Fetching .gitignore for %s' % kind) + + gifile = get_ignore_file(kind) + + if not gifile: + print('Not found in master') + gifile = get_ignore_file(kind, global_dir=True) + + if not gifile: + print('Found no exact match, let\'s give searching a try...') + filepath = search_for_file(kind) + if not filepath: + print('Can\'t help any more, exiting') + print('No changes made') + exit(0) + gifile = get_ignore_file(filepath, path=True) + + if not gifile: + print('Can\'t help any more, exiting') + print('No changes made') + exit(0) + + save_file(gifile) + + +def show_help(): + print('Usage: giti [language or stuff]') + print('giti v%s' % __version__) def main(): - if len(sys.argv) == 2: - giti(sys.argv[1]) - else: - print "Usage: giti [language or stuff]" + if len(sys.argv) == 2: + giti(sys.argv[1]) + else: + show_help() -main() +if __name__ == '__main__': + main() |