mirror of
https://github.com/solero/houdini.git
synced 2024-11-12 13:48:20 +00:00
110 lines
4.5 KiB
Python
110 lines
4.5 KiB
Python
from collections import OrderedDict
|
|
from types import FunctionType
|
|
from abc import abstractmethod
|
|
|
|
import asyncio
|
|
import logging
|
|
import importlib
|
|
import pkgutil
|
|
|
|
|
|
def get_package_modules(package):
|
|
package_modules = []
|
|
for importer, module_name, is_package in pkgutil.iter_modules(package.__path__):
|
|
full_module_name = f'{package.__name__}.{module_name}'
|
|
subpackage_object = importlib.import_module(full_module_name, package=package.__path__)
|
|
if is_package:
|
|
sub_package_modules = get_package_modules(subpackage_object)
|
|
|
|
package_modules = package_modules + sub_package_modules
|
|
package_modules.append(subpackage_object)
|
|
return package_modules
|
|
|
|
|
|
class _AbstractManager(dict):
|
|
def __init__(self, server):
|
|
self.server = server
|
|
self.logger = logging.getLogger('houdini')
|
|
|
|
super().__init__()
|
|
|
|
@abstractmethod
|
|
async def setup(self, module):
|
|
"""Setup manager class"""
|
|
|
|
@abstractmethod
|
|
async def load(self, module):
|
|
"""Loads entries from module"""
|
|
|
|
|
|
class PenguinStringCompiler(OrderedDict):
|
|
|
|
def __init__(self, *args, **kwargs):
|
|
super().__init__(*args, **kwargs)
|
|
|
|
def __setitem__(self, key, compiler_method):
|
|
assert type(compiler_method) == FunctionType
|
|
super().__setitem__(key, compiler_method)
|
|
|
|
async def compile(self, p):
|
|
compiler_method_results = []
|
|
|
|
for compiler_method in self.values():
|
|
if asyncio.iscoroutinefunction(compiler_method):
|
|
compiler_method_result = await compiler_method(p)
|
|
else:
|
|
compiler_method_result = compiler_method(p)
|
|
compiler_method_results.append(str(compiler_method_result))
|
|
|
|
compiler_result = '|'.join(compiler_method_results)
|
|
return compiler_result
|
|
|
|
@classmethod
|
|
def attribute_by_name(cls, attribute_name):
|
|
async def attribute_method(p):
|
|
return getattr(p, attribute_name) or 0
|
|
return attribute_method
|
|
|
|
@classmethod
|
|
def setup_default_builder(cls, string_builder):
|
|
string_builder.update({
|
|
'ID': PenguinStringCompiler.attribute_by_name('id'),
|
|
'Nickname': PenguinStringCompiler.attribute_by_name('nickname'),
|
|
'Approval': PenguinStringCompiler.attribute_by_name('approval'),
|
|
'Color': PenguinStringCompiler.attribute_by_name('color'),
|
|
'Head': PenguinStringCompiler.attribute_by_name('head'),
|
|
'Face': PenguinStringCompiler.attribute_by_name('face'),
|
|
'Neck': PenguinStringCompiler.attribute_by_name('neck'),
|
|
'Body': PenguinStringCompiler.attribute_by_name('body'),
|
|
'Hand': PenguinStringCompiler.attribute_by_name('hand'),
|
|
'Feet': PenguinStringCompiler.attribute_by_name('feet'),
|
|
'Flag': PenguinStringCompiler.attribute_by_name('flag'),
|
|
'Photo': PenguinStringCompiler.attribute_by_name('photo'),
|
|
'X': PenguinStringCompiler.attribute_by_name('x'),
|
|
'Y': PenguinStringCompiler.attribute_by_name('y'),
|
|
'Frame': PenguinStringCompiler.attribute_by_name('frame'),
|
|
'Member': PenguinStringCompiler.attribute_by_name('member'),
|
|
'MemberDays': PenguinStringCompiler.attribute_by_name('membership_days_total'),
|
|
'Avatar': PenguinStringCompiler.attribute_by_name('avatar'),
|
|
'PenguinState': PenguinStringCompiler.attribute_by_name('penguin_state'),
|
|
'PartyState': PenguinStringCompiler.attribute_by_name('party_state'),
|
|
'PuffleState': PenguinStringCompiler.attribute_by_name('puffle_state')
|
|
})
|
|
|
|
@classmethod
|
|
def setup_anonymous_default_builder(cls, string_builder):
|
|
string_builder.update({
|
|
'ID': PenguinStringCompiler.attribute_by_name('id'),
|
|
'Nickname': PenguinStringCompiler.attribute_by_name('nickname'),
|
|
'Approval': PenguinStringCompiler.attribute_by_name('approval'),
|
|
'Color': PenguinStringCompiler.attribute_by_name('color'),
|
|
'Head': PenguinStringCompiler.attribute_by_name('head'),
|
|
'Face': PenguinStringCompiler.attribute_by_name('face'),
|
|
'Neck': PenguinStringCompiler.attribute_by_name('neck'),
|
|
'Body': PenguinStringCompiler.attribute_by_name('body'),
|
|
'Hand': PenguinStringCompiler.attribute_by_name('hand'),
|
|
'Feet': PenguinStringCompiler.attribute_by_name('feet'),
|
|
'Flag': PenguinStringCompiler.attribute_by_name('flag'),
|
|
'Photo': PenguinStringCompiler.attribute_by_name('photo')
|
|
})
|