def _setting(update, context):
entry_cmd = update.effective_message.text
if " " in entry_cmd:
entry_cmd = entry_cmd.replace(" ", "")
if "/set" == entry_cmd.strip():
update.effective_message.reply_text(
_msg.set_multi_fav_guide(_lang), parse_mode=ParseMode.MARKDOWN_V2
)
return _stage.SET_FAV_MULTI
if "purge" == entry_cmd[4:]:
fav_count = load.db_counters.find_one({"_id": "fav_count_list"})
if fav_count is not None and fav_count['fav_sum'] != 0:
fav_sum = fav_count['fav_sum']
query = { "fav_type": {"$regex": "^fav"} }
del_query = load.fav_col.delete_many(query)
fav_sum -= int(del_query.deleted_count)
load.db_counters.update(
{"_id": "fav_count_list"},
{"fav_sum": fav_sum},
upsert=True,
)
update.effective_message.reply_text(
_text[_lang]["purge_fav"]
)
return ConversationHandler.END
else:
update.effective_message.reply_text(
_text[_lang]["show_fav_list_null"]
)
return ConversationHandler.END
if "/setlist" == entry_cmd:
global showitem
global showlist
fav_count = load.db_counters.find_one({"_id": "fav_count_list"})
fav_list = load.fav_col.find({"fav_type": "fav"})
if fav_count is not None:
if fav_count["fav_sum"] == 0:
update.effective_message.reply_text(_text[_lang]["show_fav_list_null"])
return ConversationHandler.END
elif fav_count["fav_sum"] != 0:
for item in fav_list:
showitem = (
"type : "
+ item["G_type"]
+ " | name : "
+ item["G_name"]
+ "\nid : "
+ item["G_id"]
+ "\n"
+ "--------------------\n"
)
showlist.append(showitem)
showlist = "".join(showlist)
update.effective_message.reply_text(
_text[_lang]["show_fav_list"] + "\n\n" + showlist
)
showlist = []
return ConversationHandler.END
else:
update.effective_message.reply_text(_text[_lang]["show_fav_list_null"])
return ConversationHandler.END
### set single DST ID ###
elif "quick" or "fav" in entry_cmd:
### single quick (drive or folder)
if len(entry_cmd.splitlines()) == 1:
each = entry_cmd[4:]
if "quick" == each[:5]:
if "quick+" == each[:6]:
global pick_quick
pick_quick = _func.get_name_from_id(
update, each[6:], list_name=pick_quick
)
insert_fav_quick = _func.insert_to_db_quick(pick_quick, update)
if insert_fav_quick == "is_cover":
update.effective_message.reply_text(
_text[_lang]["is_cover_quick_msg"],
parse_mode=ParseMode.MARKDOWN_V2,
reply_markup=_KB.is_cover_keyboard(),
)
return _stage.IS_COVER_QUICK
elif "quick-" == each[:6]:
_func.delete_in_db_quick
update.effective_message.reply_text(
_text[_lang]["delete_quick_success"]
)
### set fav folder(fav folder could be a drive or folder of GDrive)
elif "fav" == each[:3]:
fav_count = load.db_counters.find_one({"_id": "fav_count_list"})
fav_sum = 0
if fav_count != None:
fav_sum = fav_count["fav_sum"]
if "+" == each[3]:
global pick_fav
pick_fav = _func.get_name_from_id(
update, each[4:], list_name=pick_fav
)
for item in pick_fav:
item["fav_type"] = "fav"
try:
load.fav_col.insert_one(item)
except:
update.effective_message.reply_text(
_text[_lang]["is_set_err"],
)
else:
fav_sum += 1
load.db_counters.update(
{"_id": "fav_count_list"},
{"fav_sum": fav_sum},
upsert=True,
)
update.effective_message.reply_text(_text[_lang]["set_fav_success"])
pick_fav = []
if "-" == each[3]:
global unpick_fav
unpick_fav.append(each[4:])
for item in unpick_fav:
delete_request = {"G_id": item}
_func.delete_in_db(delete_request)
fav_count = load.fav_col.find({"fav_type": "fav"})
fav_sum = len(list(fav_count))
load.db_counters.update(
{"_id": "fav_count_list"}, {"fav_sum": fav_sum}, upsert=True
)
update.effective_message.reply_text(
_text[_lang]["delete_fav_success"]
)
unpick_fav = []
### single rule
elif "rule" == entry_cmd[4:8]:
update.effective_message.reply_text(
_msg.set_single_fav_guide(_lang), parse_mode=ParseMode.MARKDOWN_V2
)
return ConversationHandler.END
else:
update.effective_message.reply_text(
_text[_lang]["get_single_fav_error"],
parse_mode=ParseMode.MARKDOWN_V2,
)
return ConversationHandler.END
return ConversationHandler.END
else:
update.effective_message.reply_text(
_text[_lang]["get_multi_in_single"], parse_mode=ParseMode.MARKDOWN_V2
)
return ConversationHandler.END
else:
update.effective_message.reply_text(
_msg.set_help(_lang), parse_mode=ParseMode.MARKDOWN_V2
)
return ConversationHandler.END
def _multi_settings_recieved(update, context):
_tmp_quick_counter = 0
fav_msg = update.effective_message.text
fav_msg = fav_msg.replace(" ", "").splitlines()
global pick_quick
for each in fav_msg:
print(each)
### modify quick DST
if "quick+" == each[:6]:
_tmp_quick_counter += 1
if _tmp_quick_counter == 1:
global pick_quick
pick_quick += _func.get_name_from_id(
update, each[6:], list_name=pick_quick
)
insert_fav_quick = _func.insert_to_db_quick(pick_quick, update)
if insert_fav_quick == "error":
update.effective_message.reply_text(
_text[_lang]["is_cover_quick_msg"],
parse_mode=ParseMode.MARKDOWN_V2,
reply_markup=_KB.is_cover_keyboard(),
)
return _stage.IS_COVER_QUICK
elif _tmp_quick_counter < 1:
pass
elif _tmp_quick_counter > 1:
print("error!")
update.effective_message.reply_text(
_text[_lang]["get_quick_count_invaild"]
)
elif "quick-" == each[:6]:
_func.delete_in_db_quick
update.effective_message.reply_text(_text[_lang]["delete_quick_success"])
### set fav folder(fav folder could be a drive or folder of GDrive)
elif "fav" == each[:3]:
fav_count = load.db_counters.find_one({"_id": "fav_count_list"})
fav_sum = 0
if fav_count != None:
fav_sum = fav_count["fav_sum"]
if "+" == each[3]:
global pick_fav
pick_fav += _func.get_name_from_id(update, each[4:], list_name=pick_fav)
for item in pick_fav:
item["fav_type"] = "fav"
try:
load.fav_col.insert_one(item)
except:
update.effective_message.reply_text(_text[_lang]["is_set_err"],)
else:
fav_sum += 1
load.db_counters.update(
{"_id": "fav_count_list"}, {"fav_sum": fav_sum}, upsert=True
)
update.effective_message.reply_text(_text[_lang]["set_fav_success"])
pick_fav = []
if "-" == each[3]:
global unpick_fav
unpick_fav.append(each[4:])
for item in unpick_fav:
delete_request = {"G_id": item}
_func.delete_in_db(delete_request)
fav_count = load.fav_col.find({"fav_type": "fav"})
fav_sum = len(list(fav_count))
load.db_counters.update(
{"_id": "fav_count_list"}, {"fav_sum": fav_sum}, upsert=True
)
update.effective_message.reply_text(_text[_lang]["delete_fav_success"])
unpick_fav = []
else:
if "/cancel" == update.effective_message.text:
return _func.cancel(update, context)
else:
update.effective_message.reply_text(_text[_lang]["get_multi_fav_error"])
return ConversationHandler.END