main.py

from decimal import Decimal, getcontext
import time
import pathlib

class bcolors:
    HEADER = '\033[95m'
    OKBLUE = '\033[94m'
    OKCYAN = '\033[96m'
    OKGREEN = '\033[92m'
    WARNING = '\033[93m'
    FAIL = '\033[91m'
    ENDC = '\033[0m'
    BOLD = '\033[1m'
    UNDERLINE = '\033[4m'

print(
    f'''{bcolors.HEADER}Operation modes: {bcolors.OKBLUE}
    (1 -> Calculation Time)
    (2 -> Console)
    (3 -> Write To File){bcolors.ENDC}'''
)
mode = input(f'{bcolors.HEADER}Selected mode: {bcolors.OKBLUE} ')

if mode not in ['1', '2', '3']:
    print(f'{bcolors.FAIL}Incorrect Operating Mode!{bcolors.ENDC}')
else:

    if mode == '3':
        file_name = input(f'{bcolors.HEADER}File Name (pi.txt): {bcolors.ENDC}')
        file_name = file_name.strip('.txt')

    number = input(
        f'{bcolors.HEADER}Number of numbers after the comma: {bcolors.ENDC}')
    precision = input(f'{bcolors.HEADER}Precision (500): {bcolors.ENDC}')

    if number != '':
        if precision == '' or int(precision) > 500:
            precision = 500

        getcontext().prec = int(number)

        init_time = int(round(time.time()))

        pi = sum(1/Decimal(16)**k * (Decimal(4)/(8*k+1) -
                                     Decimal(2)/(8*k+4) -
                                     Decimal(1)/(8*k+5) -
                                     Decimal(1)/(8*k+6)) for k in range(int(precision)))

        if mode == '1':
            print(
                f'{bcolors.OKGREEN}Calculation Time: {int(round(time.time())) - init_time} Segundos!{bcolors.ENDC}')
        elif mode == '2':
            print(f'{bcolors.OKCYAN}{pi}{bcolors.ENDC}')
        elif mode == '3':
            if file_name != '':
                open(file_name + '.txt', 'w').write(str(pi))
                print(
                    f'{bcolors.WARNING}Created file path: {pathlib.Path(__file__).parent.absolute()}\{file_name}.txt {bcolors.ENDC}')
            else:
                open('pi.txt', 'w').write(str(pi))
                print(
                    f'{bcolors.WARNING}Created file path: {pathlib.Path(__file__).parent.absolute()}\pi.txt {bcolors.ENDC}')
            print(f'{bcolors.OKGREEN}Success! Saved to File.{bcolors.ENDC}')
    else:
        print(
            f'{bcolors.FAIL}Number of numbers after the comma undefined!{bcolors.ENDC}')