mirror of
https://github.com/catspace-dev/unicheckbot.git
synced 2024-11-21 21:46:32 +03:00
this should probably make relative imports work
This commit is contained in:
parent
eda75f552c
commit
723172387b
|
@ -8,6 +8,6 @@ RUN apk update \
|
|||
&& apk add gcc python3-dev musl-dev libffi-dev openssl-dev make cargo
|
||||
|
||||
COPY . .
|
||||
WORKDIR tgbot/tgbot
|
||||
WORKDIR tgbot
|
||||
RUN pip install --upgrade pip; pip install poetry; poetry config virtualenvs.create false; poetry install
|
||||
CMD poetry shell; python bot.py
|
||||
|
|
|
@ -5,8 +5,8 @@ from tortoise import Tortoise
|
|||
from tortoise.exceptions import DBConnectionError
|
||||
from loguru import logger
|
||||
from asyncio import sleep
|
||||
import config
|
||||
import handlers
|
||||
import tgbot.config as config
|
||||
import tgbot.handlers as handlers
|
||||
|
||||
storage = MemoryStorage()
|
||||
telegram_bot = Bot(token=config.TELEGRAM_BOT_TOKEN)
|
1
apps/tgbot/test/__init__.py
Normal file
1
apps/tgbot/test/__init__.py
Normal file
|
@ -0,0 +1 @@
|
|||
|
|
@ -1,7 +1,7 @@
|
|||
from unittest import TestCase
|
||||
|
||||
from ..handlers.default.tcp import TCPCheckerHandler
|
||||
from ..handlers.base import process_args_for_host_port,\
|
||||
from tgbot.handlers.default.tcp import TCPCheckerHandler
|
||||
from tgbot.handlers.base import process_args_for_host_port,\
|
||||
NotEnoughArgs, InvalidPort
|
||||
|
||||
|
|
@ -1,7 +1,7 @@
|
|||
from aiogram.types import Message
|
||||
from typing import Tuple, Any, List
|
||||
|
||||
from tgbot.nodes import nodes as all_nodes
|
||||
from ..nodes import nodes as all_nodes
|
||||
from httpx import Response
|
||||
from aiogram.bot import Bot
|
||||
from datetime import datetime
|
||||
|
@ -9,7 +9,7 @@ from core.coretypes import APINodeInfo
|
|||
from .helpers import send_api_requests
|
||||
from .errors import NotEnoughArgs, InvalidPort, LocalhostForbidden
|
||||
from .validators import BaseValidator, LocalhostValidator
|
||||
from tgbot.middlewares.throttling import rate_limit
|
||||
from ..middlewares.throttling import rate_limit
|
||||
from loguru import logger
|
||||
from uuid import uuid4
|
||||
from time import time
|
||||
|
|
|
@ -3,7 +3,7 @@ from httpx import Response
|
|||
from core.coretypes import ErrorPayload, ICMPCheckerResponse, ResponseStatus
|
||||
from ..base import CheckerBaseHandler, NotEnoughArgs, LocalhostForbidden
|
||||
from ..metrics import push_status_metric
|
||||
from tgbot.middlewares.throttling import rate_limit
|
||||
from ...middlewares.throttling import rate_limit
|
||||
|
||||
icmp_help_message = """
|
||||
❓ Производит проверку хоста по протоколу ICMP.
|
||||
|
|
|
@ -2,8 +2,8 @@ from aiogram.types import Message
|
|||
from typing import Union
|
||||
import ipaddress
|
||||
|
||||
from tgbot.handlers.base import SimpleCommandHandler, NotEnoughArgs
|
||||
from tgbot.middlewares.throttling import rate_limit
|
||||
from ..base import SimpleCommandHandler, NotEnoughArgs
|
||||
from ...middlewares.throttling import rate_limit
|
||||
|
||||
ipcalc_help_message = """
|
||||
❓ Калькулятор IP подсетей.
|
||||
|
|
|
@ -1,8 +1,8 @@
|
|||
from core.coretypes import ResponseStatus, ErrorPayload, MinecraftResponse
|
||||
from httpx import Response
|
||||
|
||||
from tgbot.handlers.base import CheckerTargetPortHandler, process_args_for_host_port
|
||||
from tgbot.handlers.metrics import push_status_metric
|
||||
from ..base import CheckerTargetPortHandler, process_args_for_host_port
|
||||
from ..metrics import push_status_metric
|
||||
|
||||
minecraft_help_message = """
|
||||
❓ Получает статистику о Minecraft сервере
|
||||
|
|
|
@ -1,7 +1,7 @@
|
|||
from aiogram.types import Message
|
||||
from tgbot.models.user import User
|
||||
from tgbot.middlewares.throttling import rate_limit
|
||||
from tgbot.middlewares.userdata import userdata_required
|
||||
from ...models.user import User
|
||||
from ...middlewares.throttling import rate_limit
|
||||
from ...middlewares.userdata import userdata_required
|
||||
|
||||
start_message = """
|
||||
|
||||
|
|
|
@ -3,9 +3,9 @@ from aiogram.types import Message
|
|||
from core.coretypes import ResponseStatus, ErrorPayload, PortResponse
|
||||
from httpx import Response
|
||||
|
||||
from tgbot.handlers.base import CheckerTargetPortHandler, NotEnoughArgs, InvalidPort, parse_host_port
|
||||
from tgbot.handlers.metrics import push_status_metric
|
||||
from tgbot.middlewares.throttling import rate_limit
|
||||
from ..base import CheckerTargetPortHandler, NotEnoughArgs, InvalidPort, parse_host_port
|
||||
from ..metrics import push_status_metric
|
||||
from ...middlewares.throttling import rate_limit
|
||||
|
||||
tcp_help_message = """
|
||||
❓ Производит проверку TCP порта, открыт ли он или нет
|
||||
|
|
|
@ -6,10 +6,10 @@ from dataclasses import dataclass
|
|||
from whois_vu.api import WhoisSource
|
||||
from whois_vu.errors import IncorrectZone, QueryNotMatchRegexp
|
||||
|
||||
from tgbot.handlers.whois_zones import ZONES
|
||||
from tgbot.handlers.base import SimpleCommandHandler
|
||||
from tgbot.handlers.errors import NotEnoughArgs, LocalhostForbidden
|
||||
from tgbot.middlewares.throttling import rate_limit
|
||||
from ..whois_zones import ZONES
|
||||
from ..base import SimpleCommandHandler
|
||||
from ..errors import NotEnoughArgs, LocalhostForbidden
|
||||
from ...middlewares.throttling import rate_limit
|
||||
|
||||
whois_help_message = """
|
||||
❓ Вернёт информацию о домене.
|
||||
|
|
|
@ -5,8 +5,8 @@ from ipaddress import ip_address
|
|||
from contextlib import suppress
|
||||
from loguru import logger
|
||||
from aiogram.bot import Bot
|
||||
from tgbot.handlers.metrics import push_api_request_status
|
||||
from tgbot.config import NOTIFICATION_BOT_TOKEN, NOTIFICATION_USERS
|
||||
from .metrics import push_api_request_status
|
||||
from ..config import NOTIFICATION_BOT_TOKEN, NOTIFICATION_USERS
|
||||
from traceback import format_exc
|
||||
import asyncio
|
||||
|
||||
|
|
|
@ -1,6 +1,6 @@
|
|||
from aioinflux import InfluxDBClient
|
||||
from typing import Dict
|
||||
from tgbot.config import INFLUX_DB, INFLUX_HOST, INFLUX_PORT, INFLUX_PASSWORD, INFLUX_USERNAME
|
||||
from ..config import INFLUX_DB, INFLUX_HOST, INFLUX_PORT, INFLUX_PASSWORD, INFLUX_USERNAME
|
||||
|
||||
|
||||
async def push_metric(measurement, tags: Dict, fields: Dict):
|
||||
|
|
|
@ -1,4 +1,4 @@
|
|||
from tgbot.middlewares.write_command_metric import WriteCommandMetric
|
||||
from tgbot.middlewares.logging import LoggingMiddleware
|
||||
from tgbot.middlewares.throttling import ThrottlingMiddleware
|
||||
from tgbot.middlewares.userdata import UserMiddleware
|
||||
from .write_command_metric import WriteCommandMetric
|
||||
from .logging import LoggingMiddleware
|
||||
from .throttling import ThrottlingMiddleware
|
||||
from .userdata import UserMiddleware
|
||||
|
|
|
@ -2,7 +2,7 @@ from aiogram.dispatcher.handler import current_handler
|
|||
from aiogram.dispatcher.middlewares import BaseMiddleware
|
||||
from aiogram.types import Message, CallbackQuery
|
||||
|
||||
from tgbot.models import User
|
||||
from ..models import User
|
||||
|
||||
|
||||
def userdata_required(func):
|
||||
|
|
|
@ -1,7 +1,7 @@
|
|||
from aiogram.dispatcher.middlewares import BaseMiddleware
|
||||
from aiogram.types import Message
|
||||
from tgbot.handlers.metrics import push_metric
|
||||
from tgbot.models import User, UserCheckRequests
|
||||
from ..handlers.metrics import push_metric
|
||||
from ..models import User, UserCheckRequests
|
||||
|
||||
|
||||
class WriteCommandMetric(BaseMiddleware):
|
||||
|
|
|
@ -1 +0,0 @@
|
|||
from .test_port_parsers import *
|
Loading…
Reference in New Issue
Block a user