TotalFreedomBot/telnet.py

39 lines
1.6 KiB
Python
Raw Normal View History

2020-10-30 13:56:43 +00:00
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))