Chetan Gaonker | 4a25e2b | 2016-03-04 14:45:15 -0800 | [diff] [blame] | 1 | import json |
| 2 | import requests |
| 3 | import os,sys,time |
| 4 | |
| 5 | class OnosCtrl: |
Chetan Gaonker | a2b87df | 2016-03-31 15:41:31 -0700 | [diff] [blame^] | 6 | |
| 7 | auth = ('karaf', 'karaf') |
| 8 | controller = os.getenv('ONOS_CONTROLLER_IP') or 'localhost' |
| 9 | cfg_url = 'http://%s:8181/onos/v1/network/configuration/' %(controller) |
| 10 | |
Chetan Gaonker | 4a25e2b | 2016-03-04 14:45:15 -0800 | [diff] [blame] | 11 | def __init__(self, app, controller = None): |
| 12 | self.app = app |
Chetan Gaonker | a2b87df | 2016-03-31 15:41:31 -0700 | [diff] [blame^] | 13 | if controller is not None: |
Chetan Gaonker | 4a25e2b | 2016-03-04 14:45:15 -0800 | [diff] [blame] | 14 | self.controller = controller |
| 15 | self.app_url = 'http://%s:8181/onos/v1/applications/%s' %(self.controller, self.app) |
| 16 | self.cfg_url = 'http://%s:8181/onos/v1/network/configuration/' %(self.controller) |
| 17 | self.auth = ('karaf', 'karaf') |
| 18 | |
Chetan Gaonker | a2b87df | 2016-03-31 15:41:31 -0700 | [diff] [blame^] | 19 | @classmethod |
| 20 | def config(cls, config): |
Chetan Gaonker | 4a25e2b | 2016-03-04 14:45:15 -0800 | [diff] [blame] | 21 | if config: |
| 22 | json_data = json.dumps(config) |
Chetan Gaonker | a2b87df | 2016-03-31 15:41:31 -0700 | [diff] [blame^] | 23 | resp = requests.post(cls.cfg_url, auth = cls.auth, data = json_data) |
Chetan Gaonker | 4a25e2b | 2016-03-04 14:45:15 -0800 | [diff] [blame] | 24 | return resp.ok, resp.status_code |
| 25 | return False, 400 |
| 26 | |
| 27 | def activate(self): |
| 28 | resp = requests.post(self.app_url + '/active', auth = self.auth) |
| 29 | return resp.ok, resp.status_code |
| 30 | |
| 31 | def deactivate(self): |
| 32 | resp = requests.delete(self.app_url + '/active', auth = self.auth) |
| 33 | return resp.ok, resp.status_code |
| 34 | |
Chetan Gaonker | a2b87df | 2016-03-31 15:41:31 -0700 | [diff] [blame^] | 35 | @classmethod |
| 36 | def get_devices(cls): |
| 37 | url = 'http://%s:8181/onos/v1/devices' %(cls.controller) |
| 38 | result = requests.get(url, auth = cls.auth) |
| 39 | if result.ok: |
| 40 | devices = result.json()['devices'] |
| 41 | return filter(lambda d: d['available'], devices) |
Chetan Gaonker | 4a25e2b | 2016-03-04 14:45:15 -0800 | [diff] [blame] | 42 | |
Chetan Gaonker | a2b87df | 2016-03-31 15:41:31 -0700 | [diff] [blame^] | 43 | return None |
| 44 | |
| 45 | @classmethod |
| 46 | def get_flows(cls, device_id): |
| 47 | url = 'http://%s:8181/onos/v1/flows/' %(cls.controller) + device_id |
| 48 | result = requests.get(url, auth = cls.auth) |
| 49 | if result.ok: |
| 50 | return result.json()['flows'] |
| 51 | return None |
| 52 | |
| 53 | @classmethod |
| 54 | def cord_olt_config(cls, olt_device_data = None): |
| 55 | '''Configures OLT data for existing devices/switches''' |
| 56 | if olt_device_data is None: |
| 57 | return |
| 58 | did_dict = {} |
| 59 | config = { 'devices' : did_dict } |
| 60 | devices = cls.get_devices() |
| 61 | if not devices: |
| 62 | return |
| 63 | device_ids = map(lambda d: d['id'], devices) |
| 64 | for did in device_ids: |
| 65 | access_device_dict = {} |
| 66 | access_device_dict['accessDevice'] = olt_device_data |
| 67 | did_dict[did] = access_device_dict |
| 68 | |
| 69 | ##configure the device list with access information |
| 70 | return cls.config(config) |