def remDevUser(user_id):
reUser = SESSION.query(DevUser).filter(DevUser.user_id == user_id).first()
if reUser:
SESSION.delete(reUser)
SESSION.commit()
else:
raise Exception("No user to remove.")
def getWelcomeSettings(chat_id):
try:
return SESSION.query(Welcome).filter(Welcome.chat_id == chat_id).one()
except:
return None
finally:
SESSION.close()
def getAllAutoReply():
try:
return SESSION.query(AutoReply).all()
except:
return None
finally:
SESSION.close()
def getDevsUsers():
try:
return SESSION.query(DevUser).all()
except:
return None
finally:
SESSION.close()
def getAutoReply(text):
try:
return SESSION.query(AutoReply).filter(AutoReply.text == text).one()
except:
return None
finally:
SESSION.close()
def getMutedUsers(chat_id):
try:
return SESSION.query(MutePersonal).filter(
MutePersonal.chat_id == chat_id).all()
except:
return None
finally:
SESSION.close()
def addDevUser(user_id):
addUser = SESSION.query(DevUser).filter(DevUser.user_id == user_id).first()
if addUser:
raise Exception("The user already added as Dev.")
else:
addUser = DevUser(None, user_id)
SESSION.add(addUser)
SESSION.commit()
def getDevUser(from_id):
try:
devUser = SESSION.query(DevUser).filter(
DevUser.user_id == from_id).one()
SESSION.close()
return devUser
except:
return None
def getPropSettings(chatId):
try:
return SESSION.query(PropSetting).filter(
PropSetting.chatId == chatId).get()
except:
return None
finally:
SESSION.close()
def remAutoReplySetting(text):
try:
remrep = SESSION.query(AutoReply).filter(AutoReply.text == text).one()
if remrep:
SESSION.delete(remrep)
SESSION.commit()
return True
except:
return False
def remMuteUser(chat_id, user_id):
reUser = (SESSION.query(MutePersonal).filter(
MutePersonal.chat_id == chat_id).filter(
MutePersonal.user_id == user_id).first())
if reUser:
SESSION.delete(reUser)
SESSION.commit()
else:
raise Exception("No user to remove.")
def getMutedUser(chat_id, from_id):
try:
muteUser = (SESSION.query(MutePersonal).filter(
MutePersonal.chat_id == chat_id).filter(
MutePersonal.user_id == from_id).one())
SESSION.close()
return muteUser
except:
return None
def getPropSetting(chatId, propName):
try:
return (SESSION.query(PropSetting).filter(
PropSetting.chatId == chatId
and PropSetting.propName == propName).one())
except:
return None
finally:
SESSION.close()
def addMuteUser(chat_id, first_name, user_id):
addUser = (SESSION.query(MutePersonal).filter(
MutePersonal.chat_id == chat_id).filter(
MutePersonal.user_id == user_id).first())
if addUser:
raise Exception("The user already muted.")
else:
addUser = MutePersonal(None, chat_id, first_name, user_id)
SESSION.add(addUser)
SESSION.commit()
def addWelcomeSetting(chat_id, msg_type, msg_content="", file_id=""):
addwel = SESSION.query(Welcome).get(chat_id)
if addwel:
addwel.msg_type = msg_type
addwel.msg_content = msg_content
try:
os.remove(addwel.file_id)
except Exception as e:
print("addWelcomeSetting : %s" % (e))
addwel.file_id = file_id
else:
addwel = Welcome(chat_id, msg_type, msg_content, file_id)
SESSION.add(addwel)
SESSION.commit()
def addAutoReply(text, msg_type, msg_content="", file_id=""):
try:
addRep = SESSION.query(AutoReply).filter(AutoReply.text == text).one()
except Exception as e:
addRep = None
print(str("error : togglepropsetting : %s" % (e)))
if addRep:
addRep.msg_type = msg_type
addRep.msg_content = msg_content
try:
os.remove(addRep.file_id)
except Exception as e:
print("addAutoReplySetting : %s" % (e))
addRep.file_id = file_id
else:
addRep = AutoReply(text, msg_type, msg_content, file_id)
SESSION.add(addRep)
SESSION.commit()
def togglePropSettings(chatId, propName, active=True):
try:
try:
addProp = (SESSION.query(PropSetting).filter(
PropSetting.chatId == chatId
and PropSetting.propName == propName).one())
except Exception as e:
addProp = None
print(str("error : togglepropsetting : %s" % (e)))
if addProp:
addProp.propName = propName
addProp.active = active
else:
addProp = PropSetting(chatId, propName, active)
SESSION.add(addProp)
SESSION.commit()
return True
except Exception as e:
print(str("error : togglepropsetting : %s" % (e)))
def remWelcomeSetting(chat_id):
remwel = SESSION.query(Welcome).get(chat_id)
if remwel:
SESSION.delete(remwel)
SESSION.commit()
