payment.py

import sys
import time
from pprint import pprint
import telepot
from telepot.loop import MessageLoop
from telepot.namedtuple import LabeledPrice, ShippingOption
from telepot.delegate import (
    per_invoice_payload, pave_event_space, create_open, per_message, call)

"""
Default payment bot script
"""

class OrderProcessor(telepot.helper.InvoiceHandler):
    def __init__(self, *args, **kwargs):
        super(OrderProcessor, self).__init__(*args, **kwargs)
    
    def on_shipping_query(self, msg):
        query_id, from_id, invoice_payload = telepot.glance(msg, flavor="shipping_query")
        
        print("Shipping query:")
        pprint(msg)
        
        bot.answerShippingQuery(
            query_id, True,
            shipping_options=[
                ShippingOption(id='fedex', title='FedEx', prices=[
                    LabeledPrice(label='Local', amount=345),
                    LabeledPrice(label='International', amount=2345)]),
                ShippingOption(id='dhl', title="DHL", prices=[
                    LabeledPrice(label='Local', amount=345),
                    LabeledPrice(label='International', amount=2345)])
            ])
            
    def on_pre_checkout_query(self, msg):
        query_id, from_id, invoice_payload = telepot.glance(msg, flavor='pre_checkout_query')
        
        print('Pre-Checkout query:')
        pprint(msg)
        
        bot.answerPreCheckoutQuery(query_id, True)
        
    def on_chat_message(self, msg):
        content_type, chat_type, chat_id = telepot.glance(msg)
        
        if content_type == 'successful_payment':
            print('Successful payment has been received.')
        else:
            print('Chat message:')
        pprint(msg)
        
def send_invoice(seed_tuple):
    msg = seed_tuple[1]
    
    content_type, chat_type, chat_id = telepot.glance(msg)
    
    if content_type == 'text':
        sent = bot.sendInvoice(
            chat_id, "Nick's Hand Cream", "Manly hands with a feminine touch",
            payload='something',
            provider_token=PAYMENT_PROVIDER_TOKEN,
            start_parameter='abcdefg',
            currency='USD', prices=[
                LabeledPrice(label='One Case', amount=456),
                LabeledPrice(label='Multiple', amount=667)],
            need_shipping_address=True, is_flexible=True)
            
    print("Invoice sent:")
    pprint(sent)
    
TOKEN = ""
PAYMENT_PROVIDER_TOKEN = ""
                
bot = telepot.DelegatorBot(TOKEN, [
    (per_message(flavors=['chat']), call(send_invoice)),
    pave_event_space()(
        per_invoice_payload(), create_open, OrderProcessor, timeout=30
    )
])

MessageLoop(bot).run_as_thread()

while 1:
    time.sleep(10)