#!/usr/bin/env python
"""Bump the version number"""
import re

version_filename = "setup.py"

lines = None

with open(version_filename, "r", encoding="utf-8") as f:
    lines = f.readlines()

with open(version_filename, "w", encoding="utf-8") as f:
    for line in lines:
        if line.lstrip().startswith("version="):
            # get rid of quotes around the version
            line = line.replace('"', "")
            # get rid of trailing comma
            line = line.replace(",", "")
            # split on '='
            words = line.split("=")
            # split the version into parts (by period)
            v = words[1].split(".")
            build_num = re.findall(r"\d+", v[2])[0]
            new_build_num = str(int(build_num) + 1)
            ver = f"{v[0]}.{v[1]}.{v[2].replace(build_num, new_build_num)}".replace(
                "\n", ""
            )
            f.write(f'    version="{ver}",\n')
        else:
            f.write(line)