16 lines
418 B
Python
16 lines
418 B
Python
import re
|
|
|
|
from logzero import logger
|
|
|
|
def check_pgroup(pgroup):
|
|
if pgroup is None:
|
|
return pgroup
|
|
pgroup = str(pgroup)
|
|
if re.match(r"^p\d{5}$", pgroup, re.I):
|
|
return pgroup.lower()
|
|
if re.match(r"^\d{5}$", pgroup):
|
|
return "p" + pgroup
|
|
msg = f"Pgroup {pgroup} supplied in incorrect format. Proper format 'pXXXXX', X is a digit"
|
|
logger.error(msg)
|
|
raise SystemExit(msg)
|