-
Notifications
You must be signed in to change notification settings - Fork 4
/
upload_results.py
76 lines (65 loc) · 2.17 KB
/
upload_results.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
# upload results of an autotest via http
import io, os, platform, sys, zipfile
def upload_results_http(tests, parameters, args):
upload_url = parameters["upload_url"]
upload_fields = parameters["upload_fields"]
upload_fields["exercise"] = args.exercise
upload_fields["hostname"] = platform.node()
upload_fields["login"] = getlogin()
buffer = io.BytesIO()
zip_files_for_upload(buffer, tests, parameters, args)
buffer.seek(0)
if args.debug:
print(upload_url, upload_fields)
try:
# requests may not be installed
import requests
r = requests.post(upload_url, upload_fields, files={"zip": ("zip", buffer)})
except Exception as e:
if args.debug:
print(e, file=sys.stderr)
if args.debug:
print(r.text, file=sys.stderr)
def zip_files_for_upload(stream, tests, parameters, args):
# pylint: disable=consider-using-with
# use of with triggered a bug here - in old python versions
zf = zipfile.ZipFile(stream, "w", compression=zipfile.ZIP_LZMA)
bytes_uploaded = 0
for test in tests.values():
try:
zf.writestr(test.label + ".passed", "1" if test.passed else "")
except AttributeError:
pass
# don't zip files that are supplied in autotest
supplied = os.listdir(parameters["supplied_files_directory"])
upload_files = set(args.file | args.optional_files).difference(supplied)
for filename in ["autotest.log"] + list(upload_files):
try:
bytes_uploaded += os.path.getsize(filename)
if bytes_uploaded > parameters["upload_max_bytes"]:
break
zf.write(filename)
except OSError:
pass
zf.close()
def getlogin():
"""
attempt to get username robustly whatever the platform
"""
try:
import pwd
return pwd.getpwuid(os.geteuid()).pw_name
except Exception:
pass
try:
return os.getlogin()
except Exception:
pass
try:
return (
os.getenv("LOGNAME", "")
or os.getenv("USER", "")
or os.getenv("USERNAME", "")
)
except Exception:
return ""