#!/usr/bin/env python

# -*- coding: utf-8 -*-

# Copyright (C) 2009-2012 :
#     Gabes Jean, naparuba@gmail.com
#     Gerhard Lausser, Gerhard.Lausser@consol.de
#     Gregory Starck, g.starck@gmail.com
#     Hartmut Goebel, h.goebel@goebel-consult.de
#
# This file is part of Shinken.
#
# Shinken is free software: you can redistribute it and/or modify
# it under the terms of the GNU Affero General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# Shinken 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 Affero General Public License for more details.
#
# You should have received a copy of the GNU Affero General Public License
# along with Shinken.  If not, see <http://www.gnu.org/licenses/>.



from shinken.satellitelink import SatelliteLink, SatelliteLinks
from shinken.property import BoolProp, IntegerProp, StringProp, ListProp

from shinken.pyro_wrapper import Pyro
from shinken.log  import logger


class SchedulerLink(SatelliteLink):
    """Please Add a Docstring to describe the class here"""
    
    id = 0

    # Ok we lie a little here because we are a mere link in fact
    my_type = 'scheduler'

    properties = SatelliteLink.properties.copy()
    properties.update({
        'scheduler_name':     StringProp(fill_brok=['full_status']),
        'port':               IntegerProp(default='7768', fill_brok=['full_status']),
        'weight':             IntegerProp(default='1', fill_brok=['full_status']),
        'skip_initial_broks': BoolProp(default='0', fill_brok=['full_status']),
    })
    
    running_properties = SatelliteLink.running_properties.copy() 
    running_properties.update({
        'conf':      StringProp(default=None),
        'need_conf': StringProp(default=True),
        'external_commands' : StringProp(default=[]),
        'push_flavor' : IntegerProp(default=0),
    })


    def get_name(self):
        return self.scheduler_name


    def run_external_commands(self, commands):
        if self.con is None:
            self.create_connection()
        if not self.alive:
            return None
        logger.debug("[SchedulerLink] Sending %d commands" % len(commands))
        try:
            self.con.run_external_commands(commands)
        except Pyro.errors.URIError , exp:
            self.con = None
            return False
        except Pyro.errors.ProtocolError , exp:
            self.con = None
            return False
        except TypeError , exp:            
            try:
                exp = ''.join(Pyro.util.getPyroTraceback(exp))
            except:
                pass

            logger.debug(exp)
        except Pyro.errors.CommunicationError , exp:
            self.con = None
            return False
        except Exception, exp:
            try:
                exp = ''.join(Pyro.util.getPyroTraceback(exp))
            except:
                pass
            
            logger.debug(exp)
            self.con = None
            return False



    def register_to_my_realm(self):
        self.realm.schedulers.append(self)


    def give_satellite_cfg(self):
        return {'port' : self.port, 'address' : self.address, 'name' : self.scheduler_name, 'instance_id' : self.id, 'active' : self.conf is not None,
                'push_flavor' : self.push_flavor}


    # Some parameters can give as 'overriden parameters' like use_timezone
    # so they will be mixed (in the scheduler) with the standard conf sent by the arbiter
    def get_override_configuration(self):
        r = {}
        properties = self.__class__.properties
        for prop, entry in properties.items():
            if entry.override:
                r[prop] = getattr(self, prop)
        return r




class SchedulerLinks(SatelliteLinks):
    """Please Add a Docstring to describe the class here"""
    
    name_property = "scheduler_name"
    inner_class = SchedulerLink

@@ -1226,29 +1226,25 @@
 ass here%22%22%22%0A
-    
 %0A
+
     id = 0%0A%0A
@@ -1733,20 +1733,16 @@
 %0A    %7D)%0A
-    
 %0A    run
@@ -1798,17 +1798,16 @@
 s.copy()
- 
 %0A    run
@@ -1952,17 +1952,16 @@
 ommands'
- 
 : String
@@ -2433,25 +2433,24 @@
 ors.URIError
- 
 , exp:%0A     
@@ -2533,25 +2533,24 @@
 rotocolError
- 
 , exp:%0A     
@@ -2625,27 +2625,14 @@
 rror
- 
 , exp:
-            
 %0A   
@@ -2825,17 +2825,16 @@
 ionError
- 
 , exp:%0A 
@@ -3041,21 +3041,9 @@
 ass%0A
-            
 %0A
+
     
@@ -3261,17 +3261,16 @@
  %7B'port'
- 
 : self.p
@@ -3283,17 +3283,16 @@
 address'
- 
 : self.a
@@ -3305,17 +3305,16 @@
 , 'name'
- 
 : self.s
@@ -3341,17 +3341,16 @@
 ance_id'
- 
 : self.i
@@ -3360,17 +3360,16 @@
 'active'
- 
 : self.c
@@ -3414,17 +3414,16 @@
 _flavor'
- 
 : self.p
@@ -3948,20 +3948,16 @@
 here%22%22%22%0A
-    
 %0A    nam

Fix style #373