Add convert commands (pdf to docx), add cloud commands and fix little bugs.

This commit is contained in:
stef-devs 2022-06-23 16:14:08 +02:00
parent b8d811c850
commit c017091f96
2 changed files with 41 additions and 4 deletions

View File

@ -20,6 +20,8 @@ import logging
import deepl import deepl
import spotipy import spotipy
import pyshorteners import pyshorteners
import pdf2docx
from pdf2docx import Converter, parse
from spotipy.oauth2 import SpotifyClientCredentials from spotipy.oauth2 import SpotifyClientCredentials
from youtube_search import YoutubeSearch from youtube_search import YoutubeSearch
from dotenv import load_dotenv from dotenv import load_dotenv
@ -297,7 +299,7 @@ def shutdown_step(message):
id = message.from_user.id id = message.from_user.id
text = message.text text = message.text
id_owner = os.getenv('USER_ID') id_owner = os.getenv('USER_ID')
if id == id_owner: if id == 771375637:
if text == "si" or text == "Si" or text == "y" or text == "Y" or text == "Yes" or text == "yes": if text == "si" or text == "Si" or text == "y" or text == "Y" or text == "Yes" or text == "yes":
bot.send_message(message.chat.id, "Il computer è stato spento con successo!") bot.send_message(message.chat.id, "Il computer è stato spento con successo!")
@ -306,8 +308,6 @@ def shutdown_step(message):
requests.get(url) requests.get(url)
else: else:
bot.send_message(message.chat.id, "⚠️ Hai annullato l'operazione!") bot.send_message(message.chat.id, "⚠️ Hai annullato l'operazione!")
else:
bot.send_message(message.chat.id, "Eh volevi!")
#Command /shortlink #Command /shortlink
@ -340,4 +340,39 @@ def uptime_step(message):
else: else:
bot.send_message(message.chat.id, "Il sito non è online!") bot.send_message(message.chat.id, "Il sito non è online!")
#Command /convert
@bot.message_handler(commands=["convert"])
def addfile(message):
sent_msg = bot.send_message(message.chat.id, "Manda il file pdf che vuoi convertire in docx")
bot.register_next_step_handler(sent_msg, convert)
def convert(message):
file_name = message.document.file_name
file_info = bot.get_file(message.document.file_id)
downloaded_file = bot.download_file(file_info.file_path)
with open(file_name, 'wb') as new_file:
new_file.write(downloaded_file)
docx_file = file_name + '.docx'
pdf_file = file_name
cv = Converter(pdf_file)
cv.convert(docx_file)
cv.close()
html_doc = open(docx_file, 'rb')
sent_msg = bot.send_message(message.chat.id, "Il file è stato convertito con successo!")
bot.send_document(message.chat.id, html_doc)
os.remove(pdf_file)
#Command /cloud
@bot.message_handler(commands=["cloud"])
def cloud(message):
sent_msg = bot.send_message(message.chat.id, "Invia il file che vuoi caricare")
bot.register_next_step_handler(sent_msg, cloud_step)
def cloud_step(message):
file_name = message.document.file_name
file_info = bot.get_file(message.document.file_id)
downloaded_file = bot.download_file(file_info.file_path)
with open(file_name, 'wb') as new_file:
new_file.write(downloaded_file)
bot.polling() bot.polling()

View File

@ -32,4 +32,6 @@ yt_dlp == 2022.5.18
pytelegrambotapi == 4.5.1 pytelegrambotapi == 4.5.1
pyshorteners == 1.0.1 pyshorteners == 1.0.1
pdf2docx == 0.5.4