blob: eac54f91941b958d020bd23f8bd1922c39bd7117 (
plain)
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
|
#!/usr/bin/python3
# ups.khirnov.net uploader
import os.path
import shutil
import subprocess
import sys
try:
import magic
have_magic = True
except ImportError:
have_magic = False
server = 'https://ups.khirnov.net/'
cert_path = os.path.expanduser('~/.local/var/up/cert')
if len(sys.argv) > 2:
sys.stderr.write('Too many arguments\n')
sys.exit(1)
elif len(sys.argv) == 2 and sys.argv[1] != '-':
src_name = sys.argv[1]
dst_name = os.path.basename(src_name)
stdin = subprocess.DEVNULL
else:
src_name = '-'
dst_name = 'stdin'
stdin = subprocess.PIPE
# try to guess the extension
if have_magic:
m = magic.Magic(extension = True)
e = m.from_buffer(sys.stdin.buffer.peek(2048))
if e != '???':
dst_name += '.' + e.split('/')[0]
curl_cmdline = ['curl', '--silent', '--show-error', '--cert', cert_path,
'--data-binary', '@' + src_name, server + dst_name]
child = subprocess.Popen(curl_cmdline, stdin = stdin)
if stdin != subprocess.DEVNULL:
# stream our stdin to the child process
# cannot pass stdin to the child directly because we peeked at it above
shutil.copyfileobj(sys.stdin.buffer, child.stdin)
child.stdin.close()
ret = child.wait()
if ret != 0:
sys.stderr.write('curl returned an error code: %d\n' % ret)
else:
sys.stdout.write('\n')
|