pc-monitor-bot/lock.py

31 lines
684 B
Python

#!/usr/bin/env python3
import telegram
import sys
import datetime
try:
import config
except ImportError:
print("Missing Config. Exiting.")
exit()
bot = telegram.Bot(token=config.api_key)
if len(sys.argv) != 2:
print("Invalid number of parameters")
exit(1)
arg = sys.argv[1]
if arg == "lock":
today = datetime.datetime.today()
text = today.strftime(config.lock_message)
bot.send_message(chat_id=config.master_id, text=text)
elif arg == "unlock":
today = datetime.datetime.today()
text = today.strftime(config.unlock_message)
bot.send_message(chat_id=config.master_id, text=text)
else:
print("Wrong parameter. Exiting.")
exit(1)