-
Notifications
You must be signed in to change notification settings - Fork 1
/
gh_extension.py
68 lines (50 loc) · 1.71 KB
/
gh_extension.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
import os
import subprocess
import dotbot
"""
gh extension install https://github.com/vilmibm/gh-user-status.git
"""
class GHExtension(dotbot.Plugin):
"""
Install GH exstations
"""
_default_flags = [""]
def __init__(self, context):
super(GHExtension, self).__init__(context)
self._directives = {"ghe": self._get}
# Dotbot methods
def can_handle(self, directive):
return directive in self._directives
def handle(self, directive, data):
try:
for entry in data:
self._directives[directive](entry)
command = "gh extension upgrade --all"
subprocess.call(command, cwd=self.cwd, shell=True)
return True
except ValueError as e:
self._log.error(e)
return False
# Utility
@property
def cwd(self):
return self._context.base_directory()
# Inner methods
def _get(self, data):
repo, flags = self._parse(data, "repo")
command = "gh extension install {} {}".format(repo, flags)
subprocess.call(command, cwd=self.cwd, shell=True)
def _parse(self, data, key):
if type(data) is dict:
if key not in data:
raise ValueError("Key '{}' not found in {}".format(key, data))
if "flags" not in data:
self._log.warning("Key 'flags' not found in {}".format(data))
self._log.warning(
"Using default flags {}".format(self._default_flags))
value = data[key]
flags = data.get("flags", self._default_flags)
else:
value = data
flags = self._default_flags
return value, " ".join(flags)