anafapi/tests/test_anafoauth.py

'''Created 6 Jul 2022 Levi'''
# coding: UTF-8
import platform
import unittest
import os
from typing import Union

from lxml import etree

from anafapi.anafoauth import Anafoauth, Efactoauth
from anafapi import utils

if platform.system() == 'Linux':
    token_file_path = '/home/deeejas/git/anafapi/anafapi/{}'
elif platform.system() == 'Windows':
    token_file_path = os.path.join(os.path.expanduser('~'), 'git/anafapi/anafapi/{}')

class oauthTest(unittest.TestCase):
    anoauth = Anafoauth(
        client_id='16cbef8754bef0bcf015529353427e8a7e3ee71d56879565',
        client_secret='c6aa66d89002f36461f93facd990e6bf5bc706e1df897e8a7e3ee71d56879565',
        redirect_uri='https://www.anaf.ro/oauth2callback',
    )

    def test_is_valid(self):
        self.anoauth.load_token_file(token_file_path.format('efact.json'))
        # print(self.anoauth.to)
        print(self.anoauth.is_valid)

    def test_browser_redirect(self):
        print(self.anoauth._browser_redirect())

    def test_get_token(self):
        self.anoauth.get_token()
        self.anoauth.save_token(token_file_path.format('efact.json'))

    def test_revive_token(self):
        self.anoauth.load_token_file(token_file_path.format('efact.json'))
        self.anoauth.revive_token()
        self.anoauth.save_token(token_file_path.format('efact_refreshed.json'))

class efactTest(unittest.TestCase):
    efactoauth = Efactoauth(
        client_id='16cbef8754bef0bcf015529353427e8a7e3ee71d56879565',
        client_secret='c6aa66d89002f36461f93facd990e6bf5bc706e1df897e8a7e3ee71d56879565',
        redirect_uri='https://www.anaf.ro/oauth2callback',
    )
    efactoauth.load_token_file(token_file_path.format('efact.json'))
    efactoauth.load_session()

    def test_add_invoice(self):
        # self.efactoauth.validate_invoice('C:/Users/DECEL/FACT1/MZK/2023_06/MZK2321602.xml', env='prod')
        self.efactoauth.add_invoice('17259191', 'C:/Users/DECEL/FACT1/MZK/2023_06/MZK2321602.xml', env='prod')

    def test_bulk_add_invoice(self):
        for root, _, files in os.walk('f:/DECEL/EFACT/KINVEST/2024_01/'):
            for file in files:
                if file.endswith('.xml'):
                    inv_file = os.path.join(root, file)
                    self.efactoauth.add_invoice('31434328', inv_file, env='prod')

    def test_transform_to_pdf(self):
        self.efactoauth.transform_to_pdf('c:/Users/DECEL/FACT1/MZK/2023_01_03/MZK2414.xml')

    def test_bulk_transform(self):
        for root, _, files in os.walk('f:/DECEL/EFACT/KINVEST/2024_01/'):
            for file in files:
                if file.endswith('.xml'):
                    inv_file = os.path.join(root, file)
                    self.efactoauth.transform_to_pdf(inv_file, no_validation=True)

    def test_get_all_messages(self, vat_id, days, root_path, env, filter_, extract):
        self.efactoauth.get_all_messages(vat_id, days=days, env=env, root_path=root_path, filter_=filter_, extract=extract)

    def test_pretty_messages(self):
        self.efactoauth.pretty_messages('17259191', env='prod')

    def test_download_invoice(self):
        self.efactoauth.download_invoice('3105899770', path='C:/Users/DECEL/FACT1/MZK/2023_06', env='prod')

    def test_getxml_to_erp(self):
        # self.efactoauth.getxml_from_zip(path='../efact/3016038776.zip', dest_f='')
        # latest = self.efactoauth.get_latest_invoice('efact')
        inv_files = utils.get_not_booked('17259191')
        # inv_files = ['C:/Users/Levi/git/anafapi/tests/17259191/FACTURA PRIMITA/2024_02/06/17317438/4159771982.xml']
        for latest in inv_files:
            if 'booked' not in latest:
                inv = self.efactoauth.parse_inv(invoice=latest)
                print(inv.seller_party.name, inv.invoice_id, inv.issue_date.date())
                # for line in inv.lines:
                #     print(line.tax)
                self.efactoauth.addinvoice_tonextup(invoice=inv,
                                                    kind='service',
                                                    template=32,
                                                    exp_plan=691,
                                                    warehouse_code=23,
                                                    generic=True,
                                                    generic_value='27')
                os.rename(latest, latest.replace('.xml'.lower(), '_booked.xml'))
            # os.remove(latest)

    def test_verify_invoice(self):
        self.efactoauth.verify_invoice('4075641152', env='prod')

    def test_isvalid(self):
        self.efactoauth.load_token_file(token_file_path.format('efact.json'))
        print(self.efactoauth.is_valid)

    def test_get_latest_invoice(self):
        latest = self.efactoauth.get_latest_invoice('efact')
        print(latest)
        # os.replace(latest, latest.replace('.xml'.lower(), '_booked.xml'))

    def test_validate_invoice(self):
        # print(os.listdir('c:/Users/DECEL/FACT1/MZK/2023.01.03/'))
        for root, _dir, files in os.walk('c:/Users/levi/FACT1/MZK/2024_01_23'):
            for file in files:
                # print(file)
                self.efactoauth.validate_invoice(os.path.join(root, file))
        # self.efactoauth.validate_invoice('c:/Users/levi/FACT1/MZK/2024_01_23/MZK241157.xml')

    def test_rename_all(self, dir_path, kind: Union[str, 'seller', 'buyer']):  # @UndefinedVariable
        if kind == 'seller':
            account = 'AccountingCustomerParty'
            # print(os.path.basename(dir_path))
            # dir_path = os.path.join(dir_path, os.path.basename(dir_path))
        else:
            account = 'AccountingSupplierParty'
            # next_subfolder = next(os.walk(dir_path))[1][0]
        for root_f, _dirs, files in os.walk(dir_path):
            for file in files:
                # print(root_f.split('/'))
                # print(os.path.basename(dir_path))
                # print(root_f, file)
                if file.endswith('.xml') and file.count('_') < 2 and 'cif error' not in root_f:
                    # print(os.path.join(root_f, file))
                    tree = etree.parse(os.path.join(root_f, file))
                    root = tree.getroot()
                    inv_id = root.xpath('//cbc:ID', namespaces={'cbc': 'urn:oasis:names:specification:ubl:schema:xsd:CommonBasicComponents-2'})
                    inv_id = inv_id[0].text
                    parties = root.xpath('//cbc:RegistrationName', namespaces={'cbc': 'urn:oasis:names:specification:ubl:schema:xsd:CommonBasicComponents-2'})
                    for partie in parties:
                        parent_p_type = partie.getparent().getparent().getparent()
                        if etree.QName(parent_p_type).localname == account:  # @UndefinedVariable
                            # print(partie.text)
                            file_partie = partie.text

                    is_date = root.xpath('//cbc:IssueDate', namespaces={'cbc': 'urn:oasis:names:specification:ubl:schema:xsd:CommonBasicComponents-2'})
                    is_date = is_date[0].text

                    partie_ids = root.xpath('//cbc:CompanyID', namespaces={'cbc': 'urn:oasis:names:specification:ubl:schema:xsd:CommonBasicComponents-2'})
                    for partie_id in partie_ids:
                        parent_p_type = partie_id.getparent().getparent().getparent()
                        parent_t_type = partie_id.getparent()
                        # print(parent_t_type)
                        if etree.QName(parent_t_type).localname == 'PartyTaxScheme' and etree.QName(parent_p_type).localname == account:  # @UndefinedVariable
                            # print(partie_id.text)
                            file_partn_id = partie_id.text
                    new_name = '{}_{}_{}_{}.xml'.format(file.replace('.xml', ''), inv_id, file_partie, is_date)
                    current_path = os.path.join(root_f, file)
                    new_path = os.path.join(root_f, new_name)
                    #
                    # !This will rename all files
                    #
                    os.rename(current_path, new_path)
                    print(new_name)
        else:
            print('Nothing to rename')

    def test_parse_inv(self):
        # inv = self.efactoauth.parse_inv('C:/Users/Levi/git/anafapi/tests/17259191/8262600/FACTURA PRIMITA/01_2024/31/4154212674_booked.xml')
        # print(self.efactoauth._get_series(inv.invoice_id))
        inv_files = utils.get_not_booked('17259191')
        print(inv_files)


if __name__ == "__main__":
    # unittest.main()
    # efactTest().test_add_invoice()
    efactTest().test_get_all_messages('17259191', 10, '17259191', 'prod', 'P', True)
    # efactTest().test_rename_all(os.path.join(os.path.expanduser('~'), 'git/anafapi/tests/17259191/FACTURA TRIMISA'), kind='seller')
    # efactTest().test_rename_all(os.path.join(os.path.expanduser('~'), 'git/anafapi/tests/17259191/FACTURA PRIMITA'), kind='buyer')
    # anoauth.get_token()