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
|
"""A class for remote rsync repos"""
from os import mkdtemp, rmdir, mkfifo, unlink
import re
from subprocess import run
from pomu.repo.remote import RemoteRepo, normalize_key
from pomu.util.result import Result
class RemoteRsyncRepo(RemoteRepo):
"""A class responsible for rsync remotes"""
def __init__(self, url):
self.uri = uri
def __enter__(self):
pass
def __exit__(self, *_):
pass
def fetch_tree(self):
"""Returns repos hierarchy"""
if hasattr(self, '_tree'):
return self._tree
d = mkdtemp()
p = run('rsync', '-rn', '--out-format="%n"' self.uri, d)
rmdir(d)
if p.returncode:
return Result.Err()
self._tree = ['/' + x for x in p.stdout.split('\n')]
return self._tree
def fetch_subtree(self, key):
"""Lists a subtree"""
k = normalize_key(key, True)
self.fetch_tree()
dic = dict(self._tree)
if k not in dic:
return []
l = len(key)
return Result.Ok(
[tpath[l:] for tpath in self.fetch_tree() if tpath.startswith(k)])
def fetch_file(self, key):
"""Fetches a file from the repo"""
k = normalize_key(key)
self.fetch_tree()
dic = dict(self._tree)
if k not in dic:
return Result.Err()
d = mkdtemp()
fip = path.join(d, 'fifo')
mkfifo(fip)
p = run('rsync', self.uri.rstrip('/') + key, fip)
fout = fip.read()
unlink(fip)
rmdir(d)
if p.returncode:
return Result.Err()
return Result.Ok(fout)
|