async def add_chat_service(chatid):
check = await get_service(chatid)
if not check:
cli.insert_one({'chatid': chatid})
return True
cli.update_one({
'_id': check['id'],
}, {'&set': {
'chatid': check['chatid']
}})
return False
async def set_weather(city):
to_check = await get_weather()
if to_check:
cli.update_one(
{
'_id': to_check['_id'],
'weather_city': to_check['weather_city']
}, {"$set": {
'weather_city': city
}})
else:
cli.insert_one({'weather_city': city})
async def add_to_blacklist(chat_id, trigger):
to_check = await check_blacklist(chat_id, trigger)
if not to_check:
cli.insert_one({'chat_id': chat_id, 'trigger': trigger})
return True
cli.update_one(
{
'_id': to_check["_id"],
'chat_id': to_check["chat_id"],
'trigger': to_check["trigger"],
}, {"$set": {
'trigger': trigger
}})
return False
async def set_ses(chat_id, ses_id, expires):
check = await is_chat(chat_id)
if not check:
cli.insert_one({
'chat_id': chat_id,
'ses_id': ses_id,
'expires': expires
})
return True
cli.update_one({
'_id': check['_id'],
}, {'$set': {
'ses_id': check['ses_id'],
'expires': check['expires']
}})
return False
async def add_filter(chatid, keyword, msg_id):
to_check = await get_filter(chatid, keyword)
if not to_check:
cli.insert_one({'chat_id': chatid, 'keyword': keyword, 'msg': msg_id})
return True
cli.update_one(
{
'_id': to_check["_id"],
'chat_id': to_check["chat_id"],
'keyword': to_check["keyword"],
}, {"$set": {
'msg': msg_id
}})
return False
async def add_note(chatid, name, text):
to_check = await get_note(chatid, name)
if not to_check:
cli.insert_one({'chat_id': chatid, 'name': name, 'text': text})
return True
cli.update_one(
{
'_id': to_check["_id"],
'chat_id': to_check["chat_id"],
'name': to_check["name"],
}, {"$set": {
'text': text
}})
return False
async def add_welcome_setting(chat_id, should_clean_welcome, previous_welcome,
f_mesg_id):
check = await get_current_welcome_settings(chat_id)
if not check:
cli.insert_one({
'chat_id': chat_id,
'should_clean_welcome': should_clean_welcome,
'previous_welcome': previous_welcome,
'f_mesg_id': f_mesg_id
})
return True
cli.update_one(
{
'_id': check['_id'],
'chat_id': check['chat_id'],
'should_clean_welcome': should_clean_welcome,
'previous_welcome': check['previous_welcome']
}, {'$set': {
'f_mesg_id': f_mesg_id
}})
return False
async def update(query, key, value):
return cli.update_one(
query, {"$set": {"Key": key, "Value": value}})
async def update_file(name, path, newfile):
return cli.update_one(
{"Name": name}, {"$set": {"File": newfile, "Path": path}})
async def update_user(query, newvalue):
return cli.update_one(query, {"$set": newvalue})