monitor/database/info/action.py

   1 from elixir import Entity, Field, OneToMany, ManyToOne, ManyToMany
   2 from elixir import options_defaults, using_options, setup_all, has_one
   3 from elixir import String, Integer, DateTime, PickleType, Boolean
   4 from elixir.ext.versioned import *
   5 from datetime import datetime,timedelta
   6 import elixir
   7 import traceback
   8
   9 from monitor.database.dborm import mon_metadata, mon_session
  10 __metadata__ = mon_metadata
  11 __session__  = mon_session
  12
  13 #class IssueType(Entity):
  14 #       shortname = Field(String, default=None)
  15 #       description = Field(String, default=None)
  16 #       issue_record = ManyToMany('IssueRecord')
  17
  18 #class IssueRecord(Entity):
  19 #       date_created = Field(DateTime,default=datetime.now)
  20 #       date_last_updated = Field(DateTime,default=datetime.now)
  21 #       date_action_taken = Field(DateTime,default=datetime.now)
  22 #
  23 #       hostname = Field(String,default=None)
  24 #       loginbase = Field(String)
  25 #
  26 #       ticket_id = Field(Integer, default=0)
  27 #       rt = Field(PickleType, default=None)
  28 #
  29 #       # open, paused, closed
  30 #       status = Field(String, default="open")
  31 #
  32 #       take_action = Field(Boolean, default=False)
  33 #       send_email = Field(Boolean, default=True)
  34 #
  35 #       message_series =  Field(String, default="nodedown")
  36 #       message_index = Field(Integer, default=0)
  37 #       penalty_level = Field(Integer, default=0)
  38 #
  39 #       issue_type = ManyToMany('IssueType')
  40 #       actions = OneToMany('ActionRecord', order_by='-date_created')
  41
  42 class BlacklistRecord(Entity):
  43         date_created = Field(DateTime,default=datetime.now)
  44         hostname = Field(String,default=None, primary_key=True)
  45         expires = Field(Integer,default=0)      # seconds plus
  46         acts_as_versioned(['hostname'])
  47
  48         def neverExpires(self):
  49                 if self.expires == 0:
  50                         return True
  51                 else:
  52                         return False
  53
  54         def expired(self):
  55                 if self.neverExpires():
  56                         return False
  57                 else:
  58                         if self.date_created + timedelta(0,self.expires) > datetime.now():
  59                                 return True
  60                         else:
  61                                 return False
  62
  63         def willExpire(self):
  64                 if self.neverExpires():
  65                         return "never"
  66                 else:
  67                         return self.date_created + timedelta(0, self.expires)
  68
  69 class ActionRecord(Entity):
  70         @classmethod
  71         def get_latest_by(cls, **kwargs):
  72                 # TODO: need to sort on 'round' since actions will not be globally sync'd.
  73                 return cls.query.filter_by(**kwargs).order_by(ActionRecord.id.desc()).first()
  74
  75 # ACCOUNTING
  76         date_created = Field(DateTime,default=datetime.now)
  77         loginbase = Field(String,default=None)
  78         hostname = Field(String,default=None)
  79         # NOTE:
  80         #       the expected kinds of actions are:
  81         #               * reboot node
  82         #               * open ticket, send notice
  83         #               * close ticket
  84         #               * apply penalty to site
  85         #               * backoff penalty to site
  86         action = Field(String)
  87
  88         # NOTE: describes the kind of action.  i.e. online-notice, offline-notice,
  89         # reboot-first-try, reboot-second-try, penalty-pause, penalty-warning, penalty-no-create,
  90         # penalty-disable-slices,
  91         action_type = Field(String, default=None)
  92
  93         message_id = Field(Integer, default=0)
  94         penalty_level = Field(Integer, default=0)
  95
  96         # NOTE: in case an exception is thrown while trying to perform an action.
  97         error_string = Field(String, default=None)
  98
  99         #issue = ManyToOne('IssueRecord')
 100         # NOTE: this is the parent relation to fb records.  first create the
 101         # action record, then append to this value all of the findbad records we
 102         # want to have in our set.
 103         # Model:
 104         #    - create action record
 105         #    - find fbnode records
 106         #    - append fbnode records to action record
 107         #  OR
 108         #    - find fbnode records
 109         #    - create action record with fbnodes as argument
 110         # findbad_records = OneToMany('FindbadNodeRecord', order_by='-date_checked')
 111
 112         # NOTE: can I move 'message_index, escellation_level, and penalty_level'
 113         #    into the same value?  Maybe not penalty level, since there are only two;
 114         #    and, there may be additional message and escellation levels.
 115         #send_email_to = Field(PickleType, default=None)
 116         #action_description = Field(PickleType, default=None)
 117         #message_arguments = Field(PickleType, default=None)
 118
 119         # NOTE: not sure this needs to be in the db.
 120         #escellation_level = Field(Integer, default=0)
 121         #stage = Field(String, default=None)