Statistics
| Branch: | Revision:

root / snfOCCI / APIserver.py @ 343dc811

History | View | Annotate | Download (2.8 kB)

1
#!/usr/bin/env python
2

    
3
from snfOCCI.registry import snfRegistry
4
from snfOCCI.compute import ComputeBackend
5
from snfOCCI.config import SERVER_CONFIG
6

    
7
from kamaki.clients.compute import ComputeClient
8
from kamaki.clients.cyclades import CycladesClient
9
from kamaki.config  import Config
10

    
11
from occi.core_model import Mixin
12
from occi.backend import MixinBackend
13
from occi.extensions.infrastructure import COMPUTE, START, STOP, SUSPEND, RESTART, RESOURCE_TEMPLATE, OS_TEMPLATE
14
from occi.wsgi import Application
15

    
16
from wsgiref.simple_server import make_server
17
from wsgiref.validate import validator
18

    
19

    
20

    
21

    
22
class MyAPP(Application):
23
    '''
24
    An OCCI WSGI application.
25
    '''
26

    
27
    def refresh_images(self, snf, client):
28

    
29
        images = snf.list_images()
30
        for image in images:
31
            IMAGE_ATTRIBUTES = {'occi.core.id': str(image['id'])}
32
            IMAGE = Mixin("http://schemas.ogf.org/occi/infrastructure#", str(image['name']), [OS_TEMPLATE], attributes = IMAGE_ATTRIBUTES)
33
            self.register_backend(IMAGE, MixinBackend())
34

    
35
    def refresh_flavors(self, snf, client):
36
        
37
        flavors = snf.list_flavors()
38
        for flavor in flavors:
39
            details = snf.get_flavor_details(flavor['id'])
40
            FLAVOR_ATTRIBUTES = {'occi.core.id': flavor['id'],
41
                                 'occi.compute.cores': details['cpu'],
42
                                 'occi.compute.memory': details['ram'],
43
                                 'occi.storage.size': details['disk'],
44
                                 }
45
            FLAVOR = Mixin("http://schemas.ogf.org/occi/infrastructure#", str(flavor['name']), [RESOURCE_TEMPLATE], attributes = FLAVOR_ATTRIBUTES)
46
            self.register_backend(FLAVOR, MixinBackend())
47

    
48

    
49
    def __call__(self, environ, response):
50

    
51
        conf = Config()
52
        conf.set('compute_token',environ['HTTP_AUTH_TOKEN'])
53
        compClient = ComputeClient(conf)
54
        cyclClient = CycladesClient(conf)
55

    
56
        #Up-to-date flavors and images
57
        self.refresh_images(compClient, cyclClient)
58
        self.refresh_flavors(compClient, cyclClient)
59

    
60
        # token will be represented in self.extras
61
        return self._call_occi(environ, response, security = None, token = environ['HTTP_AUTH_TOKEN'], snf = compClient, client = cyclClient)
62

    
63

    
64
def main():
65

    
66

    
67
    APP = MyAPP(registry = snfRegistry())
68
    COMPUTE_BACKEND = ComputeBackend()
69

    
70
    APP.register_backend(COMPUTE, COMPUTE_BACKEND)
71
    APP.register_backend(START, COMPUTE_BACKEND)
72
    APP.register_backend(STOP, COMPUTE_BACKEND)
73
    APP.register_backend(RESTART, COMPUTE_BACKEND)
74
    APP.register_backend(SUSPEND, COMPUTE_BACKEND)
75
    APP.register_backend(RESOURCE_TEMPLATE, MixinBackend())
76
    APP.register_backend(OS_TEMPLATE, MixinBackend())
77
 
78
    VALIDATOR_APP = validator(APP)
79
    HTTPD = make_server('', SERVER_CONFIG['port'], VALIDATOR_APP)
80
    HTTPD.serve_forever()
81