scheduler.py

#!/usr/bin/env python
# -*- coding: utf-8 -*-
#
# combacscheduler.py
#
#       Copyright 2014 BFR <info@freie-radios.de>
#
#       This program 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; Version 3 of the License
#
#       This program 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 this program; if not, the license can be downloaded here:
#
#       http://www.gnu.org/licenses/gpl.html

# Meta
__version__ = '0.1.1'
__license__ = "GNU General Public License (GPL) Version 3"
__version_info__ = (0, 1, 1)
__author__ = 'Michael Liebler <michael-liebler@radio-z.net>'

"""
Comba Scheduler Klasse 
Is holding the eventqueue
"""
import signal
import pyev
import os
import os.path
import time
import simplejson
import datetime
import decimal
import json
import traceback
import sqlalchemy
import MySQLdb
import sys

from datetime import timedelta
from dateutil.relativedelta import relativedelta
import logging
from glob import glob
import threading

# Die eigenen Bibliotheken
from libraries.base.schedulerconfig import AuraSchedulerConfig
from modules.communication.redis.messenger import RedisMessenger
from libraries.base.calendar import AuraCalendarService
from modules.scheduling.models import ModelBroadcastEventOverrides
from libraries.database.broadcasts import ScheduleEntry, Model
from libraries.exceptions.auraexceptions import NoProgrammeLoadedException


def alchemyencoder(obj):
    """JSON encoder function for SQLAlchemy special classes."""
    if isinstance(obj, datetime.date):
        return obj.isoformat()
    elif isinstance(obj, decimal.Decimal):
        return float(obj)
    elif isinstance(obj, sqlalchemy.orm.state.InstanceState):
        return ""
#    elif isinstance(obj, bool):
#        if obj:
#            return "True"
#        return "False"
    else:
        return obj

"""
Comba Scheduler Class 
Gets data from pv and importer, stores and fires events, 
Liefert Start und Stop Jobs an den Comba Controller, lädt XML-Playlisten und räumt auf 
"""


class AuraScheduler():
    redismessenger = RedisMessenger()
    liquidsoapcommunicator = None
    schedule_entries = None
    message_timer = []

    schedulerconfig = None
    job_result = {}
    programme = None
    client = None
    debug = False
    active_entry = None

    def __init__(self, config): #, liquidsoap_client):
        """
        Constructor
        @type    config:               string
        @param   config:               Pfad zur aura.ini
        """
        self.auraconfig = config
        self.debug = config.get("debug")

        # Messenger für Systemzustände initieren
        self.redismessenger.set_channel('scheduler')
        self.redismessenger.set_section('execjob')
        self.redismessenger.set_mail_addresses(self.auraconfig.get('frommail'), self.auraconfig.get('adminmail'))

        self.schedulerconfig = self.auraconfig.get("scheduler_config_file")

        # Die Signale, die Abbruch signalisieren
        self.stopsignals = (signal.SIGTERM, signal.SIGINT)

        # das pyev Loop-Object
        self.loop = pyev.default_loop()

        # Das ist kein Reload
        self.initial = True

        # Der Scheduler wartet noch auf den Start Befehl
        self.ready = False

        # DIe Config laden
        self.__load_config__()

        self.scriptdir = os.path.dirname(os.path.abspath(__file__)) + '/..'

        #errors_file = os.path.dirname(os.path.realpath(__file__)) + '/error/scheduler_error.js'
        json_data = open(self.auraconfig.get("install_dir") + "/errormessages/scheduler_error.js")
        self.errorData = simplejson.load(json_data)

        # init database ?
        self.init_database()

        self.redismessenger.send('Scheduler started', '0000', 'success', 'initApp', None, 'appinternal')

    # ------------------------------------------------------------------------------------------ #
    def init_database(self):
        # check if tables do exist. if not create them
        try:
            ScheduleEntry.select_all()
        except sqlalchemy.exc.ProgrammingError as e:
            if e.__dict__["code"] == "f405":
                Model.recreate_db()
            else:
                raise


    # ------------------------------------------------------------------------------------------ #
#    def set(self, key, value):
#        """
#        Eine property setzen
#        @type    key: string
#        @param   key: Der Key
#        @type    value: mixed
#        @param   value: Beliebiger Wert
#        """
#        self.__dict__[key] = value

    # ------------------------------------------------------------------------------------------ #
 #   def get(self, key, default=None):
 #       """
 #       Eine property holen
 #       @type    key: string
 #       @param   key: Der Key
 #       @type    default: mixed
 #       @param   default: Beliebiger Wert#
#
#        """
 #       if key not in self.__dict__:
#            if default:
#                self.set(key, default)
#            else:
#                return None
#        return self.__dict__[key]

    # ------------------------------------------------------------------------------------------ #
    def reload_config(self):
        """
        Reload Scheduler - Config neu einlesen
        """
        self.stop()
        # Scheduler Config neu laden
        if self.__load_config__():
            self.redismessenger.send('Scheduler reloaded by user', '0500', 'success', 'reload', None, 'appinternal')
        self.start()

    # ------------------------------------------------------------------------------------------ #
    def get_active_source(self):
        now_unix = time.mktime(datetime.datetime.now().timetuple())
        actsource = ""
        lastentry = None

        # load programme if necessary
        if self.programme is None:
            print("want to get active channel, but have to load programme first")
            self.load_programme_from_db()

        # get active source