imports refactor

This commit is contained in:
Klemek
2021-04-22 14:58:08 +02:00
parent 4ce3d6023e
commit 3721f1aef2
4 changed files with 44 additions and 56 deletions
+2 -2
View File
@@ -1,6 +1,6 @@
from .emoji import Emoji, get_emoji_dict from .emoji import Emoji, get_emoji_dict
from .frequency import Frequency
from .composition import Composition from .composition import Composition
from .presence import Presence
from .counter import Counter from .counter import Counter
from .frequency import Frequency
from .history import History from .history import History
from .presence import Presence
+29 -44
View File
@@ -7,22 +7,7 @@ if sys.version_info < (3, 7):
sys.exit(1) sys.exit(1)
from utils import emojis, gdpr, command_cache from utils import emojis, gdpr, command_cache
from scanners import ( import scanners
EmojisScanner,
FullScanner,
FrequencyScanner,
CompositionScanner,
PresenceScanner,
MentionsScanner,
MentionedScanner,
MessagesScanner,
ChannelsScanner,
ReactionsScanner,
FirstScanner,
RandomScanner,
LastScanner,
WordsScanner,
)
from logs import GuildLogs from logs import GuildLogs
logging.basicConfig( logging.basicConfig(
@@ -67,9 +52,9 @@ bot.register_command(
) )
bot.register_command( bot.register_command(
"words", "words",
lambda *args: WordsScanner().compute(*args), lambda *args: scanners.WordsScanner().compute(*args),
"words: (BETA) rank words by their usage", "words: (BETA) rank words by their usage",
WordsScanner.help(), scanners.WordsScanner.help(),
) )
bot.register_command( bot.register_command(
"repeat", "repeat",
@@ -85,81 +70,81 @@ bot.register_command(
) )
bot.register_command( bot.register_command(
"last", "last",
lambda *args: LastScanner().compute(*args), lambda *args: scanners.LastScanner().compute(*args),
"last: read last message", "last: read last message",
LastScanner.help(), scanners.LastScanner.help(),
) )
bot.register_command( bot.register_command(
"rand(om)?", "rand(om)?",
lambda *args: RandomScanner().compute(*args), lambda *args: scanners.RandomScanner().compute(*args),
"rand: read a random message", "rand: read a random message",
RandomScanner.help(), scanners.RandomScanner.help(),
) )
bot.register_command( bot.register_command(
"first", "first",
lambda *args: FirstScanner().compute(*args), lambda *args: scanners.FirstScanner().compute(*args),
"first: read first message", "first: read first message",
FirstScanner.help(), scanners.FirstScanner.help(),
) )
bot.register_command( bot.register_command(
"mentioned", "mentioned",
lambda *args: MentionedScanner().compute(*args), lambda *args: scanners.MentionedScanner().compute(*args),
"mentioned: rank specific user mentions by their usage", "mentioned: rank specific user mentions by their usage",
MentionedScanner.help(), scanners.MentionedScanner.help(),
) )
bot.register_command( bot.register_command(
"(mentions?)", "(mentions?)",
lambda *args: MentionsScanner().compute(*args), lambda *args: scanners.MentionsScanner().compute(*args),
"mentions: rank mentions by their usage", "mentions: rank mentions by their usage",
MentionsScanner.help(), scanners.MentionsScanner.help(),
) )
bot.register_command( bot.register_command(
"(emojis?|emotes?)", "(emojis?|emotes?)",
lambda *args: EmojisScanner().compute(*args), lambda *args: scanners.EmojisScanner().compute(*args),
"emojis: rank emojis by their usage", "emojis: rank emojis by their usage",
EmojisScanner.help(), scanners.EmojisScanner.help(),
) )
bot.register_command( bot.register_command(
"(react(ions?)?)", "(react(ions?)?)",
lambda *args: ReactionsScanner().compute(*args), lambda *args: scanners.ReactionsScanner().compute(*args),
"react: rank users by their reactions", "react: rank users by their reactions",
ReactionsScanner.help(), scanners.ReactionsScanner.help(),
) )
bot.register_command( bot.register_command(
"(channels?|chan)", "(channels?|chan)",
lambda *args: ChannelsScanner().compute(*args), lambda *args: scanners.ChannelsScanner().compute(*args),
"chan: rank channels by their messages", "chan: rank channels by their messages",
ChannelsScanner.help(), scanners.ChannelsScanner.help(),
) )
bot.register_command( bot.register_command(
"(messages?|msg)", "(messages?|msg)",
lambda *args: MessagesScanner().compute(*args), lambda *args: scanners.MessagesScanner().compute(*args),
"msg: rank users by their messages", "msg: rank users by their messages",
MessagesScanner.help(), scanners.MessagesScanner.help(),
) )
bot.register_command( bot.register_command(
"pres(ence)?", "pres(ence)?",
lambda *args: PresenceScanner().compute(*args), lambda *args: scanners.PresenceScanner().compute(*args),
"pres: presence analysis", "pres: presence analysis",
PresenceScanner.help(), scanners.PresenceScanner.help(),
) )
bot.register_command( bot.register_command(
"compo(sition)?", "compo(sition)?",
lambda *args: CompositionScanner().compute(*args), lambda *args: scanners.CompositionScanner().compute(*args),
"compo: composition analysis", "compo: composition analysis",
CompositionScanner.help(), scanners.CompositionScanner.help(),
) )
bot.register_command( bot.register_command(
"freq(ency)?", "freq(ency)?",
lambda *args: FrequencyScanner().compute(*args), lambda *args: scanners.FrequencyScanner().compute(*args),
"freq: frequency analysis", "freq: frequency analysis",
FrequencyScanner.help(), scanners.FrequencyScanner.help(),
) )
bot.register_command( bot.register_command(
"(full|scan)", "(full|scan)",
lambda *args: FullScanner().compute(*args), lambda *args: scanners.FullScanner().compute(*args),
"scan: full analysis", "scan: full analysis",
FullScanner.help(), scanners.FullScanner.help(),
) )
bot.start() bot.start()
+10 -9
View File
@@ -1,15 +1,16 @@
from .scanner import Scanner from .scanner import Scanner
from .emojis_scanner import EmojisScanner
from .frequency_scanner import FrequencyScanner
from .composition_scanner import CompositionScanner
from .presence_scanner import PresenceScanner
from .full_scanner import FullScanner
from .mentions_scanner import MentionsScanner
from .mentioned_scanner import MentionedScanner
from .messages_scanner import MessagesScanner
from .channels_scanner import ChannelsScanner from .channels_scanner import ChannelsScanner
from .reactions_scanner import ReactionsScanner from .composition_scanner import CompositionScanner
from .emojis_scanner import EmojisScanner
from .first_scanner import FirstScanner from .first_scanner import FirstScanner
from .frequency_scanner import FrequencyScanner
from .full_scanner import FullScanner
from .last_scanner import LastScanner from .last_scanner import LastScanner
from .mentioned_scanner import MentionedScanner
from .mentions_scanner import MentionsScanner
from .messages_scanner import MessagesScanner
from .presence_scanner import PresenceScanner
from .random_scanner import RandomScanner from .random_scanner import RandomScanner
from .reactions_scanner import ReactionsScanner
from .words_scanner import WordsScanner from .words_scanner import WordsScanner
+3 -1
View File
@@ -5,7 +5,9 @@ import discord
# Custom libs # Custom libs
from .scanner import Scanner from .scanner import Scanner
from . import FrequencyScanner, CompositionScanner, PresenceScanner from .composition_scanner import CompositionScanner
from .frequency_scanner import FrequencyScanner
from .presence_scanner import PresenceScanner
from data_types import Frequency, Composition, Presence from data_types import Frequency, Composition, Presence
from logs import ChannelLogs, MessageLog from logs import ChannelLogs, MessageLog
from utils import generate_help from utils import generate_help