ci_update_session_record_cs.py

import base64
import os
import re
import logging


# SDK_REPO = 'c:/github/azure-sdk-for-java/sdk/resourcemanager'
SDK_REPO = "c:/github/azure-libraries-for-net"
RESOURCE_PROVIDER = "Microsoft.Resources"
VERSION_CHANGES = {"2019-08-01": "2021-01-01"}


def main():
    logging.basicConfig(level=logging.INFO)
    process_session_records()


def process_session_records():
    for root, dirs, files in os.walk(SDK_REPO):
        for name in files:
            filepath = os.path.join(root, name)
            if (
                os.path.splitext(filepath)[1] == ".json"
                and "SessionRecords" in filepath
                and "netcoreapp2.0" not in filepath
            ):
                update(filepath)


def update(filepath: str):
    with open(filepath, encoding="utf-8") as f:
        lines = f.readlines()

    modified = False
    uri = None
    out_lines = []
    for line in lines:
        if '"RequestUri"' in line and f"/providers/{RESOURCE_PROVIDER}/" in line:
            for before, after in VERSION_CHANGES.items():
                if f"api-version={before}" in line:
                    modified = True

                    line = line.replace(f"api-version={before}", f"api-version={after}")
        elif RESOURCE_PROVIDER == "Microsoft.Resources":
            # resource groups
            if (
                re.match(
                    r' {6}"RequestUri": "/subscriptions/[-a-z0-9]*/resource[Gg]roups/[-\w._()]+\?api-version=[-0-9]+",',
                    line,
                )
                or re.match(
                    r' {6}"RequestUri": "/subscriptions/[-a-z0-9]*/resource[Gg]roups\?api-version=[-0-9]+",', line
                )
                or re.match(
                    r' {6}"RequestUri": "/subscriptions/[-a-z0-9]*/resource[Gg]roups/[-\w._()]+/resources\?api-version=[-0-9]+",',
                    line,
                )
            ):
                for before, after in VERSION_CHANGES.items():
                    if f"api-version={before}" in line:
                        modified = True

                        line = line.replace(f"api-version={before}", f"api-version={after}")

        if '"RequestUri"' in line:
            m = re.search(' {6}"RequestUri": "(.+?)",\n', line)
            if m:
                uri = m.group(1)
        else:
            if uri and '"EncodedRequestUri"' in line:
                encoded_uri = str(base64.b64encode(uri.encode("utf-8")), "utf-8")
                line = f'      "EncodedRequestUri": "{encoded_uri}",\n'
                modified = True
            uri = None

        out_lines.append(line)

    if modified:
        logging.info(f"update session record {filepath}")

        with open(filepath, "w", encoding="utf-8") as f:
            f.write("".join(out_lines))


main()