def getRobot_byCond(db, robotId=None, robotName=None):
if (not robotId and not robotName):
return None
if (robotId):
cond = "WHERE ID='%s'" % (robotId)
else:
cond = "WHERE R_Name='%s'" % (robotName)
dbObj = db.fetchone(TB_ROBOT, cond=cond)
if (not dbObj):
WARNING("robot %s not exist" % cond)
return None
robot = WCRobot(db, dbObj=dbObj)
robot.loadFromObj()
return robot
def add(self):
userObj = {
"ID": getUuid(),
"Q_Robots": self.robots,
"Q_Message": self.messageCapacity,
"Q_Group": self.group,
"Q_LastSync": get_current_time(),
"Q_CreateTime": get_current_time()
}
ret = self.db.insert(TB_QUOTA, userObj)
if (ret == -1):
WARNING("add user %s error for db operation" % self.myId)
return DB_ERR
DEBUG(userObj)
return OCT_SUCCESS
def update(self):
self.infoObj["name"] = self.name
userObj = {
"P_Name": self.name,
"P_State": self.state,
"P_Info": transToStr(self.infoObj),
"P_LastSync": get_current_time(),
"P_Description": self.desc
}
cond = "WHERE ID='%s'" % self.myId
ret = self.db.update(TB_PRODUCT, userObj, cond=cond)
if (ret == -1):
WARNING("update user %s error for db operation" % self.name)
return DB_ERR
return 0
def add(self):
robotObj = {
"ID": getUuid(),
"R_UId": self.uId,
"R_AccountId": self.accountId,
"R_Name": self.name,
"R_CreateTime": get_current_time(),
"R_LastSync": get_current_time(),
}
ret = self.db.insert(TB_ROBOT, robotObj)
if (ret == -1):
WARNING("add robot %s error for db operation" % self.name)
return DB_ERR
DEBUG(robotObj)
return OCT_SUCCESS
def resetPassword(self, newPassword):
if (not newPassword):
ERROR(
"new password not specified when doing password reset operation"
)
return NOT_ENOUGH_PARAS
userObj = {
"U_Password": getEncPass(newPassword),
"U_LastSync": get_current_time(),
}
cond = "WHERE ID='%s'" % self.myId
ret = self.db.update(TB_ACCOUNT, userObj, cond=cond)
if (ret == -1):
WARNING("update user %s error for db operation" % self.name)
return DB_ERR
return OCT_SUCCESS
def getAccount(db, userId=None, userName=None, uid=None):
if (not userId and not userName and not uid):
return None
if (userId):
cond = "WHERE ID='%s'" % (userId)
elif userName:
cond = "WHERE U_Name='%s'" % (userName)
else:
cond = "WHERE U_LdapUid='%s'" % (uid)
dbObj = db.fetchone(TB_ACCOUNT, cond=cond)
if (not dbObj):
WARNING("user %s not exist" % cond)
return None
user = Account(db, dbObj=dbObj)
user.loadFromObj()
return user
def add(self):
userObj = {
"ID": getUuid(),
"U_Name": self.name,
"U_Password": getEncPass(self.password),
"U_Email": self.email,
"U_PhoneNumber": self.phone,
"U_CreateTime": get_current_time(),
"U_LastSync": get_current_time(),
}
ret = self.db.insert(TB_ACCOUNT, userObj)
if (ret == -1):
WARNING("add user %s error for db operation" % self.name)
return DB_ERR
DEBUG(userObj)
return OCT_SUCCESS
def add(self):
if not self.myId:
self.myId = getUuid()
dbObj = {
"ID": self.myId,
"A_Name": self.name,
"A_Address": self.address,
"A_CreateTime": get_current_time(),
"A_LastSync": get_current_time(),
}
ret = self.db.insert(TB_AGENT, dbObj)
if (ret == -1):
WARNING("add agent %s error for db operation" % self.name)
return DB_ERR
DEBUG(dbObj)
return OCT_SUCCESS
def logout(self):
if self.state == ROBOT_STATE_OFFLINE:
WARNING("robot %s already logout" % self.myId)
return OCT_SUCCESS, None
DEBUG(SystemConf.robots)
if self.state == ROBOT_STATE_WAITINGSCAN or self.state == ROBOT_STATE_ONLINE:
rob = SystemConf.robots.get(self.myId)
if not rob:
ERROR("rob thread %s not running" % self.myId)
else:
DEBUG("rob %s is gone to stop" % self.myId)
rob.running_state = False
del SystemConf.robots[self.myId]
self.state = ROBOT_STATE_OFFLINE
self.updateState()
return OCT_SUCCESS, None
def syncmsg(db, env, arg):
paras = arg["paras"]
agentId = paras["agentId"]
agentObj = getAgentFromCache(agentId)
if not agentObj:
WARNING("agent not exist in cache %s" % agentId)
agent = getAgent(db, agentId)
if not agent:
agent = addAgent(db, agentId, arg["REMOTE_ADDR"],
arg["REMOTE_ADDR"])
agentObj = agent.toObj()
AddAgentToCache(agentObj)
msg = Msg(db)
msg.msg = paras["data"]
msg.type = paras["type"]
msg.agentId = agentId
msg.add()
return (OCT_SUCCESS, None)
def add(self):
self.myId = getUuid()
self.createTime = get_current_time()
self.expireTime = get_current_time() + SESSION_EXPIRE_TIME
obj = {
"ID": self.myId,
"S_UserId": self.userId,
"S_UserName": self.username,
"S_UserType": self.role,
"S_Cookie": transToStr(self.cookie),
"S_CreateTime": self.createTime,
"S_ExpireTime": self.expireTime,
}
ret = self.db.insert(TB_SESSION, obj)
if ret == -1:
WARNING("add session %s error for db operation" % self.myId)
return DB_ERR
return OCT_SUCCESS
def add(self):
self.myId = getUuid()
obj = {
"ID": self.myId,
"P_Name": self.name,
"P_Type": self.type,
"P_TypeName": PRODUCT_TYPES[self.type],
"P_Description": self.desc,
"P_Info": transToStr(self.infoObj),
"P_CreateTime": get_current_time(),
"P_LastSync": get_current_time()
}
ret = self.db.insert(TB_PRODUCT, obj)
if (ret == -1):
WARNING("add product %s error for db operation" % self.name)
return DB_ERR
DEBUG(obj)
return OCT_SUCCESS
def add(self):
self.myId = getUuid()
obj = {
"ID": self.myId,
"QR_Name": self.name,
"QR_Type": self.type,
"QR_Price": self.price,
"QR_Points": self.points,
"QR_Info": transToStr(self.info),
"QR_Paras": transToStr(self.paras),
"QR_CreateTime": get_current_time(),
"QR_Description": self.desc,
}
ret = self.db.insert(TB_QUERYRESULT, obj)
if (ret == -1):
WARNING("add user %s error for db operation" % self.name)
return DB_ERR
DEBUG(obj)
return OCT_SUCCESS
if VERSION != '5' and VERSION != '3':
ERROR("unsupport version rvm[%s]" %VERSION)
sys.exit(1)
with open("/OCT/etc/internal_version", "r") as fd:
content = fd.readline()
if "center" in content or "RVMCenter" in content:
PLATFORM = "center"
elif "allinone" in content:
PLATFORM = "allinone"
else:
PLATFORM = "server"
if PLATFORM == "server":
WARNING("rvm server no need get log info")
sys.exit(1)
if VERSION == '3':
DB_NAME = 'octlog'
else:
DB_NAME = 'dbcenter'
BANK_ADDR = "/OCT/OCTFrame/core/membank/%s"
membody = memcache.get_mem(DB_NAME, BANK_ADDR)
tablelist = membody.get('tablelist', {})
db = dbmysql.mysqldb()
loginfo = get_log(db, DB_NAME)
print(loginfo)
