mirror of
https://github.com/solero/houdini.git
synced 2025-10-24 00:08:08 +00:00
Redemption join handlers
This commit is contained in:
Ben
@@ -1,6 +1,101 @@
|
||||
from houdini.data import db
|
||||
|
||||
|
||||
class RedemptionCode(db.Model):
|
||||
__tablename__ = 'redemption_code'
|
||||
|
||||
id = db.Column(db.Integer, primary_key=True,
|
||||
server_default=db.text("nextval('\"redemption_code_id_seq\"'::regclass)"))
|
||||
code = db.Column(db.String(16), nullable=False, unique=True)
|
||||
type = db.Column(db.String(8), nullable=False, server_default=db.text("'BLANKET'::character varying"))
|
||||
coins = db.Column(db.Integer, nullable=False, server_default=db.text("0"))
|
||||
expires = db.Column(db.DateTime)
|
||||
uses = db.Column(db.Integer)
|
||||
|
||||
def __init__(self, **kwargs):
|
||||
super().__init__(**kwargs)
|
||||
self._cards = set()
|
||||
self._flooring = set()
|
||||
self._furniture = set()
|
||||
self._igloos = set()
|
||||
self._items = set()
|
||||
self._locations = set()
|
||||
self._puffles = set()
|
||||
self._puffle_items = set()
|
||||
|
||||
@property
|
||||
def cards(self):
|
||||
return self._cards
|
||||
|
||||
@cards.setter
|
||||
def cards(self, child):
|
||||
if isinstance(child, RedemptionAwardCard):
|
||||
self._cards.add(child)
|
||||
|
||||
@property
|
||||
def flooring(self):
|
||||
return self._flooring
|
||||
|
||||
@flooring.setter
|
||||
def flooring(self, child):
|
||||
if isinstance(child, RedemptionAwardFlooring):
|
||||
self._flooring.add(child)
|
||||
|
||||
@property
|
||||
def furniture(self):
|
||||
return self._furniture
|
||||
|
||||
@furniture.setter
|
||||
def furniture(self, child):
|
||||
if isinstance(child, RedemptionAwardFurniture):
|
||||
self._furniture.add(child)
|
||||
|
||||
@property
|
||||
def igloos(self):
|
||||
return self._igloos
|
||||
|
||||
@igloos.setter
|
||||
def igloos(self, child):
|
||||
if isinstance(child, RedemptionAwardIgloo):
|
||||
self._igloos.add(child)
|
||||
|
||||
@property
|
||||
def items(self):
|
||||
return self._items
|
||||
|
||||
@items.setter
|
||||
def items(self, child):
|
||||
if isinstance(child, RedemptionAwardItem):
|
||||
self._items.add(child)
|
||||
|
||||
@property
|
||||
def locations(self):
|
||||
return self._locations
|
||||
|
||||
@locations.setter
|
||||
def locations(self, child):
|
||||
if isinstance(child, RedemptionAwardLocation):
|
||||
self._locations.add(child)
|
||||
|
||||
@property
|
||||
def puffles(self):
|
||||
return self._puffles
|
||||
|
||||
@puffles.setter
|
||||
def puffles(self, child):
|
||||
if isinstance(child, RedemptionAwardPuffle):
|
||||
self._puffles.add(child)
|
||||
|
||||
@property
|
||||
def puffle_items(self):
|
||||
return self._puffle_items
|
||||
|
||||
@puffle_items.setter
|
||||
def puffle_items(self, child):
|
||||
if isinstance(child, RedemptionAwardPuffleItem):
|
||||
self._puffle_items.add(child)
|
||||
|
||||
|
||||
class RedemptionBook(db.Model):
|
||||
__tablename__ = 'redemption_book'
|
||||
|
||||
@@ -19,17 +114,6 @@ class RedemptionBookWord(db.Model):
|
||||
answer = db.Column(db.String(20), nullable=False)
|
||||
|
||||
|
||||
class RedemptionCode(db.Model):
|
||||
__tablename__ = 'redemption_code'
|
||||
|
||||
id = db.Column(db.Integer, primary_key=True,
|
||||
server_default=db.text("nextval('\"redemption_code_id_seq\"'::regclass)"))
|
||||
code = db.Column(db.String(16), nullable=False, unique=True)
|
||||
type = db.Column(db.String(8), nullable=False, server_default=db.text("'BLANKET'::character varying"))
|
||||
coins = db.Column(db.Integer, nullable=False, server_default=db.text("0"))
|
||||
expires = db.Column(db.DateTime)
|
||||
|
||||
|
||||
class RedemptionAwardCard(db.Model):
|
||||
__tablename__ = 'redemption_award_card'
|
||||
code_id = db.Column(db.ForeignKey('redemption_code.id', ondelete='CASCADE', onupdate='CASCADE'), primary_key=True,
|
||||
@@ -56,8 +140,8 @@ class RedemptionAwardFurniture(db.Model):
|
||||
|
||||
class RedemptionAwardIgloo(db.Model):
|
||||
__tablename__ = 'redemption_award_igloo'
|
||||
code_id = db.Column('CodeID', db.ForeignKey('redemption_code.id', ondelete='CASCADE', onupdate='CASCADE'),
|
||||
primary_key=True,nullable=False)
|
||||
code_id = db.Column(db.ForeignKey('redemption_code.id', ondelete='CASCADE', onupdate='CASCADE'),
|
||||
primary_key=True, nullable=False)
|
||||
igloo_id = db.Column(db.ForeignKey('igloo.id', ondelete='CASCADE', onupdate='CASCADE'), primary_key=True,
|
||||
nullable=False)
|
||||
|
||||
@@ -94,11 +178,19 @@ class RedemptionAwardPuffleItem(db.Model):
|
||||
primary_key=True, nullable=False)
|
||||
|
||||
|
||||
class PenguinRedemption(db.Model):
|
||||
__tablename__ = 'penguin_redemption'
|
||||
class PenguinRedemptionCode(db.Model):
|
||||
__tablename__ = 'penguin_redemption_code'
|
||||
|
||||
penguin_id = db.Column(db.ForeignKey('penguin.id', ondelete='CASCADE', onupdate='CASCADE'), primary_key=True,
|
||||
nullable=False)
|
||||
code_id = db.Column(db.ForeignKey('redemption_code.id', ondelete='CASCADE', onupdate='CASCADE'), primary_key=True,
|
||||
nullable=False, index=True)
|
||||
book_id = db.Column(db.ForeignKey('redemption_book.id', ondelete='CASCADE', onupdate='CASCADE'), index=True)
|
||||
|
||||
|
||||
class PenguinRedemptionBook(db.Model):
|
||||
__tablename__ = 'penguin_redemption_book'
|
||||
|
||||
penguin_id = db.Column(db.ForeignKey('penguin.id', ondelete='CASCADE', onupdate='CASCADE'), primary_key=True,
|
||||
nullable=False)
|
||||
book_id = db.Column(db.ForeignKey('redemption_book.id', ondelete='CASCADE', onupdate='CASCADE'), primary_key=True,
|
||||
nullable=False, index=True)
|
||||
|
||||
@@ -0,0 +1,68 @@
|
||||
from houdini import handlers
|
||||
from houdini.handlers import XTPacket
|
||||
from houdini.constants import ClientType
|
||||
|
||||
from houdini.data import db
|
||||
from houdini.data.redemption import RedemptionCode, RedemptionAwardCard, RedemptionAwardFlooring, \
|
||||
RedemptionAwardFurniture, RedemptionAwardIgloo, RedemptionAwardItem, RedemptionAwardLocation,\
|
||||
RedemptionAwardPuffle, RedemptionAwardPuffleItem, PenguinRedemptionBook, PenguinRedemptionCode
|
||||
|
||||
from datetime import datetime
|
||||
|
||||
|
||||
@handlers.handler(XTPacket('rjs', ext='red'), pre_login=True, client=ClientType.Vanilla)
|
||||
@handlers.allow_once
|
||||
async def handle_join_redemption_server_vanilla(p, credentials: str, confirmation_hash: str, lang: str):
|
||||
pid, _, username, login_key, rdnk, approved, rejected = credentials.split('|')
|
||||
|
||||
if login_key != p.login_key:
|
||||
return await p.close()
|
||||
|
||||
tr = p.server.redis.multi_exec()
|
||||
tr.setex(f'{username}.lkey', p.server.config.auth_ttl, login_key)
|
||||
tr.setex(f'{username}.ckey', p.server.config.auth_ttl, confirmation_hash)
|
||||
await tr.execute()
|
||||
|
||||
redeemed_books = await PenguinRedemptionBook.query.where(PenguinRedemptionBook.penguin_id == p.id).gino.all()
|
||||
await p.send_xt('rjs', ','.join(str(redeemed_book.book_id) for redeemed_book in redeemed_books), 'houdini',
|
||||
int(p.is_member))
|
||||
|
||||
|
||||
@handlers.handler(XTPacket('rsc', ext='red'), pre_login=True)
|
||||
@handlers.depends_on_packet(XTPacket('rjs', ext='red'))
|
||||
async def handle_code(p, redemption_code: str):
|
||||
query = RedemptionCode.load(cards=RedemptionAwardCard,
|
||||
flooring=RedemptionAwardFlooring,
|
||||
furniture=RedemptionAwardFurniture,
|
||||
igloos=RedemptionAwardIgloo,
|
||||
items=RedemptionAwardItem,
|
||||
locations=RedemptionAwardLocation,
|
||||
puffles=RedemptionAwardPuffle,
|
||||
puffle_items=RedemptionAwardPuffleItem)\
|
||||
.query.where(RedemptionCode.code == redemption_code)
|
||||
code = await query.gino.first()
|
||||
|
||||
if code is None:
|
||||
return await p.send_error(720)
|
||||
|
||||
if code.uses is not None:
|
||||
redeemed_count = await db.select([db.func.count(PenguinRedemptionCode.code_id)]).where(
|
||||
PenguinRedemptionCode.code_id == code.id).gino.scalar()
|
||||
|
||||
if redeemed_count >= code.uses:
|
||||
return await p.send_error(721)
|
||||
|
||||
penguin_redeemed = await PenguinRedemptionCode.query.where((PenguinRedemptionCode.code_id == code.id) &
|
||||
(PenguinRedemptionCode.penguin_id == p.id)).gino.scalar()
|
||||
if penguin_redeemed:
|
||||
return await p.send_error(721)
|
||||
|
||||
if code.expires is not None and code.expires < datetime.now():
|
||||
return await p.send_error(726)
|
||||
|
||||
if code.type == 'CATALOG':
|
||||
owned_ids = ','.join((item.item_id for item in code.items if item.item_id in p.inventory))
|
||||
return await p.send_xt('rsc', 'treasurebook', 3, owned_ids, 0)
|
||||
|
||||
await p.send_xt('rsc', code.type, '', code.coins)
|
||||
|
||||
|
||||
Reference in New Issue
Block a user