soroban-budget-assert/resolve_567.py at main · Tollcraft/soroban-budget-assert · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
with open("cargo-budget-report/src/main.rs", "r") as f:
    text = f.read()

import re

# Block 1: Keep HEAD
text = re.sub(
    r"<<<<<<< HEAD\n(.*?)\n=======\n(?:.*?)\n>>>>>>> origin/main\n",
    r"\1\n",
    text,
    count=1,
    flags=re.DOTALL
)

# Block 2: Keep origin/main
text = re.sub(
    r"<<<<<<< HEAD\n(?:.*?)\n=======\n(.*?)\n>>>>>>> origin/main\n",
    r"\1\n",
    text,
    count=1,
    flags=re.DOTALL
)

with open("cargo-budget-report/src/main.rs", "w") as f:
    f.write(text)

with open("cargo-budget-report/tests/integration.rs", "r") as f:
    text_test = f.read()

# Block 3: Keep origin/main
text_test = re.sub(
    r"<<<<<<< HEAD\n(?:.*?)\n=======\n(.*?)\n>>>>>>> origin/main\n",
    r"\1\n",
    text_test,
    count=1,
    flags=re.DOTALL
)

with open("cargo-budget-report/tests/integration.rs", "w") as f:
    f.write(text_test)