diff options
| -rw-r--r-- | main.py | 9 |
1 files changed, 4 insertions, 5 deletions
@@ -50,11 +50,9 @@ async def get_user(id): return await get_user(id) async def search_teammates(who, query): - now = datetime.datetime.utcnow() cal = parsedatetime.Calendar() - isrange = " for " in query or " until " in query - if isrange: + if " for " in query or " until " in query: delimit = query.rfind("for") if delimit == -1: delimit = query.rfind("until") @@ -65,7 +63,8 @@ async def search_teammates(who, query): startstr = query endstr = "for 30 minutes" - start, end, calnow = [cal.parseDT(d)[0] for d in (startstr, endstr, "now")] + start = cal.parseDT(startstr)[0] + end = cal.parseDT(endstr, sourceTime=start)[0] db.playwithme.insert_one({ "player": who.id, @@ -76,7 +75,7 @@ async def search_teammates(who, query): print(who.name + " requested a game " + ", ".join([str(d) for d in (start, end)])) players = [] - for teammate in db.playwithme.find({"start": {"$lt": start}, "end": {"$gt": start}}): + for teammate in db.playwithme.find({"end": {"$gt": start, "$lt": end}}): mention = (await get_user(teammate["player"]))["mention"] if not mention in players: players += [mention] |