Files
osmedeus/lib/alias/git_format
2019-09-09 11:32:56 +07:00

59 lines
1.5 KiB
Python
Executable File

#!/usr/bin/env python3
import os
import sys
import time
import argparse
import urllib.parse
sys.path.append(os.path.dirname(os.path.dirname(os.path.realpath(__file__))))
from core import utils
# git_format -i /tmp/target.txt -o '$WORKSPACE/gitscan/$OUTPUT'
# git_format -i 'repo:sda' -o '$WORKSPACE/gitscan/$OUTPUT'
def parse_input(data, out):
repo = f'{out}-repo.txt'
user = f'{out}-user.txt'
if ':' not in data:
utils.just_append(repo, data.strip())
else:
key = data.split(':')[0]
real_data = ''.join(data.split(':')[1:])
if 'user' in key or 'org' in key:
utils.just_append(user, real_data + "\n")
else:
utils.just_append(repo, real_data + "\n")
# print(real_data)
try:
utils.clean_up(repo)
utils.clean_up(user)
except:
pass
def routine(args):
target = args.input
output = args.output
if utils.not_empty_file(f'{target}'):
contents = utils.just_read(target, get_list=True)
else:
contents = [target]
for line in contents:
if ',' in line:
datas = line.split(',')
else:
datas = [line]
for data in datas:
parse_input(data, output)
# time.sleep(0.3)
parser = argparse.ArgumentParser(description="Git Format alias")
parser.add_argument('-i', '--input', action='store', dest='input', help='input')
parser.add_argument('-o', '--output', action='store', dest='output', help='output')
args = parser.parse_args()
routine(args)