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
|
#!/usr/bin/env python
#
# Copyright 2015-present Facebook. All Rights Reserved.
#
# This program file is free software; you can redistribute it and/or modify it
# under the terms of the GNU General Public License as published by the
# Free Software Foundation; version 2 of the License.
#
# This program is distributed in the hope that it will be useful, but WITHOUT
# ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
# FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License
# for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program in a file named COPYING; if not, write to the
# Free Software Foundation, Inc.,
# 51 Franklin Street, Fifth Floor,
# Boston, MA 02110-1301 USA
#
import urllib2, urllib
import logging
import os
import subprocess
import time
import argparse
SPATULA_FILE = '/etc/spatula/spatula'
LOG_FORMAT = '[%(levelname)s] %(asctime)s: %(message)s'
DATE_FORMAT = '%Y-%m-%d %H:%M:%S'
DEFAULT_SLEEP = 900 # default sleep 15 mins
class SpatulaWrapper(object):
def __init__(self, address='fe80::2', interface='usb0',
port=8087, ssl=False):
proto = 'http'
if ssl:
proto = 'https'
self.url = '{proto}://{address}%{iface}:{port}'.format(
proto = proto,
address = address,
iface = interface,
port = port)
def _getSpatula(self, endpoint='/spatula'):
'''
Get the executable spatula script from the host.
'''
url = '{}{}'.format(self.url, endpoint)
try:
request = urllib2.Request(url)
response = urllib2.urlopen(request)
return response.read()
except Exception as err:
raise Exception('failed getting Spatula {}: {}'.format(url, err))
def _success(self, endpoint='/success'):
'''
Api to report the timestamp of a successful run
'''
query = urllib.urlencode({'timestamp': time.time()})
url = '{}{}?{}'.format(self.url, endpoint, query)
try:
request = urllib2.Request(url)
response = urllib2.urlopen(request)
return response.read()
except Exception as err:
raise Exception('failed report success {}: {}'.format(url, err))
def _error(self, error, endpoint='/error'):
'''
Api to report the timestamp and the error when spatula fails
'''
query = urllib.urlencode({'timestamp': time.time(), 'error': error})
url = '{}{}?{}'.format(self.url, endpoint, query)
try:
request = urllib2.Request(url)
response = urllib2.urlopen(request)
return response.read()
except Exception as err:
raise Exception('failed report error [{}]: {}'.format(url, err))
def _execute(self):
try:
# get the executable from host
if not os.path.exists(os.path.dirname(SPATULA_FILE)):
os.makedirs(os.path.dirname(SPATULA_FILE))
with open(SPATULA_FILE, 'w+') as file:
file.write(self._getSpatula())
file.close()
# set the permission
os.chmod(SPATULA_FILE, 0755)
# run the executable file
spatula = subprocess.Popen([SPATULA_FILE],
stdout=subprocess.PIPE,
stderr=subprocess.PIPE)
out, err = spatula.communicate()
if spatula.returncode != 0:
raise Exception('spatula failed: {}'.format(err))
self._success()
except Exception as err:
self._error(err)
raise err
def run(self, sleep):
while True:
try:
self._execute()
except Exception as err:
logging.error(err)
time.sleep(sleep)
if __name__ == '__main__':
args = argparse.ArgumentParser()
args.add_argument('-s', '--sleep', default=DEFAULT_SLEEP,
help='Sleep time between spatula runs (default=%(default)s)')
params = args.parse_args()
logging.basicConfig(format=LOG_FORMAT, datefmt=DATE_FORMAT)
wrapper = SpatulaWrapper()
wrapper.run(params.sleep)
|
