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)
45 loginbase = Field(String,default=None)
46 expires = Field(Integer,default=0) # seconds plus
47 acts_as_versioned(['hostname'])
50 def getLoginbaseBlacklist(cls):
51 # TODO: need to sort on 'round' since actions will not be globally sync'd.
52 return cls.query.filter(cls.loginbase!=None).order_by(cls.loginbase.desc())
55 def getHostnameBlacklist(cls):
56 # TODO: need to sort on 'round' since actions will not be globally sync'd.
57 return cls.query.filter(cls.hostname!=None).order_by(cls.hostname.desc())
59 def neverExpires(self):
66 if self.neverExpires():
69 if self.date_created + timedelta(0,self.expires) > datetime.now():
75 if self.neverExpires():
78 return self.date_created + timedelta(0, self.expires)
80 class ActionRecord(Entity):
82 def get_latest_by(cls, **kwargs):
83 # TODO: need to sort on 'round' since actions will not be globally sync'd.
84 return cls.query.filter_by(**kwargs).order_by(ActionRecord.id.desc()).first()
87 date_created = Field(DateTime,default=datetime.now)
88 loginbase = Field(String,default=None)
89 hostname = Field(String,default=None)
91 # the expected kinds of actions are:
93 # * open ticket, send notice
95 # * apply penalty to site
96 # * backoff penalty to site
97 action = Field(String)
99 # NOTE: describes the kind of action. i.e. online-notice, offline-notice,
100 # reboot-first-try, reboot-second-try, penalty-pause, penalty-warning, penalty-no-create,
101 # penalty-disable-slices,
102 action_type = Field(String, default=None)
104 message_id = Field(Integer, default=0)
105 penalty_level = Field(Integer, default=0)
107 # NOTE: in case an exception is thrown while trying to perform an action.
108 error_string = Field(String, default=None)
110 #issue = ManyToOne('IssueRecord')
111 # NOTE: this is the parent relation to fb records. first create the
112 # action record, then append to this value all of the findbad records we
113 # want to have in our set.
115 # - create action record
116 # - find fbnode records
117 # - append fbnode records to action record
119 # - find fbnode records
120 # - create action record with fbnodes as argument
121 # findbad_records = OneToMany('FindbadNodeRecord', order_by='-date_checked')
123 # NOTE: can I move 'message_index, escellation_level, and penalty_level'
124 # into the same value? Maybe not penalty level, since there are only two;
125 # and, there may be additional message and escellation levels.
126 #send_email_to = Field(PickleType, default=None)
127 #action_description = Field(PickleType, default=None)
128 #message_arguments = Field(PickleType, default=None)
130 # NOTE: not sure this needs to be in the db.
131 #escellation_level = Field(Integer, default=0)
132 #stage = Field(String, default=None)