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
|
"""
Package source manager is responsible for selecting the correct source
for a requested package, by a provided string.
Package sources can provide several handlers with different priorities.
A handler is a String -> Result function, which tries to parse the passed
value and return Ok(value).
The package would be handled by the handler with the lowest priority, which
was added the first.
Example:
@dispatcher.source
class BgoSource():
@dispatcher.handler(priority=5)
def parse_int(uri):
if uri.is_decimal():
return Result.Ok(int(uri))
elif uri[0] == '#' and uri[1:].is_decimal():
return Result.Ok(int(uri[1:]))
return Result.Err('NaN')
@dispatcher.handler(priority=1)
def parse_url(uri):
if uri.startswith('http://bugs.gentoo.org/'):
...
"""
#TODO: efficient sorted insertion
#import bisect
import inspect
from pomu.repo.repo import pomu_active_repo
from pomu.util.result import Result
class PackageDispatcher():
def __init__(self):
self.handlers = []
def source(self, cls):
for m, obj in inspect.getmembers(cls):
if isinstance(obj, self.handler._handler):
self.register_package_handler(cls, obj.handler, obj.priority)
return cls
class handler():
class _handler():
def __init__(self, handler):
self.handler = handler
def __call__(self, *args):
return self.handler(*args)
def __init__(self, priority=1000, *args, **kwargs):
self.priority = priority
def __call__(self, func, *args, **kwargs):
x = self._handler(func)
x.priority = self.priority
return x
def register_package_handler(self, source, handler, priority):
i = 0
for i in range(len(self.handlers)):
if self.handlers[0] > priority:
break
self.handlers.insert(i, (priority, source, handler))
def get_package_source(self, uri):
for priority, source, handler in self.handlers:
if handler(uri).is_ok():
return source
return None
def get_package(self, uri):
for priority, source, handler in self.handlers:
res = handler(uri)
if res.is_ok():
return Result.Ok(source.fetch_package(res.ok()))
return Result.Err('No handler found for package ' + uri)
def install_package(self, uri):
pkg = self.get_package(uri).unwrap()
#TODO: write a helper function which expects pomu_active_repo
#alternatively, test for it before any command which requires a repo
return pomu_active_repo().unwrap().merge(pkg)
def uninstall_package(self, uri):
pkg = self.get_package(uri).unwrap()
return pomu_active_repo().unwrap().unmerge(pkg)
|
GitWeb