async def get_all_events(ctx: Context_T) -> List[Event]:
try:
return await Event.query.where(Event.context_id == ctx_id_by_user(ctx)
).gino.all()
except Exception as e:
logger.exception(e)
return []
async def bind_event_with_qq_group(event: Event, qq_group_number: int) -> bool:
try:
await event.update(qq_group_number=qq_group_number).apply()
return True
except Exception as e:
logger.exception(e)
return False
async def get_signup_count(event: Event) -> Optional[int]:
try:
return await db.select([db.func.count(Signup.id)]
).where(Signup.event_id == event.id
).gino.scalar()
except Exception as e:
logger.exception(e)
return None
async def get_signup(ctx: Context_T, event: Event) -> Optional[Signup]:
try:
return await Signup.query.where(
(Signup.context_id == ctx_id_by_user(ctx))
& (Signup.event_id == event.id)).gino.first()
except Exception as e:
logger.exception(e)
return None
async def end_event(event: Event) -> bool:
try:
await event.update(end_time=dt.beijing_now().timestamp(),
qq_group_number=None).apply()
return True
except Exception as e:
logger.exception(e)
return False
async def create_signup(ctx: Context_T, event: Event,
field_values: List[str]) -> Optional[Signup]:
try:
return await Signup.create(context_id=ctx_id_by_user(ctx),
event_id=event.id,
field_values=field_values,
qq_number=ctx.get('user_id'))
except Exception as e:
logger.exception(e)
return None
async def start_event(ctx: Context_T,
title: str,
fields: List[Dict[str, Any]],
max_signups: int = 0) -> Optional[Event]:
try:
return await Event.create(
context_id=ctx_id_by_user(ctx),
title=title,
code=random_string(8, string.ascii_uppercase + string.digits),
fields=fields,
start_time=dt.beijing_now().timestamp(),
max_signups=max_signups,
)
except Exception as e:
logger.exception(e)
return None
async def _(session: RequestSession):
group_id = session.ctx['group_id']
user_id = session.ctx['user_id']
try:
count = await db.select([
db.func.count(Signup.id)
]).where((Signup.event_id == Event.id) & (Signup.qq_number == user_id)
& (Event.qq_group_number == group_id)).gino.scalar()
if count > 0:
# the user has signed up
await session.approve()
else:
event_count = await db.select([
db.func.count(Event.id)
]).where(Event.qq_group_number == group_id).gino.scalar()
if event_count > 0:
await session.reject('请先报名才能加群哦')
except Exception as e:
logger.exception(e)
async def get_all_signups(event: Event) -> List[Signup]:
try:
return await Signup.query.where(Signup.event_id == event.id).gino.all()
except Exception as e:
logger.exception(e)
return []
async def get_event(code: str) -> Optional[Event]:
try:
return await Event.query.where(Event.code == code).gino.first()
except Exception as e:
logger.exception(e)
return None