Tony Mack | 79a49c8 | 2013-06-15 23:51:57 -0400 | [diff] [blame] | 1 | import threading |
| 2 | import requests, json |
| 3 | from core.models import * |
| 4 | from openstack.manager import OpenStackManager |
| 5 | |
| 6 | # decorator that marks dispatachable event methods |
| 7 | def event(func): |
| 8 | setattr(func, 'event', func.__name__) |
| 9 | return func |
| 10 | |
| 11 | class EventHandler: |
| 12 | |
| 13 | def __init__(self): |
| 14 | self.manager = OpenStackManager() |
| 15 | |
Tony Mack | e9ba12d | 2013-06-24 09:37:49 -0400 | [diff] [blame] | 16 | @staticmethod |
| 17 | def get_events(): |
Tony Mack | 79a49c8 | 2013-06-15 23:51:57 -0400 | [diff] [blame] | 18 | events = [] |
Tony Mack | e9ba12d | 2013-06-24 09:37:49 -0400 | [diff] [blame] | 19 | for name in dir(EventHandler): |
| 20 | attribute = getattr(EventHandler, name) |
| 21 | if hasattr(attribute, 'event'): |
| 22 | events.append(getattr(attribute, 'event')) |
Tony Mack | 79a49c8 | 2013-06-15 23:51:57 -0400 | [diff] [blame] | 23 | return events |
| 24 | |
| 25 | def dispatch(self, event, *args, **kwds): |
| 26 | if hasattr(self, event): |
| 27 | return getattr(self, event)(*args, **kwds) |
| 28 | |
| 29 | |
| 30 | @event |
| 31 | def save_site(self, id): |
| 32 | sites = Site.objects.filter(id=id) |
| 33 | if sites: |
| 34 | self.manager.save_site(sites[0]) |
| 35 | |
| 36 | @event |
| 37 | def delete_site(self, tenant_id): |
| 38 | self.manager.driver.delete_tenant(tenant_id) |
| 39 | |
| 40 | @event |
| 41 | def save_site_privilege(self, id): |
| 42 | site_privileges = SitePrivilege.objects.filter(id=id) |
| 43 | if site_privileges: |
| 44 | site_priv = self.manager.save_site_privilege(site_privileges[0]) |
| 45 | |
| 46 | @event |
| 47 | def delete_site_privilege(self, kuser_id, tenant_id, role_type): |
| 48 | self.manager.driver.delete_user_role(kuser_id, tenant_id, role_type) |
| 49 | |
| 50 | @event |
| 51 | def save_slice(self, id): |
| 52 | slices = Slice.objects.filter(id=id) |
| 53 | if slices: |
| 54 | self.manager.save_slice(slices[0]) |
| 55 | |
| 56 | @event |
| 57 | def delete_slice(self, tenant_id, network_id, router_id, subnet_id): |
| 58 | self.manager._delete_slice(tenant_id, network_id, router_id, subnet_id) |
| 59 | |
| 60 | @event |
| 61 | def save_user(self, id): |
| 62 | users = User.objects.filter(id=id) |
| 63 | if users: |
| 64 | self.manager.save_user(users[0]) |
| 65 | |
| 66 | @event |
| 67 | def delete_user(self, kuser_id): |
| 68 | self.manager.driver.delete_user(kuser_id) |
| 69 | |
| 70 | @event |
| 71 | def save_sliver(self, id): |
| 72 | slivers = Sliver.objects.filter(id=id) |
| 73 | if slivers: |
| 74 | self.manager.save_sliver(slivers[0]) |
| 75 | |
| 76 | @event |
| 77 | def delete_sliver(self, instance_id): |
| 78 | self.manager.destroy_instance(instance_id) |
| 79 | |
| 80 | |
| 81 | |
| 82 | class EventListener: |
| 83 | |
| 84 | def __init__(self): |
| 85 | self.handler = EventHandler() |
| 86 | |
| 87 | def listen_for_event(self, event, hash): |
| 88 | url = 'http://www.feefie.com/command' |
| 89 | params = {'action': 'subscribe', |
| 90 | 'hash': hash, |
| 91 | 'htm': 1} |
| 92 | while True: |
| 93 | r = requests.get(url, params=params) |
| 94 | r_data = json.loads(r) |
| 95 | payload = r_data.get('payload') |
| 96 | self.handler.dispatch(event, **payload) |
| 97 | |
| 98 | |
| 99 | def run(self): |
| 100 | # register events |
Tony Mack | e9ba12d | 2013-06-24 09:37:49 -0400 | [diff] [blame] | 101 | event_names = [{'title': name} for name in EventHandler.get_events()] |
Tony Mack | 79a49c8 | 2013-06-15 23:51:57 -0400 | [diff] [blame] | 102 | url = 'http://www.feefie.com/command' |
| 103 | params = {'action': 'add', |
| 104 | 'u': 'pl', |
Tony Mack | e9ba12d | 2013-06-24 09:37:49 -0400 | [diff] [blame] | 105 | 'events': json.dumps(event_names)} |
Tony Mack | 79a49c8 | 2013-06-15 23:51:57 -0400 | [diff] [blame] | 106 | r = requests.get(url, params=params) |
Tony Mack | 79a49c8 | 2013-06-15 23:51:57 -0400 | [diff] [blame] | 107 | r_data = json.loads(r) |
| 108 | events = r_data.get('events', []) |
| 109 | # spanw a thread for each event |
| 110 | for event in events: |
| 111 | args = (event['title'], event['hash']) |
| 112 | listener_thread = threading.Thread(target=self.listen_for_event, args=args) |
| 113 | listener_tread.start() |
| 114 | |