main.py

import argparse
import configparser
import logging
import os
import sys

from sync import constants, sync, oauth2, integrations

DEFAULT_CONFIG_PATH = "config.ini"


def ReadConfig(filename):
    """Reads and returns the ConfigParser instance."""
    config = configparser.RawConfigParser()
    config.read(os.path.join(os.path.abspath(os.path.dirname(__file__)), filename))
    return config


def CommandSync(config, args):
    """Cleanup and do syncing of products."""
    sync.DoCleanupProcedure(config)
    sync.DoSyncProcedure(config, read_only=args.readonly)


def CommandShopeeGenerateAuthorizationURL(config, args):
    """Generate authorization URL to connect Shopee app with Shopee shop."""
    sync.DoGenerateShopeeShopAuthorizationURL(config)


def CommandLazadaReauthenticate(config, args):
    """Refresh authentication token."""
    sync.DoLazadaResetAccessToken(config, args.token)


def CommandTiktokReauthenticate(config, args):
    """Refresh authentication token."""
    sync.DoTiktokResetAccessToken(config, args.token)


def CommandCleanup(config, args):
    """Cleanup dangling data."""
    sync.DoCleanupProcedure(config)


def CommandCheckConfig(config, args):
    """Check if auth config is still working."""
    logging.info(config.sections())
    oauth2_service = oauth2.Oauth2Service(dbpath=config.get("Common", "Store"))
    with oauth2_service:
        try:
            lazada_oauth2_dict = oauth2_service.GetOauth2Tokens(
                constants._SYSTEM_LAZADA
            )
            logging.info(lazada_oauth2_dict)
        except Exception as e:
            logging.error(e)
        try:
            tiktok_oauth2_dict = oauth2_service.GetOauth2Tokens(
                constants._SYSTEM_TIKTOK
            )
            logging.info(tiktok_oauth2_dict)
        except Exception as e:
            logging.error(e)


def CommandSandbox(config, args):
    """Free-for-all testing func."""

    update_oauth2_dict = {
        "access_token": "ROW_lcUwnAAAAAAdnfoUlQX2jmrN0oUL_xVy_JQLR3vxaCerOK2FjsdMFAybXqDLGmgqV6HG-hBrhgnb03b76MMglUGkPCOsUBeA",
        "access_token_expire_in": 1657951435,
        "refresh_token": "MWQ1N2NlODE4NjU0OTE2NTQ0ODg1MzQ2OWM0NjU5NGY3YzkzNzIwYWEwYjIwNw",
    }

    oauth2_service = oauth2.Oauth2Service(dbpath=config.get("Common", "Store"))
    oauth2_service.SaveOauth2Tokens(
        constants._SYSTEM_TIKTOK,
        update_oauth2_dict.get("access_token"),
        update_oauth2_dict.get("refresh_token"),
        update_oauth2_dict.get("access_token_expire_in"),
    )

    tiktok_oauth2_dict = oauth2_service.GetOauth2Tokens(constants._SYSTEM_TIKTOK)
    tiktok_client = integrations.tiktok.TiktokClient(
        domain=config.get(constants._CONFIG_TIKTOK, "Domain"),
        app_key=config.get(constants._CONFIG_TIKTOK, "AppKey"),
        app_secret=config.get(constants._CONFIG_TIKTOK, "AppSecret"),
        access_token=tiktok_oauth2_dict["access_token"],
        shop_id=config.get(constants._CONFIG_TIKTOK, "ShopID"),
        warehouse_id=config.get(constants._CONFIG_TIKTOK, "WarehouseID"),
        with_refresh=False,
    )
    sync.UpdateTiktokOauth2Tokens(oauth2_service, tiktok_client)

    # oauth2_service = oauth2.Oauth2Service(dbpath=config.get("Common", "Store"))
    # lazada_oauth2_dict = oauth2_service.GetOauth2Tokens(constants._SYSTEM_LAZADA)
    # lazada_client = integrations.lazada.LazadaClient(
    #     domain=config.get(constants._CONFIG_LAZADA, "Domain"),
    #     app_key=config.get(constants._CONFIG_LAZADA, "AppKey"),
    #     app_secret=config.get(constants._CONFIG_LAZADA, "AppSecret"),
    #     access_token=lazada_oauth2_dict["access_token"],
    #     with_confirm=True,
    #     with_refresh=False,
    # )
    # product = lazada_client.GetProductDirect("2033")
    # logging.info(
    #     f"sku:{product.model} item_id:{product.item_id} sku_id:{product.sku_id} stocks:{product.stocks}"
    # )

    # result = lazada_client.UpdateProductStocks("2787", 1)
    # logging.info(f"success updating product?: {result.error_code}")

    # lazada_client.UpdateProductStocks("1512", 10)
    # lazada_client.Refresh()
    # product = lazada_client.GetProduct("1512")
    # logging.info(f"{product.model}: {product.stocks}")


if __name__ == "__main__":
    logging.basicConfig(level=logging.DEBUG)

    # Command name to function mapping.
    COMMAND_LOOKUP = {
        "sync": CommandSync,
        "shreauth": CommandShopeeGenerateAuthorizationURL,
        "lzreauth": CommandLazadaReauthenticate,
        "tkreauth": CommandTiktokReauthenticate,
        "cleanup": CommandCleanup,
        "chkconfig": CommandCheckConfig,
        "sandbox": CommandSandbox,
    }

    # Setup argument parser.
    parser = argparse.ArgumentParser(
        description="Opencart-Lazada-Shopee-WooCommerce-Tiktok syncing script."
    )
    parser.add_argument(
        "mode",
        choices=COMMAND_LOOKUP.keys(),
        action="store",
        help="OCLZSH syncing script mode",
    )
    parser.add_argument(
        "--config",
        action="store",
        default=os.getenv("CONFIG_PATH", DEFAULT_CONFIG_PATH),
        help="path of the config file",
    )
    parser.add_argument(
        "--token",
        action="store",
        default="",
        help="token to use when reauthenticating to Lazada",
    )
    parser.add_argument(
        "--readonly",
        action="store_true",
        default=False,
        help="set to true to only read but not write",
    )
    args = parser.parse_args(sys.argv[1:])

    # Setup config path.
    config = ReadConfig(args.config)

    # Invoke action based on matching mode.
    COMMAND_LOOKUP[args.mode](config, args)