RetniNet/index.py

import psutil
import requests
import random
import pyfiglet
import socket
import threading
import datetime
import telebot
import time
import ast
import ffmpeg
import os
import python_weather
import asyncio
import json
import goslate
import urllib.request
import urllib.parse
import logging
import deepl
import spotipy
import pyshorteners
import pdf2docx
from fileinput import filename
from pdf2docx import Converter, parse
from spotipy.oauth2 import SpotifyClientCredentials
from youtube_search import YoutubeSearch
from dotenv import load_dotenv
from yt_dlp import YoutubeDL
from platform import system
from tqdm.auto import tqdm
from importlib.metadata import files
from dataclasses import dataclass
from jmespath import search
from telebot import types, telebot
from pytube import YouTube
from bs4 import BeautifulSoup
from gc import callbacks
import logging
import sys

logging.basicConfig(
    level=logging.INFO,
    format="%(asctime)s [%(levelname)s] %(message)s",
    handlers=[
        logging.FileHandler("debug.log"),
        logging.StreamHandler(sys.stdout)
    ]
)

load_dotenv()
API_TOKEN = os.getenv('BOT_TOKEN')
bot = telebot.TeleBot(API_TOKEN)

#Start bot
logging.info("Il bot si è avviato con successo!")


#Command /start
@bot.message_handler(commands=['start'])
def send_welcome(message):
    chat_id = message.chat.id
    logging.info("Triggered command START.")
    bot.send_photo(chat_id, photo='https://i.imgur.com/6YPJBze.png')
    messageText = "✋ Benvenuto su <b>RetniNet!</b>\n\n<b>RetniNet</b> è un bot privato per <b>automatizzare</b> e <b>semplificare</b> cose che facciamo quotidianamente. \n\n👨‍💻 Creato & sviluppato da @Stef58_Official"
    bot.send_message(chat_id, messageText, parse_mode="HTML")

#Command /music


@bot.message_handler(commands=['music'])
def select_music(pm):
    logging.info("Triggered command MUSIC.")
    sent_msg = bot.send_message(pm.chat.id, "Inserisci il link della canzone:")
    bot.register_next_step_handler(sent_msg, music_step)


def music_step(pm):
    text = pm.text
    #Check if the link is correct (Youtube and Spotify links only)
    if text.startswith('https://www.youtube.com/') or text.startswith('https://youtu.be/'):
        ytdl_opts = {
            'format': 'bestaudio/best',
            'outtmpl': 'music_cache/%(title)s.%(ext)s',
            'postprocessors': [{
                'key': 'FFmpegExtractAudio',
                'preferredcodec': 'mp3',
                'preferredquality': '192',
            }],
        }
        with YoutubeDL(ytdl_opts) as ydl:
            url = pm.text
            info = ydl.extract_info(url, download=False)
            name = info.get('title')
            id = info.get('id')
            ydl.download([id])
            send_message = "🎶 La canzone <b>" + name + \
                    "</b> è stata scaricata con successo!"
            bot.send_message(pm.chat.id, send_message, parse_mode="HTML")
            bot.send_audio(pm.chat.id, audio=open("music_cache/" + name + '.mp3', 'rb'))
            os.remove("music_cache/" + name + '.mp3')
    #If the link for spotify ends with /album, /user, /playlist ecc.. send a error message
    elif pm.text.startswith('https://open.spotify.com/track/'):
        ytdl_opts = {
            'format': 'bestaudio/best',
            'outtmpl': 'music_cache/%(title)s.%(ext)s',
            'postprocessors': [{
                'key': 'FFmpegExtractAudio',
                'preferredcodec': 'mp3',
                'preferredquality': '192',
            }],
        }
        with YoutubeDL(ytdl_opts) as ydl:
            message = pm.text
            sp = spotipy.Spotify(auth_manager=SpotifyClientCredentials(client_id=os.getenv('SPOTIFY_ID'), client_secret=os.getenv('SPOTIFY_SECRET')))
            track = sp.track(message)
            name = track["name"]
            artist = track["artists"][0]["name"]
            results = YoutubeSearch(artist + " - " + name, max_results=1).to_json()
            get_link = json.loads(results)
            link = get_link['videos'][0]['url_suffix']
            print_link = 'https://youtube.com' + link
            url = print_link
            info = ydl.extract_info(url, download=False)
            name = info.get('title')
            id = info.get('id')
            ydl.download([id])
            send_message = "🎶 La canzone <b>" + name + \
                "</b> è stata scaricata con successo!"
            bot.send_message(pm.chat.id, send_message, parse_mode="HTML")
            bot.send_audio(pm.chat.id, audio=open("music_cache/" + name + '.mp3', 'rb'))
            os.remove("music_cache/" + name + '.mp3')
            logging.info("\"" + name + "\" è stato scaricato ed eliminato con successo!")

    elif pm.text.startswith('https://youtube.com/') or pm.text.startswith('youtube.com/'):
        bot.send_message(
            pm.chat.id, "🚫 <b>Errore</b>\n\n<b>Il link</b> inserito non è valido!")
    else:
        try:
            with YoutubeDL(ytdl_opts) as ydl:
                url = pm.text
                info = ydl.extract_info(url, download=False)
                name = info.get('title')
                id = info.get('id')
                ydl.download([id])
                send_message = "🎶 La canzone <b>" + name + \
                    "</b> è stata scaricata con successo!"
                bot.send_message(pm.chat.id, send_message, parse_mode="HTML")
                bot.send_audio(pm.chat.id, audio=open("music_cache/" + name + '.mp3', 'rb'))
                os.remove("music_cache/" + name + '.mp3')
        except:
            chat = pm.chat.id
            send_msg = "🚫 Errore!\n\n<b>Errore:</b> <i>Impossibile scaricare la canzone</i>"
            bot.send_message(chat, send_msg, parse_mode="HTML")

        else:
            chat = pm.chat.id
            bot.send_message(chat, pm.chat.id,"🚫 Devi inserire un <b>link</b> valido!")


#Command /meteo
@bot.message_handler(commands=['meteo'])
def meteo(pm):
    logging.info("Triggered command METEO.")
    sent_msg = bot.send_message(pm.chat.id, "🏙️ Inserisci la città:")
    bot.register_next_step_handler(sent_msg, meteo_step)


def meteo_step(message):
    translator = deepl.Translator(os.getenv('DEEPL_TOKEN'))
    result = translator.translate_text('hi', target_lang='IT')
    translated_text = result.text
    city = message.text
    token_weather = os.environ.get('WEATHER_TOKEN')
    response = requests.get("https://api.openweathermap.org/data/2.5/weather?q="+city+",it&APPID="+token_weather).json()
    weather = response['weather'][0]['main']
    temp = response['main']['temp']
    temp = temp - 273.15
    weather_translate = translator.translate_text(weather, target_lang='IT')
    result_weather = weather_translate.text
    bot.send_message(message.chat.id, "🌡️ La temperatura in " + city + " è di " + str(temp) + "°C")
    bot.send_message(message.chat.id, "🌧️ La condizione è " + result_weather)

#Command /stats


@bot.message_handler(commands=['stats'])
def uptime(message):
    print("Triggered command STATS.")
    cpuUsage = psutil.cpu_percent(interval=1)
    ramTotal = int(psutil.virtual_memory().total/(1024*1024))  # GB
    ramUsage = int(psutil.virtual_memory().used/(1024*1024))  # GB
    ramFree = int(psutil.virtual_memory().free/(1024*1024))  # GB
    ramUsagePercent = psutil.virtual_memory().percent
    msg = '''
CPU & RAM Info

🟩 Utilizzo CPU = {} %
RAM
Totale = {} MB
Usato = {} MB
Libero  = {} MB
In uso = {} %\n'''.format(cpuUsage, ramTotal, ramUsage, ramFree, ramUsagePercent)
    bot.send_message(message.chat.id, msg)

#Command /pastebin


@bot.message_handler(commands=['pastebin'])
def pastebin(message):
    logging.info("Triggered command PASTEBIN.")
    sent_msg = bot.send_message(message.chat.id, "📋 Inserisci il testo:")
    bot.register_next_step_handler(sent_msg, pastebin_step)


def pastebin_step(message):
    chat = message.chat.id
    text = message.text
    site = 'https://pastebin.com/api/api_post.php'
    dev_key = os.environ.get('PASTEBIN_TOKEN')
    code = text
    our_data = urllib.parse.urlencode(
        {"api_dev_key": dev_key, "api_option": "paste", "api_paste_code": code})
    our_data = our_data.encode()
    resp = urllib.request.urlopen(site, our_data)
    resp = resp.read()
    send_msg = "📋 Il tuo <b>codice</b> è stato inviato con successo!\n\n<b>Link:</b> " + \
        str(resp)
    bot.send_message(chat, send_msg, parse_mode="HTML")

#Command /epicgames


@bot.message_handler(commands=['epicgames'])
def epicgames(message):
    text = message.text
    bot.send_message(
        message.chat.id, "🎮 Vuoi vedere il gioco disponibile al momento o quello futuro? (disponibile/futuro)")
    bot.register_next_step_handler(message, epicgames_step)


def epicgames_step(message):
    text = message.text
    if text == 'disponibile' or text == 'Disponibile':
        logging.info("Triggered command EPICGAMES DISPONIBILE.")
        # URL API
        url = "https://api.plenusbot.xyz/epic_games?country=IT"
        response = requests.get(url).json()
        # Title of current games
        current_games = response['currentGames'][0]['title']
        # Image current games
        image_currentgames = response['currentGames'][0]['keyImages'][0]['url']
        # Description current games
        current_games_description = response['currentGames'][0]['description']
        # Token for translate
        translator = deepl.Translator(os.getenv('DEEPL_TOKEN'))
        # Translate description current games
        description_translate = translator.translate_text(
            current_games_description, target_lang='IT')
        result_description = description_translate.text
        send_img = bot.send_photo(message.chat.id, image_currentgames)
        sent_msg = bot.send_message(message.chat.id, "🎮 Il gioco gratis di oggi è " + current_games + "\n\n" + result_description)
    else:
        logging.info("Triggered command EPICGAMES FUTURO.")
        # URL API
        url = "https://api.plenusbot.xyz/epic_games?country=IT"
        response = requests.get(url).json()
        # Title of future games
        future_games1 = response['nextGames'][0]['title']
        # Image future games
        image_futuregames1 = response['nextGames'][0]['keyImages'][0]['url']
        # Description future games
        future_games_description1 = response['nextGames'][0]['description']
        # Token for translate
        translator = deepl.Translator(os.getenv('DEEPL_TOKEN'))
        description_translate1 = translator.translate_text(future_games_description1, target_lang='IT')
        result_description1 = description_translate1.text
        send_img = bot.send_photo(message.chat.id, image_futuregames1)
        sent_msg = bot.send_message(message.chat.id, "🎮 Il gioco futuro è " + future_games1 + "\n\n" + result_description1)
        # Title of future games
        future_games2 = response['nextGames'][1]['title']
        # Image future games
        image_futuregames2 = response['nextGames'][1]['keyImages'][0]['url']
        # Description future games
        future_games_description2 = response['nextGames'][1]['description']
        # Traslate description future games
        description_translate2 = translator.translate_text(future_games_description2, target_lang='IT')
        result_description2 = description_translate2.text
        send_img = bot.send_photo(message.chat.id, image_futuregames2)
        sent_msg = bot.send_message(message.chat.id, "🎮 Il gioco futuro è " + future_games2 + "\n\n" + result_description2)

#Command /shutdown
@bot.message_handler(commands=['shutdown'])
def shutdown(message):
    text = message.text
    sent_msg = bot.send_message(message.chat.id, "Sei sicuro di voler spegnere il pc?")
    bot.register_next_step_handler(sent_msg, shutdown_step)

def shutdown_step(message):
    id = message.from_user.id
    text = message.text
    id_owner = os.getenv('USER_ID')
    if id == 771375637:

        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!")
            logging.info("Triggered Shutdown")
            url = os.getenv("PASSWORD")
            requests.get(url)
        else:
            bot.send_message(message.chat.id, "⚠️ Hai annullato l'operazione!")

#Command /shortlink

@bot.message_handler(commands=['shortlink'])
def shortlink(message):
    text = message.text
    sent_msg = bot.send_message(message.chat.id, "Inserisci il link:")
    bot.register_next_step_handler(sent_msg, shortlink_step)

def shortlink_step(message):
    text = message.text
    type_tiny = pyshorteners.Shortener()
    short_url = type_tiny.tinyurl.short(text)
    bot.send_message(message.chat.id, "Ecco a te lo shortlink: " + short_url)

#Command /uptime
@bot.message_handler(commands=['uptime'])
def uptime(message):
    logging.info("Triggered UPTIME")
    sent_msg = bot.send_message(message.chat.id, "Manda il link del sito che vuoi controllare.")
    bot.register_next_step_handler(sent_msg, uptime_step)

def uptime_step(message):
    text = message.text
    url= text
    status_code = urllib.request.urlopen(url).getcode()
    website_is_up = status_code == 200
    if website_is_up == True:
        bot.send_message(message.chat.id, "Il sito è online!")
    else:
        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)
    os.replace(file_name, 'cloud/'+file_name)

bot.polling()