-
Notifications
You must be signed in to change notification settings - Fork 0
/
wit.py
79 lines (66 loc) · 2.08 KB
/
wit.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
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
# upload 170
import os
import sys
import wit_add
import wit_commit
import wit_checkout
import wit_merge
import wit_status
import wit_graph
import wit_branch
class NoWitError(Exception):
pass
def check_for_wit():
for _, dirs, _ in os.walk(os.getcwd()):
for d in dirs:
if d == '.wit':
return True
return False
def init():
dir1 = os.path.join(f'{os.getcwd()}\\.wit')
if os.path.exists(dir1):
return True
else:
os.mkdir(dir1)
images_dir = os.path.join(f'{dir1}\\images')
staging_dir = os.path.join(f'{dir1}\\staging_area')
activated_path = os.path.join(f'{dir1}\\activated.txt')
os.mkdir(images_dir)
os.mkdir(staging_dir)
with open(activated_path, 'w+') as active:
active.write('Master')
def check_input(): # input in cmd is string that is appended to sys.argv, thus this function is necessary
action = None
if sys.argv[1] == 'init':
action = init()
if 'add' in sys.argv and sys.argv[1] == 'add':
path = sys.argv[2]
action = add(filename=path)
if 'commit' in sys.argv and sys.argv[1] == 'commit':
message = None
if len(sys.argv) == 3:
message = sys.argv[2]
action = commit(message=message)
if 'status' in sys.argv and sys.argv[1] == 'status':
action = status()
if 'checkout' in sys.argv and sys.argv[1] == 'checkout':
if len(sys.argv) == 3:
commit_id = sys.argv[2]
action = checkout(commit_id)
if 'graph' in sys.argv and sys.argv[1] == 'graph':
action = graph()
if 'branch' in sys.argv and sys.argv[1] == 'branch':
branch_name = sys.argv[2]
action = branch(branch_name=branch_name)
if 'merge' in sys.argv and sys.argv[1] == 'merge':
branch_name = sys.argv[2]
action = merge(branch_name=branch_name)
return action
add = wit_add.add
commit = wit_commit.commit
status = wit_status.status
checkout = wit_checkout.checkout
graph = wit_graph.graph
branch = wit_branch.branch
merge = wit_merge.merge
check_input()