Add support of socket

georideapilib/api.py

@@ -16,6 +16,12 @@ from georideapilib.objects import (
     GeorideSharedTrip
 )
 
+from georideapilib.exception import (
+    LoginException,
+    SeverException,
+    UnauthorizedException
+)
+
 GEORIDE_API_HOST = "https://api.georide.fr"
 GEORIDE_API_ENDPOINT_LOGIN = "/user/login"
 GEORIDE_API_ENDPOINT_NEW_TOKEN = "/user/new-token"
@@ -33,16 +39,25 @@ _SESSION = requests.Session()
 _LOGGER = logging.getLogger(__name__)
 
 def get_authorisation_token(email, password):
-    """ return an authorization token """
+    """ return an authorization token or no"""
     data = {"email": email, "password": password}
     encoded_data = json.dumps(data).encode('utf-8')
+
     response = _SESSION.post( 
         GEORIDE_API_HOST + GEORIDE_API_ENDPOINT_LOGIN,
         data=encoded_data,
         headers={'Content-Type': 'application/json'})
-    response_data = response.json()
-    _LOGGER.debug(response_data)
-    account = GeorideAccount.from_json(response_data)
+    
+    if response.status_code == 200:
+        _LOGGER.debug("Login success")
+        response_data = response.json()
+        account = GeorideAccount.from_json(response_data)
+    elif response.status_code == 403:
+        _LOGGER.warnning("Login failed")
+        raise LoginException("Login failed")
+    else:
+        _LOGGER.error("Georide login, http error code: %s", response.status_code)
+        raise SeverException("Georide login, http error code: {}".format(response.status_code))
     return account
 
 
@@ -52,19 +67,29 @@ def renew_token(token):
     response = _SESSION.get(
         GEORIDE_API_HOST + GEORIDE_API_ENDPOINT_NEW_TOKEN,
         headers=headers)
-    response_data = response.json()
-    _LOGGER.debug(response_data)
-    new_token = response_data['authToken']
+    if response.status_code == 200:
+        _LOGGER.debug("Token updated")
+        response_data = response.json()
+        new_token = response_data['authToken']
+    elif response.status_code == 401:
+        _LOGGER.warnning("Renew token refused")
+        raise UnauthorizedException("Renew token refused")
+    else:
+        _LOGGER.error("Georide login, http error code: %s", response.status_code)
+        raise SeverException("Georide login, http error code: {}".format(response.status_code))
     return new_token
 
-
 def revoke_token(token):
     """ invalidate the authorization token """
     headers = {"Authorization": "Bearer " + token}
     response = _SESSION.post(
         GEORIDE_API_HOST + GEORIDE_API_ENDPOINT_LOGOUT,
         headers=headers)
-    if response.status_code != 204:
+    if response.status_code == 401:
+        _LOGGER.warnning("Token allready revoked")
+        raise UnauthorizedException("Token allready revoked")
+    if response.status_code == 401:
+        _LOGGER.warnning("Token allready revoked")
         return False
     return True
 

georideapilib/exception.py

@@ -0,0 +1,23 @@
+""" all geroide exception """
+
+class Error(Exception):
+    """Base class for exceptions in this module."""
+    pass
+
+class LoginException(Error):
+    """loggin exception"""
+    def __init__(self, expression, message):
+        self.expression = expression
+        self.message = message
+
+class UnauthorizedException(Error):
+    """loggin exception"""
+    def __init__(self, expression, message):
+        self.expression = expression
+        self.message = message
+
+class SeverException(Error):
+    """loggin exception"""
+    def __init__(self, expression, message):
+        self.expression = expression
+        self.message = message

georideapilib/socket.py

@@ -0,0 +1,120 @@
+""" Georide socket-io implementation """
+import logging
+import socketio
+
+from georideapilib.api import GEORIDE_API_HOST
+
+_LOGGER = logging.getLogger(__name__)
+_LOGGER.setLevel(logging.DEBUG)
+# create console handler and set level to debug
+CH = logging.StreamHandler()
+CH.setLevel(logging.DEBUG)
+
+sio = socketio.Client(reconnection=True)
+
+@sio.on('connect')
+def on_connect():
+    """ event: connected """
+    _LOGGER.debug('GeoRide socket connected')
+
+@sio.on('disconnect')
+def on_disconnect():
+    """ event: disconnected """
+    _LOGGER.debug('GeoRide socket disconnected')
+
+
+class GeorideSocket():
+    """docstring for GeorideSocket"""
+    def __init__(self):
+        self._on_message_callback = None
+        self._on_device_callback = None
+        self._on_position_callback = None
+        self._on_alarm_callback = None
+        self._on_refresh_tracker_callback = None
+        self._on_locked_callback = None
+        self._initialised = False
+
+
+    def subscribe_on_message(self, callback_function):
+        """event: tells you authentication informations."""
+        self._on_message_callback = callback_function
+
+    def subscribe_device(self, callback_function):
+        """event: tells you when a device is added to the account."""
+        self._on_device_callback = callback_function
+
+    def subscribe_position(self, callback_function):
+        """event: tells you when a device sent a position."""
+        self._on_position_callback = callback_function
+
+    def subscribe_alarm(self, callback_function):
+        """event: tells you when a device trigger an alarm."""
+        self._on_alarm_callback = callback_function
+
+    def subscribe_refresh_tracker(self, callback_function):
+        """event: tells you when you need to refresh your list of trackers"""
+        self._on_refresh_tracker_callback = callback_function
+
+    def subscribe_locked(self, callback_function):
+        """event: tells you when a device has been locked or unlocked."""
+        self._on_locked_callback = callback_function
+
+
+    def init(self):
+        """init the context"""
+        @sio.on('message')
+        def on_message(data):
+            """ on_message """
+            _LOGGER.debug('Message recieved: %s', data)
+            if self._on_message_callback is not None:
+                self._on_message_callback()
+        
+        @sio.on('device')
+        def on_device(data):
+            """ on_device """
+            _LOGGER.debug('Device recieved: %s', data)
+            if self._on_device_callback is not None:
+                self._on_device_callback()
+        
+        @sio.on('position')
+        def on_position(data):
+            """ on_position """
+            _LOGGER.debug('Position recieved:%s', data)
+            if self._on_position_callback is not None:
+                self._on_position_callback()
+
+        @sio.on('alarm')
+        def on_alarm(data):
+            """ on_alarm """
+            _LOGGER.debug('Alarm recieved: %s', data)
+            if self._on_alarm_callback is not None:
+                self._on_alarm_callback(data)
+            
+        @sio.on('refreshTrackersInstruction')
+        def on_refresh_tracker():
+            """ on_refresh_tracker """
+            _LOGGER.debug('Refresh tracker recieved')
+            if self._on_refresh_tracker_callback is not None:
+                self._on_refresh_tracker_callback()
+
+        @sio.on('lockedPosition')
+        def on_locked(data):
+            """ on_locked """
+            _LOGGER.debug('Locked recieved: %s', data)
+            if self._on_locked_callback is not None:
+                self._on_locked_callback()
+
+        self._initialised = True
+
+    def connect(self, auth_token):
+        """ connect to the georide socket"""
+        _LOGGER.info("Start conection")
+        if self._initialised is not False:
+            sio.connect(GEORIDE_API_HOST, headers={'token': auth_token})
+            sio.wait()
+        else:
+            _LOGGER.error("Please call init() before")
+        
+    def disconnect(self):
+        """disconnect from the georide socket"""
+        sio.disconnect()

setup.py

@@ -19,7 +19,7 @@ CURRENT_DIR = os.path.dirname(__file__)
 setup(
     name='georideapilib',
     packages=['georideapilib'],  # this must be the same as the name above
-    version='0.2.0',
+    version='0.3.0',
     description='Lib to control georide tracker devices with their rest api',
     author='Matthieu DUVAL',
     author_email='georideapilib@duval-dev.fr',
@@ -28,7 +28,7 @@ setup(
     download_url='https://codeload.github.com/ptimatth/pyGeoride/tar.gz/0.1.0',
     keywords=['rest', 'georide', 'api', 'grutier'],  # arbitrary keywords
     classifiers=[],
-    install_requires=[],
+    install_requires=["python-socketio[client]"],
     tests_require=[
         'pytest>=3.7',
         'pytest-pep8',