Fix missing import of colorama Fore

This commit is contained in:
2023-08-27 14:34:47 -04:00
parent c41ec58e66
commit 7c4c0e4381

View File

@ -29,6 +29,7 @@ from lxml.etree import fromstring, tostring
from re import sub from re import sub
from yaml import load as yload from yaml import load as yload
from yaml import SafeLoader as SafeYAMLLoader from yaml import SafeLoader as SafeYAMLLoader
from colorama import Fore
from pvc.cli.helpers import * from pvc.cli.helpers import *
from pvc.cli.waiters import * from pvc.cli.waiters import *
@ -1231,11 +1232,11 @@ def cli_vm_modify(
) )
for line in diff: for line in diff:
if re.match(r"^\+", line) is not None: if re.match(r"^\+", line) is not None:
echo(CLI_CONFIG, colorama.Fore.GREEN + line + colorama.Fore.RESET) echo(CLI_CONFIG, Fore.GREEN + line + Fore.RESET)
elif re.match(r"^\-", line) is not None: elif re.match(r"^\-", line) is not None:
echo(CLI_CONFIG, colorama.Fore.RED + line + colorama.Fore.RESET) echo(CLI_CONFIG, Fore.RED + line + Fore.RESET)
elif re.match(r"^\^", line) is not None: elif re.match(r"^\^", line) is not None:
echo(CLI_CONFIG, colorama.Fore.BLUE + line + colorama.Fore.RESET) echo(CLI_CONFIG, Fore.BLUE + line + Fore.RESET)
else: else:
echo(CLI_CONFIG, line) echo(CLI_CONFIG, line)
echo(CLI_CONFIG, "") echo(CLI_CONFIG, "")
@ -4583,11 +4584,11 @@ def cli_provisioner_userdata_modify(name, filename, editor):
) )
for line in diff: for line in diff:
if re.match(r"^\+", line) is not None: if re.match(r"^\+", line) is not None:
echo(CLI_CONFIG, colorama.Fore.GREEN + line + colorama.Fore.RESET) echo(CLI_CONFIG, Fore.GREEN + line + Fore.RESET)
elif re.match(r"^\-", line) is not None: elif re.match(r"^\-", line) is not None:
echo(CLI_CONFIG, colorama.Fore.RED + line + colorama.Fore.RESET) echo(CLI_CONFIG, Fore.RED + line + Fore.RESET)
elif re.match(r"^\^", line) is not None: elif re.match(r"^\^", line) is not None:
echo(CLI_CONFIG, colorama.Fore.BLUE + line + colorama.Fore.RESET) echo(CLI_CONFIG, Fore.BLUE + line + Fore.RESET)
else: else:
echo(CLI_CONFIG, line) echo(CLI_CONFIG, line)
echo(CLI_CONFIG, "") echo(CLI_CONFIG, "")
@ -4771,11 +4772,11 @@ def cli_provisioner_script_modify(name, filename, editor):
) )
for line in diff: for line in diff:
if re.match(r"^\+", line) is not None: if re.match(r"^\+", line) is not None:
echo(CLI_CONFIG, colorama.Fore.GREEN + line + colorama.Fore.RESET) echo(CLI_CONFIG, Fore.GREEN + line + Fore.RESET)
elif re.match(r"^\-", line) is not None: elif re.match(r"^\-", line) is not None:
echo(CLI_CONFIG, colorama.Fore.RED + line + colorama.Fore.RESET) echo(CLI_CONFIG, Fore.RED + line + Fore.RESET)
elif re.match(r"^\^", line) is not None: elif re.match(r"^\^", line) is not None:
echo(CLI_CONFIG, colorama.Fore.BLUE + line + colorama.Fore.RESET) echo(CLI_CONFIG, Fore.BLUE + line + Fore.RESET)
else: else:
echo(CLI_CONFIG, line) echo(CLI_CONFIG, line)
echo(CLI_CONFIG, "") echo(CLI_CONFIG, "")