teleprinter/main.py at master · AdrienCos/teleprinter · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
from telegram.ext import Updater, CommandHandler, MessageHandler, Filters
import os

tokenpath = "./token"
token = open(tokenpath, "r").read()

valid_types = [
    "application/pdf",
    "image/png",
    "image/jpg",
    "image/bmp",
    "text/plain"
]

jobs = {}

def get_lp_command(args):
    # Double or single sided
    duplex = "-o sides=two-sided-long-edge"
    if "simple" in args or "single" in args:
            duplex = "-o sides=one-sided"
    # Number of copies
    copies = "-n 1"
    for arg in args:
        if arg.isdigit():
            copies = "-n %s" % arg
    # Generate the lp command
    command = "lp %s %s" % (copies, duplex)
    return command


def request_print(bot, update, args):
    # Get the message in which the file was sent
    file_msg = update.message.reply_to_message
    # Check if we can print the document
    file_type = file_msg.document.mime_type
    if file_type in valid_types:
        # Get the file id, name and link
        file_id = file_msg.document.file_id
        file_name = file_msg.document.file_name
        file_link = bot.get_file(file_id)
        # Download the file
        file_link.download("to_print")
        # Generate the printing command
        command = get_lp_command(args)
        # Print the user
        reply = "Printing file %s" % (file_name)
        msg = update.message.reply_text(reply, quote=True)
        msg_id = msg.message_id
        # Print the file and store the job ID in the jobs dict
        lp_response = os.popen("%s ./to_print" % command).read()
        job_id = lp_response.split()[3]
        jobs[msg_id] = job_id

def cancel_print(bot, update):
    msg_id = update.message.reply_to_message.message_id
    # Check if the message is linked to a print job
    try:
        job_id = jobs[msg_id]
        exit_code = os.system("cancel %s 2>/dev/null" % job_id)
        # Check if the cancel worked
        if exit_code == 0:
            update.message.reply_text("Sucessfully cancelled job")
        else:
            update.message.reply_text("Failed to cancel job")
        # Remove the job from memory
        del jobs[msg_id]
    except:
        update.message.reply_text("No active job is linked to this message")

def recognize_file(bot, update):
    # Notify the user that we can print this file
    file_type = update.message.document.mime_type
    if file_type in valid_types:
        reply = "If you want me to print this file, reply to it with the /print command"
        update.message.reply_text(reply, quote=True)

def print_message(bot, update):
    print(update.message)
    sent = update.message.reply_text("Yes hello")
    print(sent)

# Start the bot
updater = Updater(token)

# Debug handler
updater.dispatcher.add_handler(MessageHandler(Filters.text, print_message))
# Handler to notify the user we can print their document
updater.dispatcher.add_handler(MessageHandler(Filters.document, recognize_file))
# Handler to execute a print command
updater.dispatcher.add_handler(CommandHandler('print', request_print, pass_args=True))
# Handler to cancel print jobs
updater.dispatcher.add_handler(CommandHandler('cancel', cancel_print))

updater.start_polling()
updater.idle()