def __init__(self, id=0):
'''
This initializes a user object. If id is given, loads that user. If not, creates a new user object with default values.
'''
logging.debug("Initializing user: %s" % id)
self.db = mysql.Database()
if id == 0:
self.name = ''
self.phone = ''
self.email = ''
self.state = 0
self.team = 'default'
self.lastAlert = 0
self.id = id
else:
self.load_user(id)
def query(q_string):
'''
Query the db with the given string and return with an array of user objects.
'''
try:
_db = mysql.Database()
_db._cursor.execute('''%s''' % (q_string))
logging.debug("Running mysql query: %s" % q_string)
temp = _db._cursor.fetchall()
users = []
for t in temp:
t = mysql.users_convert_to_dict(t)
users.append(User(t['id']))
return users
except Exception, e:
logging.error(e.__str__())
def __init__(self, id=0):
'''
Initialize the alert object.
If id is given, load alert with that id. Otherwise, create a new alert with default attributes.
'''
logging.debug("Initializing alert: %s" % id)
self.db = mysql.Database()
if id == 0:
self.subject = ''
self.message = ''
self.team = 'default'
self.ack = 0
self.ackby = 0
self.acktime = ''
self.lastAlertSent = ''
self.tries = 0
self.id = id
else:
self.load_alert(id)
return
if self.state != None:
try:
self.state = int(self.state)
except Exception, e:
self.populate(1004, "State is not a integer")
return
if self.user_id != 0:
try:
self.user_id = int(self.user_id)
except Exception, e:
self.populate(1005, "User_id is not a integer")
return
#init database connection
self.db = mysql.Database()
if self.action.lower() == "query":
self.query()
elif self.action.lower() == "create":
self.create()
elif self.action.lower() == "edit":
self.edit()
elif self.action.lower() == "delete":
self.delete()
else:
self.populate(1001, "Invalid action type")
def query(self):
if self.target == "alert" or self.target == "alerts":
objects = Alert.all_alerts()