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
|
#!/usr/bin/env python3
# Create the psycopg-binary package by renaming and patching psycopg-c
from __future__ import annotations
import os
import re
import shutil
from pathlib import Path
curdir = Path(__file__).parent
pdir = curdir / "../.."
if (target := (pdir / "psycopg_binary")).exists():
raise Exception(f"path {target} already exists")
def sed_i(pattern: str, repl: str, filename: str | Path) -> None:
with open(filename, "rb") as f:
data = f.read()
if (newdata := re.sub(pattern.encode("utf8"), repl.encode("utf8"), data)) != data:
with open(filename, "wb") as f:
f.write(newdata)
shutil.copytree(pdir / "psycopg_c", target)
shutil.move(str(target / "psycopg_c"), str(target / "psycopg_binary"))
shutil.move(str(target / "README-binary.rst"), str(target / "README.rst"))
sed_i("psycopg-c", "psycopg-binary", target / "pyproject.toml")
sed_i("psycopg-c", "psycopg-binary", target / "psycopg_binary/version.py")
sed_i(r'"psycopg_c([\./][^"]+)?"', r'"psycopg_binary\1"', target / "pyproject.toml")
sed_i(r"__impl__\s*=.*", '__impl__ = "binary"', target / "psycopg_binary/pq.pyx")
for dirpath, dirnames, filenames in os.walk(target):
for filename in filenames:
if os.path.splitext(filename)[1] not in (".pyx", ".pxd", ".py"):
continue
sed_i(r"\bpsycopg_c\b", "psycopg_binary", Path(dirpath) / filename)
|
