python-telegram-bot/examples/passportbot.py
2024-06-30 18:22:12 +02:00

126 lines
4.4 KiB
Python

#!/usr/bin/env python
# pylint: disable=unused-argument
# This program is dedicated to the public domain under the CC0 license.
"""
Simple Bot to print/download all incoming passport data
See https://telegram.org/blog/passport for info about what telegram passport is.
See https://github.com/python-telegram-bot/python-telegram-bot/wiki/Telegram-Passport
for how to use Telegram Passport properly with python-telegram-bot.
Note:
To use Telegram Passport, you must install PTB via
`pip install "python-telegram-bot[passport]"`
"""
import logging
from pathlib import Path
from telegram import Update
from telegram.ext import Application, ContextTypes, MessageHandler, filters
# Enable logging
logging.basicConfig(
format="%(asctime)s - %(name)s - %(levelname)s - %(message)s", level=logging.INFO
)
# set higher logging level for httpx to avoid all GET and POST requests being logged
logging.getLogger("httpx").setLevel(logging.WARNING)
logger = logging.getLogger(__name__)
async def msg(update: Update, context: ContextTypes.DEFAULT_TYPE) -> None:
"""Downloads and prints the received passport data."""
# Retrieve passport data
passport_data = update.message.passport_data
# If our nonce doesn't match what we think, this Update did not originate from us
# Ideally you would randomize the nonce on the server
if passport_data.decrypted_credentials.nonce != "thisisatest":
return
# Print the decrypted credential data
# For all elements
# Print their decrypted data
# Files will be downloaded to current directory
for data in passport_data.decrypted_data: # This is where the data gets decrypted
if data.type == "phone_number":
logger.info("Phone: %s", data.phone_number)
elif data.type == "email":
logger.info("Email: %s", data.email)
if data.type in (
"personal_details",
"passport",
"driver_license",
"identity_card",
"internal_passport",
"address",
):
logger.info(data.type, data.data)
if data.type in (
"utility_bill",
"bank_statement",
"rental_agreement",
"passport_registration",
"temporary_registration",
):
logger.info(data.type, len(data.files), "files")
for file in data.files:
actual_file = await file.get_file()
logger.info(actual_file)
await actual_file.download_to_drive()
if (
data.type in ("passport", "driver_license", "identity_card", "internal_passport")
and data.front_side
):
front_file = await data.front_side.get_file()
logger.info(data.type, front_file)
await front_file.download_to_drive()
if data.type in ("driver_license" and "identity_card") and data.reverse_side:
reverse_file = await data.reverse_side.get_file()
logger.info(data.type, reverse_file)
await reverse_file.download_to_drive()
if (
data.type in ("passport", "driver_license", "identity_card", "internal_passport")
and data.selfie
):
selfie_file = await data.selfie.get_file()
logger.info(data.type, selfie_file)
await selfie_file.download_to_drive()
if data.translation and data.type in (
"passport",
"driver_license",
"identity_card",
"internal_passport",
"utility_bill",
"bank_statement",
"rental_agreement",
"passport_registration",
"temporary_registration",
):
logger.info(data.type, len(data.translation), "translation")
for file in data.translation:
actual_file = await file.get_file()
logger.info(actual_file)
await actual_file.download_to_drive()
def main() -> None:
"""Start the bot."""
# Create the Application and pass it your token and private key
private_key = Path("private.key")
application = (
Application.builder().token("TOKEN").private_key(private_key.read_bytes()).build()
)
# On messages that include passport data call msg
application.add_handler(MessageHandler(filters.PASSPORT_DATA, msg))
# Run the bot until the user presses Ctrl-C
application.run_polling(allowed_updates=Update.ALL_TYPES)
if __name__ == "__main__":
main()