mirror of
https://github.com/solero/houdini.git
synced 2024-11-14 14:48:21 +00:00
87 lines
3.7 KiB
Python
87 lines
3.7 KiB
Python
from collections import OrderedDict
|
|
from aiocache import cached
|
|
from types import FunctionType
|
|
import asyncio
|
|
import enum
|
|
|
|
|
|
class ConflictResolution(enum.Enum):
|
|
Silent = 0
|
|
Append = 1
|
|
Exception = 2
|
|
|
|
|
|
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)
|
|
|
|
@cached(namespace='houdini')
|
|
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)
|
|
return attribute_method
|
|
|
|
@classmethod
|
|
def data_attribute_by_name(cls, attribute_name):
|
|
async def attribute_method(p):
|
|
return getattr(p.data, attribute_name)
|
|
return attribute_method
|
|
|
|
@classmethod
|
|
def setup_default_builder(cls, string_builder):
|
|
string_builder.update({
|
|
'ID': PenguinStringCompiler.data_attribute_by_name('ID'),
|
|
'Nickname': PenguinStringCompiler.data_attribute_by_name('Nickname'),
|
|
'Approval': PenguinStringCompiler.attribute_by_name('approval'),
|
|
'Color': PenguinStringCompiler.data_attribute_by_name('Color'),
|
|
'Head': PenguinStringCompiler.data_attribute_by_name('Head'),
|
|
'Face': PenguinStringCompiler.data_attribute_by_name('Face'),
|
|
'Neck': PenguinStringCompiler.data_attribute_by_name('Neck'),
|
|
'Body': PenguinStringCompiler.data_attribute_by_name('Body'),
|
|
'Hand': PenguinStringCompiler.data_attribute_by_name('Name'),
|
|
'Photo': PenguinStringCompiler.data_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'),
|
|
'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('Name'),
|
|
'Photo': PenguinStringCompiler.attribute_by_name('Photo')
|
|
})
|