Rename; Removed deps from repo

jellypy/mobile_app.py

@@ -0,0 +1,194 @@
+# -*- coding: utf-8 -*-
+
+#  This file is part of Tautulli.
+#
+#  Tautulli is free software: you can redistribute it and/or modify
+#  it under the terms of the GNU General Public License as published by
+#  the Free Software Foundation, either version 3 of the License, or
+#  (at your option) any later version.
+#
+#  Tautulli is distributed in the hope that it will be useful,
+#  but WITHOUT ANY WARRANTY; without even the implied warranty of
+#  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+#  GNU General Public License for more details.
+#
+#  You should have received a copy of the GNU General Public License
+#  along with Tautulli.  If not, see <http://www.gnu.org/licenses/>.
+
+from __future__ import unicode_literals
+from future.builtins import str
+
+import requests
+import threading
+
+import jellypy
+if jellypy.PYTHON2:
+    import database
+    import helpers
+    import logger
+else:
+    from jellypy import database
+    from jellypy import helpers
+    from jellypy import logger
+
+
+TEMP_DEVICE_TOKEN = None
+INVALIDATE_TIMER = None
+
+_ONESIGNAL_APP_ID = '3b4b666a-d557-4b92-acdf-e2c8c4b95357'
+
+
+def set_temp_device_token(token=None):
+    global TEMP_DEVICE_TOKEN
+    TEMP_DEVICE_TOKEN = token
+
+    if TEMP_DEVICE_TOKEN:
+        logger._BLACKLIST_WORDS.add(TEMP_DEVICE_TOKEN)
+    else:
+        logger._BLACKLIST_WORDS.discard(TEMP_DEVICE_TOKEN)
+
+    if TEMP_DEVICE_TOKEN is not None:
+        global INVALIDATE_TIMER
+        if INVALIDATE_TIMER:
+            INVALIDATE_TIMER.cancel()
+        invalidate_time = 5 * 60  # 5 minutes
+        INVALIDATE_TIMER = threading.Timer(invalidate_time, set_temp_device_token, args=[None])
+        INVALIDATE_TIMER.start()
+
+
+def get_temp_device_token():
+    return TEMP_DEVICE_TOKEN
+
+
+def get_mobile_devices(device_id=None, device_token=None):
+    where = where_id = where_token = ''
+    args = []
+
+    if device_id or device_token:
+        where = 'WHERE '
+        if device_id:
+            where_id += 'device_id = ?'
+            args.append(device_id)
+        if device_token:
+            where_token = 'device_token = ?'
+            args.append(device_token)
+        where += ' AND '.join([w for w in [where_id, where_token] if w])
+
+    db = database.MonitorDatabase()
+    result = db.select('SELECT * FROM mobile_devices %s' % where, args=args)
+
+    return result
+
+
+def get_mobile_device_by_token(device_token=None):
+    if not device_token:
+        return None
+
+    return get_mobile_devices(device_token=device_token)
+
+
+def add_mobile_device(device_id=None, device_name=None, device_token=None, friendly_name=None, onesignal_id=None):
+    db = database.MonitorDatabase()
+
+    keys = {'device_id': device_id}
+    values = {'device_name': device_name,
+              'device_token': device_token,
+              'onesignal_id': onesignal_id,
+              'official': validate_onesignal_id(onesignal_id=onesignal_id)}
+
+    if friendly_name:
+        values['friendly_name'] = friendly_name
+
+    try:
+        result = db.upsert(table_name='mobile_devices', key_dict=keys, value_dict=values)
+        blacklist_logger()
+    except Exception as e:
+        logger.warn("Tautulli MobileApp :: Failed to register mobile device in the database: %s." % e)
+        return
+
+    if result == 'insert':
+        logger.info("Tautulli MobileApp :: Registered mobile device '%s' in the database." % device_name)
+    else:
+        logger.debug("Tautulli MobileApp :: Re-registered mobile device '%s' in the database." % device_name)
+
+    return True
+
+
+def get_mobile_device_config(mobile_device_id=None):
+    if str(mobile_device_id).isdigit():
+        mobile_device_id = int(mobile_device_id)
+    else:
+        logger.error("Tautulli MobileApp :: Unable to retrieve mobile device config: invalid mobile_device_id %s." % mobile_device_id)
+        return None
+
+    db = database.MonitorDatabase()
+    result = db.select_single('SELECT * FROM mobile_devices WHERE id = ?',
+                              args=[mobile_device_id])
+
+    return result
+
+
+def set_mobile_device_config(mobile_device_id=None, **kwargs):
+    if str(mobile_device_id).isdigit():
+        mobile_device_id = int(mobile_device_id)
+    else:
+        logger.error("Tautulli MobileApp :: Unable to set exisiting mobile device: invalid mobile_device_id %s." % mobile_device_id)
+        return False
+
+    keys = {'id': mobile_device_id}
+    values = {'friendly_name': kwargs.get('friendly_name', '')}
+
+    db = database.MonitorDatabase()
+    try:
+        db.upsert(table_name='mobile_devices', key_dict=keys, value_dict=values)
+        logger.info("Tautulli MobileApp :: Updated mobile device agent: mobile_device_id %s." % mobile_device_id)
+        blacklist_logger()
+        return True
+    except Exception as e:
+        logger.warn("Tautulli MobileApp :: Unable to update mobile device: %s." % e)
+        return False
+
+
+def delete_mobile_device(mobile_device_id=None, device_id=None):
+    db = database.MonitorDatabase()
+
+    if mobile_device_id:
+        logger.debug("Tautulli MobileApp :: Deleting mobile_device_id %s from the database." % mobile_device_id)
+        result = db.action('DELETE FROM mobile_devices WHERE id = ?', args=[mobile_device_id])
+        return True
+    elif device_id:
+        logger.debug("Tautulli MobileApp :: Deleting device_id %s from the database." % device_id)
+        result = db.action('DELETE FROM mobile_devices WHERE device_id = ?', args=[device_id])
+        return True
+    else:
+        return False
+
+
+def set_last_seen(device_token=None):
+    db = database.MonitorDatabase()
+
+    last_seen = helpers.timestamp()
+
+    try:
+        result = db.action('UPDATE mobile_devices SET last_seen = ? WHERE device_token = ?',
+                           args=[last_seen, device_token])
+    except Exception as e:
+        logger.warn("Tautulli MobileApp :: Failed to set last_seen time for device: %s." % e)
+        return
+
+
+def validate_onesignal_id(onesignal_id):
+    if onesignal_id is None:
+        return False
+
+    headers = {'Content-Type': 'application/json'}
+    payload = {'app_id': _ONESIGNAL_APP_ID}
+
+    r = requests.get('https://onesignal.com/api/v1/players/{}'.format(onesignal_id), headers=headers, json=payload)
+    return r.status_code == 200
+
+
+def blacklist_logger():
+    devices = get_mobile_devices()
+    for d in devices:
+        logger.blacklist_config(d)