import requests, base64

from django.conf import settings


class Atmos:
    def __init__(self, terminal_id = None):
        self.consumer_key = settings.CONSUMER_KEY
        self.consumer_secret = settings.CONSUMER_SECRET
        self.terminal_id = terminal_id
        self.store_id = settings.STORE_ID
    
    def login(self):
        credentials = f"{self.consumer_key}:{self.consumer_secret}"
        encoded_credentials = base64.b64encode(credentials.encode()).decode()

        headers = {
            "Authorization": f"Basic {encoded_credentials}",
            "Content-Type": "application/x-www-form-urlencoded",
        }

        data = {
            "grant_type": "client_credentials"
        }
        url = 'https://partner.atmos.uz/token'
        res = requests.post(url, headers=headers, data=data)
        return res.json()['access_token']
    
    def create_transaction(self, amount, account):
        access_token = self.login()

        url = 'https://partner.atmos.uz/merchant/pay/create'
        headers = {
            'Authorization': f'Bearer {access_token}',
        }
        data = {
            'amount': int(amount),
            'account': str(account),
            'store_id': f'{self.store_id}'
        }

        res = requests.post(url, headers=headers, json=data)
        return res.json()
    
    def generate_url(self, transaction_id, redirect_url):
        url = f'http://test-checkout.pays.uz/invoice/get?storeId={self.store_id}&transactionId={transaction_id}&redirectLink={redirect_url}'
        return url