mirror of
https://github.com/hastagAB/Awesome-Python-Scripts.git
synced 2024-11-23 20:11:07 +00:00
129 lines
4.1 KiB
Python
Executable File
129 lines
4.1 KiB
Python
Executable File
import json
|
|
import requests
|
|
import time
|
|
import urllib
|
|
from dbhelper import DBHelper
|
|
import datetime
|
|
import os
|
|
|
|
db = DBHelper()
|
|
|
|
TOKEN = os.environ['TOKEN']
|
|
URL = "https://api.telegram.org/bot{}/".format(TOKEN)
|
|
|
|
|
|
def get_url(url):
|
|
response = requests.get(url)
|
|
content = response.content.decode("utf8")
|
|
return content
|
|
|
|
|
|
def get_json_from_url(url):
|
|
content = get_url(url)
|
|
js = json.loads(content)
|
|
return js
|
|
|
|
|
|
def get_updates(offset=None):
|
|
url = URL + "getUpdates"
|
|
if offset:
|
|
url += "?offset={}".format(offset)
|
|
js = get_json_from_url(url)
|
|
return js
|
|
|
|
|
|
def get_last_update_id(updates):
|
|
update_ids = []
|
|
for update in updates["result"]:
|
|
update_ids.append(int(update["update_id"]))
|
|
return max(update_ids)
|
|
|
|
|
|
def handle_updates(updates):
|
|
text = ""
|
|
for update in updates["result"]:
|
|
try:
|
|
text = update["message"]["text"]
|
|
except Exception as e:
|
|
pass
|
|
chat = update["message"]["chat"]["id"]
|
|
if text.startswith("/delete"):
|
|
t = text.split(" ")
|
|
if len(t) < 2:
|
|
send_message("Delete Item Properly.", chat)
|
|
else:
|
|
items = db.get_items(chat)
|
|
if items == []:
|
|
send_message("Nothing to Delete.", chat)
|
|
elif t[1] in items[0]:
|
|
db.delete_item(t[1], chat)
|
|
send_message("Item " + t[1] + " Deleted Successfully.", chat)
|
|
else:
|
|
send_message("Item Not Found.", chat)
|
|
elif text == "/start":
|
|
send_message("Send /help to show help.", chat)
|
|
elif text == "/help":
|
|
message = "Send /add item_name due_date(dd-mm-yyyy) to add an item.\nSend /show to display items in to-do list.\nSend /due to check remaining days of each item.\nSend /delele to delete item."
|
|
send_message(message, chat)
|
|
elif text == "/show":
|
|
items = db.get_items(chat)
|
|
if items == []:
|
|
send_message("Nothing to show.", chat)
|
|
for i in range(len(items)):
|
|
message = items[i][0] + " " + items[i][1]
|
|
send_message(message, chat)
|
|
elif text == "/due":
|
|
items = db.get_items(chat)
|
|
due = [0 for _ in range(0, len(items))]
|
|
if items == []:
|
|
send_message("Nothing to show.", chat)
|
|
today = datetime.datetime.today().strftime("%d-%m-%Y")
|
|
for i in range(0, len(items)):
|
|
due[i] = (datetime.datetime.strptime(items[i][1], "%d-%m-%Y") -
|
|
datetime.datetime.strptime(today, "%d-%m-%Y")).days
|
|
message = items[i][0] + " " + str(due[i]) + " days Remaining"
|
|
send_message(message, chat)
|
|
elif text.startswith("/add"):
|
|
t = text.split(" ")
|
|
if len(t) < 3:
|
|
send_message("Insert Item Properly.\nSend /add item_name due_date to add an item.", chat)
|
|
elif len(t[1]) < 1 or len(t[2]) < 1:
|
|
send_message("Insert Item Properly.\nSend /add item_name due_date to add an item.", chat)
|
|
else:
|
|
db.add_item(t[1], t[2], chat)
|
|
send_message("Item " + t[1] + " Inserted Successfully.", chat)
|
|
else:
|
|
send_message("Not a valid Input.", chat)
|
|
|
|
|
|
def get_last_chat_id_and_text(updates):
|
|
num_updates = len(updates["result"])
|
|
last_update = num_updates - 1
|
|
text = updates["result"][last_update]["message"]["text"]
|
|
chat_id = updates["result"][last_update]["message"]["chat"]["id"]
|
|
return text, chat_id
|
|
|
|
|
|
def send_message(text, chat_id, reply_markup=None):
|
|
text = urllib.parse.quote_plus(text)
|
|
url = URL + "sendMessage?text={}&chat_id={}&parse_mode=Markdown".format(text, chat_id)
|
|
if reply_markup:
|
|
url += "&reply_markup={}".format(reply_markup)
|
|
get_url(url)
|
|
|
|
|
|
def main():
|
|
db.setup()
|
|
last_update_id = None
|
|
while True:
|
|
updates = get_updates(last_update_id)
|
|
if len(updates["result"]) > 0:
|
|
last_update_id = get_last_update_id(updates) + 1
|
|
handle_updates(updates)
|
|
time.sleep(0.5)
|
|
|
|
|
|
if __name__ == '__main__':
|
|
main()
|
|
|