CC-5990, CC-5991 - Python cleanup, removed need for /usr/lib/airtime

python_apps/media-monitor/media_monitor/mm1.py

@@ -0,0 +1,142 @@
+# -*- coding: utf-8 -*-
+
+import time
+import logging.config
+import sys
+import os
+import traceback
+import locale
+
+from configobj import ConfigObj
+
+from api_clients import api_client as apc
+from std_err_override import LogWriter
+
+from multiprocessing import Queue as mpQueue
+
+from threading import Thread
+
+from pyinotify import WatchManager
+
+from airtimefilemonitor.airtimenotifier import AirtimeNotifier
+from airtimefilemonitor.mediamonitorcommon import MediaMonitorCommon
+from airtimefilemonitor.airtimeprocessevent import AirtimeProcessEvent
+from airtimefilemonitor.mediaconfig import AirtimeMediaConfig
+from airtimefilemonitor.workerprocess import MediaMonitorWorkerProcess
+from airtimefilemonitor.airtimemediamonitorbootstrap import AirtimeMediaMonitorBootstrap
+
+def configure_locale():
+    logger.debug("Before %s", locale.nl_langinfo(locale.CODESET))
+    current_locale = locale.getlocale()
+
+    if current_locale[1] is None:
+        logger.debug("No locale currently set. Attempting to get default locale.")
+        default_locale = locale.getdefaultlocale()
+
+        if default_locale[1] is None:
+            logger.debug("No default locale exists. Let's try loading from /etc/default/locale")
+            if os.path.exists("/etc/default/locale"):
+                config = ConfigObj('/etc/default/locale')
+                lang = config.get('LANG')
+                new_locale = lang
+            else:
+                logger.error("/etc/default/locale could not be found! Please run 'sudo update-locale' from command-line.")
+                sys.exit(1)
+        else:
+            new_locale = default_locale
+
+        logger.info("New locale set to: %s", locale.setlocale(locale.LC_ALL, new_locale))
+
+
+
+    reload(sys)
+    sys.setdefaultencoding("UTF-8")
+    current_locale_encoding = locale.getlocale()[1].lower()
+    logger.debug("sys default encoding %s", sys.getdefaultencoding())
+    logger.debug("After %s", locale.nl_langinfo(locale.CODESET))
+
+    if current_locale_encoding not in ['utf-8', 'utf8']:
+        logger.error("Need a UTF-8 locale. Currently '%s'. Exiting..." % current_locale_encoding)
+        sys.exit(1)
+
+# configure logging
+try:
+    logging.config.fileConfig("%s/logging.cfg" % os.path.dirname(os.path.realpath(__file__)))
+
+    #need to wait for Python 2.7 for this..
+    #logging.captureWarnings(True)
+
+    logger = logging.getLogger()
+    LogWriter.override_std_err(logger)
+
+except Exception, e:
+    print 'Error configuring logging: ', e
+    sys.exit(1)
+
+logger.info("\n\n*** Media Monitor bootup ***\n\n")
+
+
+try:
+    configure_locale()
+
+    config = AirtimeMediaConfig(logger)
+    api_client = apc.AirtimeApiClient()
+    api_client.register_component("media-monitor")
+
+    logger.info("Setting up monitor")
+    response = None
+    while response is None:
+        response = api_client.setup_media_monitor()
+        time.sleep(5)
+
+    storage_directory = response["stor"]
+    watched_dirs = response["watched_dirs"]
+    logger.info("Storage Directory is: %s", storage_directory)
+    config.storage_directory = os.path.normpath(storage_directory)
+    config.imported_directory = os.path.normpath(os.path.join(storage_directory, 'imported'))
+    config.organize_directory = os.path.normpath(os.path.join(storage_directory, 'organize'))
+    config.recorded_directory = os.path.normpath(os.path.join(storage_directory, 'recorded'))
+    config.problem_directory = os.path.normpath(os.path.join(storage_directory, 'problem_files'))
+
+    dirs = [config.imported_directory, config.organize_directory, config.recorded_directory, config.problem_directory]
+    for d in dirs:
+        if not os.path.exists(d):
+            os.makedirs(d, 02775)
+
+    multi_queue = mpQueue()
+    logger.info("Initializing event processor")
+
+    wm = WatchManager()
+    mmc = MediaMonitorCommon(config, wm=wm)
+    pe = AirtimeProcessEvent(queue=multi_queue, airtime_config=config, wm=wm, mmc=mmc, api_client=api_client)
+
+    bootstrap = AirtimeMediaMonitorBootstrap(logger, pe, api_client, mmc, wm, config)
+    bootstrap.scan()
+
+    notifier = AirtimeNotifier(wm, pe, read_freq=0, timeout=0, airtime_config=config, api_client=api_client, bootstrap=bootstrap, mmc=mmc)
+    notifier.coalesce_events()
+
+    #create 5 worker threads
+    wp = MediaMonitorWorkerProcess(config, mmc)
+    for i in range(5):
+        threadName = "Thread #%d" % i
+        t = Thread(target=wp.process_file_events, name=threadName, args=(multi_queue, notifier))
+        t.start()
+
+    wdd = notifier.watch_directory(storage_directory)
+    logger.info("Added watch to %s", storage_directory)
+    logger.info("wdd result %s", wdd[storage_directory])
+
+    for dir in watched_dirs:
+        wdd = notifier.watch_directory(dir)
+        logger.info("Added watch to %s", dir)
+        logger.info("wdd result %s", wdd[dir])
+
+    notifier.loop(callback=pe.notifier_loop_callback)
+
+except KeyboardInterrupt:
+    notifier.stop()
+    logger.info("Keyboard Interrupt")
+except Exception, e:
+    logger.error('Exception: %s', e)
+    logger.error("traceback: %s", traceback.format_exc())