config.py

from configparser import ConfigParser
from shutil import copyfile
import logging
import math
import sys
import os

import utils

home = os.path.expanduser("~")
ATM_data_dir = home + "/.lightningATM/"
config_file_path = ATM_data_dir + "config.ini"
# check the config directory exists, create it if not
if not os.path.exists(ATM_data_dir):
    os.makedirs(ATM_data_dir)

# Set to logging.DEBUG if more info needed
logging.disable(logging.DEBUG)

# Set to logging.DEBUG if more "requests" debugging info needed
logging.getLogger("requests").setLevel(logging.INFO)
logging.getLogger("urllib3.connectionpool").setLevel(logging.INFO)

# Configure basigConfig for the "logging" module
logging.basicConfig(
    filename="{}/debug.log".format(ATM_data_dir),
    format="%(asctime)-23s %(name)-9s %(levelname)-7s | %(message)s",
    datefmt="%Y/%m/%d %I:%M:%S %p",
    level=logging.DEBUG,
)

# Create logger for this config file
logger = logging.getLogger("CONFIG")


yes = ["yes", "ye", "y"]
no = ["no", "n"]


def ask_scan_config_val(section, variable):
    while True:
        try:
            res = input(
                "Do you want to scan to input {} {}".format(section, variable)
            ).lower()
            if res in yes:
                # value = scan the qr for the value
                # update_config(section, variable, value
                ...
            elif res in no:
                return
            else:
                print("Input invalid, please try again or KeyboardInterrupt to exit")
        except KeyboardInterrupt:
            return


def check_config():
    """Checks the config and prompt the user to provide values for missing keys
    """
    if conf["lnd"]["macaroon"] is (None or ""):
        logger.warning("Missing value for lnd macaroon in config")
        ask_scan_config_val("lnd", "macaroon")
    if conf["lntxbot"]["creds"] is (None or ""):
        logger.warning("Missing value for lntxbot credential in config")
        ask_scan_config_val("lntxbot", "creds")


def update_config(section, variable, value):
    """Update the config with the new value for the variable.
    If dangermode is on, we save them to config.ini, else we write them to the temporary
    dictionary
    """
    if conf["atm"]["dangermode"].lower() == "on":
        config = create_config()
        config[section][variable] = value

        with open(CONFIG_FILE, "w") as configfile:
            config.write(configfile)
    else:
        conf[section][variable] = value


def check_dangermode():
    if conf["atm"]["dangermode"].lower() == "on":
        return True
    else:
        return False


# config file handling
def get_config_file():
    # check that the config file exists, if not copy over the example_config
    if not os.path.exists(ATM_data_dir + "config.ini"):
        example_config = os.path.join(os.path.dirname(__file__), "example_config.ini")
        copyfile(example_config, ATM_data_dir + "config.ini")
    return os.environ.get("CONFIG_FILE", config_file_path)


def create_config(config_file=None):
    parser = ConfigParser(comment_prefixes="/", allow_no_value=True, strict=False)
    parser.read(config_file or CONFIG_FILE)
    return parser


CONFIG_FILE = get_config_file()
conf = create_config()

######################################################
### (Do not change any of these parameters unless  ###
### you know exactly what you are doing.           ###
######################################################

# TODO: Add variable to set certificate check to true or false

# Papirus eInk size is 128 x 96 pixels
WHITE = 1
BLACK = 0
PAPIRUSROT = 0
if "papirus" in conf["atm"]["display"]:
    try:
        from papirus import Papirus
        PAPIRUS = Papirus(rotation=PAPIRUSROT)
    except ImportError:
        logger.warning("Papirus display library not installed.")
        print("Papirus library not installed.")
        sys.exit("Exiting...")

# Display - Waveshare 2.13 V2 is 250 * 122 pixels
elif "waveshare2in13v2" in conf["atm"]["display"]:
    try:
        from waveshare_epd import epd2in13_V2
        WAVESHARE = epd2in13_V2.EPD()
    except ImportError:
        logger.warning("Waveshare display library not installed.")
        print("Waveshare2in13v2 library not installed.")
        sys.exit("Exiting...")

# Display - Waveshare 2.13 V3 is 250 * 122 pixels
elif "waveshare2in13v3" in conf["atm"]["display"]:
    try:
        from waveshare_epd import epd2in13_V3
        WAVESHARE = epd2in13_V3.EPD()
    except ImportError:
        logger.warning("Waveshare display library not installed.")
        print("Waveshare2in13v3 library not installed.")
        sys.exit("Exiting...")

# Display - Waveshare 2.13 (D) is 212 * 104 pixels
elif "waveshare2in13d" in conf["atm"]["display"]:
    try:
        from waveshare_epd import epd2in13d
        WAVESHARE = epd2in13d.EPD()
    except ImportError:
        logger.warning("Waveshare display library not installed.")
        print("Waveshare2in13d library not installed.")
        sys.exit("Exiting...")

# Display - Waveshare 2.66 is 296 * 152 pixels
elif "waveshare2in66" in conf["atm"]["display"]:
    try:
        from waveshare_epd import epd2in66
        WAVESHARE = epd2in66.EPD()
    except ImportError:
        logger.warning("Waveshare display library not installed.")
        print("Waveshare2in66 library not installed.")
        sys.exit("Exiting...")

# Display - Waveshare 2.7 is 264 * 176 pixels
elif "waveshare2in7" in conf["atm"]["display"]:
    try:
        from waveshare_epd import epd2in7
        WAVESHARE = epd2in7.EPD()
    except ImportError:
        logger.warning("Waveshare display library not installed.")
        print("Waveshare2in7 library not installed.")
        sys.exit("Exiting...")

# Display - Waveshare 2.9 (D) is 296 * 128 pixels
elif "waveshare2in9d" in conf["atm"]["display"]:
    try:
        from waveshare_epd import epd2in9d
        WAVESHARE = epd2in9d.EPD()
    except ImportError:
        logger.warning("Waveshare display library not installed.")
        print("Waveshare2in9d library not installed.")
        sys.exit("Exiting...")

# Display - Inky pHAT
elif "inkyphat" in conf["atm"]["display"]:
    try:
        from inky.auto import auto

        WHITE = 0
        BLACK = 1
        INKY = auto(ask_user=True, verbose=True)
        INKY.set_border(INKY.WHITE)
    except ImportError:
        logger.warning("Inky display library not installed.")
        print("Inkyphat library not installed.")
        sys.exit("Exiting...")

# This could help for testing purposes (e.g. tests/lnbits_lnurlw.py)
elif "testing" in conf["atm"]["display"]:
    logger.warning("Display is in testing mode. This won't work in production.")

# Display - No configuration match
else:
    logger.warning("No display configuration match.")
    print("No display configuration matched.")
    sys.exit("Exiting...")

# API URL for coingecko
COINGECKO_URL_BASE = "https://api.coingecko.com/api/v3/"

# Fiat and satoshi variables
FIAT = 0
SATS = 0
SATSFEE = 0
INVOICE = ""

# Set btc and sat price
BTCPRICE = utils.get_btc_price(conf["atm"]["cur"])
SATPRICE = (1 / (BTCPRICE * 100)) * 1e8

# Button / Acceptor Pulses
LASTIMPULSE = 0
PULSES = 0
LASTPUSHES = 0
PUSHES = 0
COINCOUNT = 0

# State
CONNECTIVITY=True

# Determine different coin types
COINTYPES={}
try:
    for coin_type in conf['coins']['coin_types'].split("\n"):
        coin_type_pulses,coin_type_fiat,coin_type_name=coin_type.split(',')
        COINTYPES[int(coin_type_pulses)]={'fiat': coin_type_fiat, 'name': coin_type_name}
except:
    print("Pulses not set in the new way, please update config.")
    sys.exit(2)