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
9 from monitor.database.dborm import mon_metadata, mon_session
10 __metadata__ = mon_metadata
11 __session__ = mon_session
13 #class IssueType(Entity):
14 # shortname = Field(String, default=None)
15 # description = Field(String, default=None)
16 # issue_record = ManyToMany('IssueRecord')
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)
23 # hostname = Field(String,default=None)
24 # loginbase = Field(String)
26 # ticket_id = Field(Integer, default=0)
27 # rt = Field(PickleType, default=None)
29 # # open, paused, closed
30 # status = Field(String, default="open")
32 # take_action = Field(Boolean, default=False)
33 # send_email = Field(Boolean, default=True)
35 # message_series = Field(String, default="nodedown")
36 # message_index = Field(Integer, default=0)
37 # penalty_level = Field(Integer, default=0)
39 # issue_type = ManyToMany('IssueType')
40 # actions = OneToMany('ActionRecord', order_by='-date_created')
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'])
48 def neverExpires(self):
55 if self.neverExpires():
58 if self.date_created + timedelta(0,self.expires) > datetime.now():
64 if self.neverExpires():
67 return self.date_created + timedelta(0, self.expires)
69 class ActionRecord(Entity):
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()
76 date_created = Field(DateTime,default=datetime.now)
77 loginbase = Field(String,default=None)
78 hostname = Field(String,default=None)
80 # the expected kinds of actions are:
82 # * open ticket, send notice
84 # * apply penalty to site
85 # * backoff penalty to site
86 action = Field(String)
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)
93 message_id = Field(Integer, default=0)
94 penalty_level = Field(Integer, default=0)
96 # NOTE: in case an exception is thrown while trying to perform an action.
97 error_string = Field(String, default=None)
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.
104 # - create action record
105 # - find fbnode records
106 # - append fbnode records to action record
108 # - find fbnode records
109 # - create action record with fbnodes as argument
110 # findbad_records = OneToMany('FindbadNodeRecord', order_by='-date_checked')
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)
119 # NOTE: not sure this needs to be in the db.
120 #escellation_level = Field(Integer, default=0)
121 #stage = Field(String, default=None)