Function main

tools/generate_exceptions.py:68–184

()

def main():
    parser = argparse.ArgumentParser(
        description='generate _exceptions.py from postgres/errcodes.txt')
    parser.add_argument('errcodesfile', type=str,
                        help='path to errcodes.txt in PostgreSQL source')

    args = parser.parse_args()

    with open(args.errcodesfile, 'r') as errcodes_f:
        errcodes = errcodes_f.read()

    section_re = re.compile(r'^Section: .*')

    tpl = """\
class {clsname}({base}):
    {docstring}{sqlstate}"""

    new_section = True
    section_class = None

    buf = '# GENERATED FROM postgresql/src/backend/utils/errcodes.txt\n' + \
          '# DO NOT MODIFY, use tools/generate_exceptions.py to update\n\n' + \
          'from ._base import *  # NOQA\nfrom . import _base\n\n\n'

    classes = []
    clsnames = set()

    def _add_class(clsname, base, sqlstate, docstring):
        if sqlstate:
            sqlstate = "sqlstate = '{}'".format(sqlstate)
        else:
            sqlstate = ''

        txt = tpl.format(clsname=clsname, base=base, sqlstate=sqlstate,
                         docstring=docstring)

        if not sqlstate and not docstring:
            txt += 'pass'

        if len(txt.splitlines()[0]) > 79:
            txt = txt.replace('(', '(\n        ', 1)

        classes.append(txt)
        clsnames.add(clsname)

    for line in errcodes.splitlines():
        if not line.strip() or line.startswith('#'):
            continue

        if section_re.match(line):
            new_section = True
            continue

        parts = re.split(r'\s+', line)

        if len(parts) < 4:
            continue