SantaSpeen/IP-CALC-Telegram-Bot
1
0
Fork 0
mirror of https://github.com/Marsh232/Telegram-IP-CALC.git synced 2025-07-03 00:06:15 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add /start /calc /help

Browse Source 
Refactor calc_subnet()
This commit is contained in:
Maxim Khomutov 2022-09-12 14:49:38 +03:00
parent c6767c5bfa
commit 854c7a7a73
1 changed files with 63 additions and 20 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

83
src/main.py
View File

@ -6,33 +6,45 @@
import ipaddress
import logging
from aiogram import Bot
from aiogram import Bot, Dispatcher, types
from aiogram.types import ChatType, ParseMode
from aiogram.utils import executor
from src.config import Config
config = Config("config.json")
log = logging.getLogger("Bot")
bot = Bot(token=config.token)
dp = Dispatcher(bot)
def main(ip):
list_ip = ip.split('/') # Разделяет вводимый ip на часть с маской, и без
print('Ваш ip адрес:', ip) # Выводит например "Ваш ip адрес: 192.168.10.128/24"
def calc_subnet(_ip):
splt = _ip.split('/') # Разделяет вводимый ip на часть с маской, и без
if len(splt) == 1:
splt.append("24")
ip, mask = splt
addr = ip + "/" + mask
net = ipaddress.ip_network(addr, strict=False) # В функцию кладётся сетевая часть ip, без хостовой части
dict_out = {
"addr": addr,
"mask": f"{net.netmask} - {mask}", # Выводит маску
"net": f"{net}", # Выводит сеть
"broadcast": f'{net.broadcast_address}', # Выводит broadcast
"max": f"{net[1]}",
"min": f"{net[-2]}",
"hosts": f"{len(list(net.hosts()))}",
"num": None
}
net = ipaddress.ip_network(ip, strict=False) # В функцию кладётся сетевая часть ip, без хостовой части
print('Маска:', net.netmask, '=', list_ip[1], '\n') # Выводит маску
print('Network:', net) # Выводит сеть
print('Broadcast:', net.broadcast_address) # Выводит broad
print('HostMin:', net[1])
print('HostMax:', net[-2])
print('Hosts:', len(list(net.hosts()))) # Выводит кол-во хостовых ip
count = 0
for n_ip in net.hosts():
count += 1
if str(n_ip) == list_ip[0]:
print('№ в сети:', count) # Выводит какой ip по счёту в сети
break
if str(n_ip) == ip:
dict_out["num"] = count # Выводит какой ip по счёту в сети
return dict_out
def subnets(ip, prefix):
@ -50,13 +62,44 @@ def subnets(ip, prefix):
print('HostMin:', subnet2[1])
print('HostMax:', subnet2[-2])
print('Hosts:', len(list((subnet2.hosts()))))
print()
@dp.message_handler(commands=["start"], chat_type=ChatType.PRIVATE)
async def start(msg: types.Message):
log.info(f"New message from {msg.from_user.id}(@{msg.from_user.username}) in {msg.chat.id}: '{msg.text}'")
await msg.reply(f"Привет, дорогой мой {msg.from_user.username}")
@dp.message_handler(commands=["help"], chat_type=ChatType.PRIVATE)
async def start(msg: types.Message):
log.info(f"New message from {msg.from_user.id}(@{msg.from_user.username}) in {msg.chat.id}: '{msg.text}'")
await msg.reply("Ты беспомощный")
@dp.message_handler(commands=["calc"], chat_type=ChatType.PRIVATE)
async def calc(msg: types.Message):
log.info(f"New message from {msg.from_user.id}(@{msg.from_user.username}) in {msg.chat.id}: '{msg.text}'")
text = msg.text
splt = text.split(" ")
if len(splt) > 1:
ip = splt[1]
c = calc_subnet(ip)
await msg.reply(f"Данные ввода: `{c['addr']}`\n"
f"Маска: `{c['mask']}`\n"
f"Сеть: `{c['net']}`\n"
f"Broadcast: `{c['broadcast']}`\n"
f"Макс адресов: `{c['max']}`\n"
f"Мин адресов: `{c['min']}`\n"
f"Всего адресов: `{c['hosts']}`\n"
f"Номер в сети: `{c['num']}`",
parse_mode=ParseMode.MARKDOWN)
else:
await msg.reply("Пример выполнения команды: `/calc 192.168.0.1/24`", parse_mode=ParseMode.MARKDOWN)
if __name__ == '__main__':
addr = input('Введите ip: ') # Пользователь вводит ip
main(addr)
executor.start_polling(dp)
exit(0)
# Тут должна быть кнопка типа "Подсети"
new_prefix = input('\nВведите префикс: ')
subnets(addr, new_prefix)
# new_prefix = input('\nВведите префикс: ')
# subnets(addr, new_prefix)