File: fix_virtualenv.py

package info (click to toggle)
pygame-sdl2 7.4.2-1
  • links: PTS, VCS
  • area: main
  • in suites: bullseye
  • size: 29,188 kB
  • sloc: ansic: 9,664; python: 1,946; makefile: 11
file content (44 lines) | stat: -rw-r--r-- 931 bytes parent folder | download | duplicates (2)
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
41
42
43
44
from __future__ import unicode_literals, print_function

import os
import argparse
import sys
import shutil
import sysconfig

def main():
    venv = os.environ.get("VIRTUAL_ENV", None)
    if not venv:
        print("VIRTUAL_ENV is not set.")
        sys.exit(1)

    for dn in os.listdir(os.path.join(venv, "include")):
        if dn.startswith("python"):
            break

    target = os.path.join(venv, "include", dn)

    try:
        source = os.readlink(target)
    except:
        print(target, "is not a symlink. Perhaps this script has already been run.")
        sys.exit(1)

    tmp = target + ".tmp"

    if os.path.exists(tmp):
        shutil.rmtree(tmp)

    os.mkdir(tmp)

    for i in os.listdir(source):
        if i == "pygame_sdl2":
            continue

        os.symlink(os.path.join(source, i), os.path.join(tmp, i))

    os.unlink(target)
    os.rename(tmp, target)

if __name__ == "__main__":
    main()