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)