diff options
Diffstat (limited to 'resource_manager.py')
| -rw-r--r-- | resource_manager.py | 10 |
1 files changed, 3 insertions, 7 deletions
diff --git a/resource_manager.py b/resource_manager.py index 752efe4..fb3b5f3 100644 --- a/resource_manager.py +++ b/resource_manager.py @@ -1,8 +1,8 @@ from operator import itemgetter import sys from BeautifulSoup import BeautifulSoup -import ParrotProtocol +from message import Message class ResourceManagerBase(object): resources = [ @@ -37,12 +37,8 @@ class ResourceManagerBase(object): def _create_message(self, resource, method, arg=None): assert resource in self.resources, 'Unknown resource {}'.format(resource) - assert method in self.resources[resource], 'Unhandled method {} for {}'.format( - method, resource) - if method == 'set': - return ParrotProtocol.setRequest(resource + '/' + method, str(arg).lower()) - else: - return ParrotProtocol.getRequest(resource + '/' + method) + assert method in self.resources[resource], 'Unhandled method {} for {}'.format(method, resource) + return Message(resource, method, arg) def send_message(self, message): try: |