mirror of
https://github.com/TotalFreedomMC/TotalFreedomBot.git
synced 2024-12-22 15:44:57 +00:00
38 lines
1.6 KiB
Python
38 lines
1.6 KiB
Python
import discord
|
|
import time
|
|
|
|
from telnetlib import Telnet
|
|
from discord.ext import commands
|
|
|
|
class telnet(commands.Cog):
|
|
def __init__(self, bot):
|
|
self.bot = bot
|
|
self.ip = [ip]
|
|
self.telnet_port = [port]
|
|
self.telnet_password = [password]
|
|
|
|
self.bot.telnet_session = Telnet(self.ip, self.telnet_port)
|
|
self.read_timeout = 3
|
|
if b'Username:' in self.bot.telnet_session.read_until(b'Username:', timeout=self.read_timeout):
|
|
self.bot.telnet_session.write(bytes('TotalFreedom', 'ascii') + b"\r\n")
|
|
|
|
if b'Password:' in self.bot.telnet_session.read_until(b'Password:', timeout=self.read_timeout):
|
|
time.sleep(2)
|
|
self.bot.telnet_session.write(bytes(self.telnet_password, 'ascii') + b"\r\n")
|
|
else:
|
|
raise PyboardError('Failed to establish a telnet connection with the board')
|
|
|
|
def connect():
|
|
self.bot.telnet_session = Telnet(self.ip, self.telnet_port)
|
|
self.read_timeout = 3
|
|
if b'Username:' in self.bot.telnet_session.read_until(b'Username:', timeout=self.read_timeout):
|
|
self.bot.telnet_session.write(bytes('TotalFreedom', 'ascii') + b"\r\n")
|
|
|
|
if b'Password:' in self.bot.telnet_session.read_until(b'Password:', timeout=self.read_timeout):
|
|
time.sleep(2)
|
|
self.bot.telnet_session.write(bytes(self.telnet_password, 'ascii') + b"\r\n")
|
|
else:
|
|
raise PyboardError('Failed to establish a telnet connection with the board')
|
|
|
|
def setup(bot):
|
|
bot.add_cog(telnet(bot))
|