forked from pdonorio/rest-mock
-
Notifications
You must be signed in to change notification settings - Fork 0
/
do_actions.py
138 lines (105 loc) · 4.58 KB
/
do_actions.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
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
# -*- coding: utf-8 -*-
# TO FIX: import logger?
class InvalidArgument(BaseException):
pass
class NotImplementedAction(BaseException):
pass
class ImplementedActions(object):
def __init__(self, compose, vanilla_services):
self._compose_project = compose
self._vanilla_services = vanilla_services
def _exec(self, command):
print ("\nCommand to be executed:\n\tdocker-compose %s\n\n" % command)
def service_mandatory(self, service):
if service is None:
raise InvalidArgument(
'Service parameter is mandatory for this action'
)
def service_incompatible(self, service):
if service is not None:
raise InvalidArgument(
'Service parameter is incompatible with this action'
)
def do_check(self, project, action, service, **kwargs):
self.service_incompatible(service)
raise NotImplementedAction(
'verify if the %s blueprint is well-configured ' +
'[verify all blueprint-specific dir and configuration files]'
% project
)
def do_init(self, command, project, action, service, **kwargs):
self.service_incompatible(service)
raise NotImplementedAction(
'init the %s blueprint (in old do command) ' +
'[clone backend and frontend, pull docker images, pull bower libs'
% project
)
def do_update(self, command, project, action, **kwargs):
print("TEST", self._compose_project)
# TODO: git pull
# TO FIX: plumbum?
# TODO: also in init!
# images pull
command.append('pull')
# log.pp(self._vanilla_services)
# log.pp(self._compose_project)
raise NotImplementedAction("Missing recursion on service links")
for service in self._compose_project.services:
if service.name not in self._vanilla_services:
continue
# recursion on links
for link in service.links:
linked_service, name = link
print("TEST 2", linked_service)
# log.pp(service.__dict__)
command.append(service)
self._exec(command)
# bower/npm/yarn install
raise NotImplementedAction(
'pull git, docker images and bower libs in blueprint %s' % project
)
def do_start(self, command, action, service, **kwargs):
self.service_mandatory(service)
self._exec("%s %s %s" % (command, action, service))
def do_stop(self, command, action, service, **kwargs):
self.service_mandatory(service)
self._exec("%s %s %s" % (command, action, service))
def do_restart(self, command, action, service, **kwargs):
self.service_mandatory(service)
self._exec("%s %s %s" % (command, action, service))
def do_graceful(self, command, action, service, **kwargs):
self.service_mandatory(service)
self._exec("%s %s %s" % (command, action, service))
def do_scale(self, command, action, service, num, **kwargs):
self.service_mandatory(service)
self._exec(
"%s %s %s=%s" % (command, action, service, num))
def do_logs(self, command, action, service, **kwargs):
if service is None:
self._exec("%s %s" % (command, action))
else:
self._exec("%s %s %s" % (command, action, service))
def do_remove(self, command, action, service, **kwargs):
# service is required or not for this action?
self.service_mandatory(service)
self._exec("%s %s %s" % (command, action, service))
def do_clean(self, command, action, service, **kwargs):
# service is required or not for this action?
self.service_mandatory(service)
self._exec("%s %s %s" % (command, action, service))
def do_command(self, command, action, service, arguments, **kwargs):
self.service_mandatory(service)
if len(arguments) == 0:
raise InvalidArgument('Missing arguments for command action')
self._exec(
"%s exec %s %s" % (command, service, arguments))
def do_shell(self, command, action, service, **kwargs):
self.service_mandatory(service)
self.do_command(
command, action, service, arguments='bash')
def do_bower(self, command, action, service, arguments, **kwargs):
self.service_incompatible(service)
if len(arguments) == 0:
raise InvalidArgument('Missing arguments for bower action')
self.do_command(
command, action, service='bower', arguments=arguments)