Generate a unique token for each mobile device

plexpy/mobile_app.py

@@ -19,6 +19,9 @@ import helpers
 import logger
 
 
+TEMP_DEVICE_TOKEN = None
+
+
 def get_mobile_devices(device_id=None, device_token=None):
     where = where_id = where_token = ''
     args = []
@@ -33,19 +36,57 @@ def get_mobile_devices(device_id=None, device_token=None):
             args.append(device_token)
         where += ' AND '.join([w for w in [where_id, where_token] if w])
 
-    monitor_db = database.MonitorDatabase()
-    result = monitor_db.select('SELECT * FROM mobile_devices %s' % where, args=args)
+    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):
+    db = database.MonitorDatabase()
+
+    keys = {'device_id': device_id}
+    values = {'device_name': device_name,
+              'device_token': device_token}
+
+    try:
+        result = db.upsert(table_name='mobile_devices', key_dict=keys, value_dict=values)
+    except Exception as e:
+        logger.warn(u"PlexPy MobileApp :: Failed to register mobile device in the database: %s." % e)
+        return
+
+    if result == 'insert':
+        logger.info(u"PlexPy MobileApp :: Registered mobile device '%s' in the database." % device_name)
+    else:
+        logger.debug(u"PlexPy MobileApp :: Re-registered mobile device '%s' in the database." % device_name)
+
+    return True
+
+
 def delete_mobile_device(device_id=None):
-    monitor_db = database.MonitorDatabase()
+    db = database.MonitorDatabase()
 
     if device_id:
-        logger.debug(u"PlexPy Notifiers :: Deleting device_id %s from the database." % device_id)
-        result = monitor_db.action('DELETE FROM mobile_devices WHERE device_id = ?', args=[device_id])
+        logger.debug(u"PlexPy 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 blacklist_logger():
+    devices = get_mobile_devices()
+
+    blacklist = []
+
+    for d in devices:
+        blacklist.append(d['device_token'])
+
+    logger._BLACKLIST_WORDS.extend(blacklist)