parcel.py 650 B

12345678910111213141516171819
  1. from nio.events.room_events import RoomMessageText
  2. from nio.rooms import MatrixRoom
  3. async def command_parcel(room: MatrixRoom, event: RoomMessageText, bot):
  4. prompt = event.body.split()[2:]
  5. if prompt:
  6. bot.logger.log("Looking up parcels...")
  7. for parcel in prompt:
  8. status, tokens_used = bot.parcel_api.lookup_parcel(parcel, user=room.room_id)
  9. await bot.send_message(room, status, True)
  10. bot.log_api_usage(event, room, f"{bot.parcel_api.api_code}-{bot.parcel_api.parcel_api}", tokens_used)
  11. return
  12. await bot.send_message(room, "You need to provide tracking numbers.", True)