actions_python_cpp_compiler/util_funcs.py at master · Zoynels/actions_python_cpp_compiler · 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
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
# -*- coding: utf-8 -*-
import argparse
import glob
import os
import shutil
import subprocess
import sys
import zipfile


def install_dist(args):
    if str(args.func).lower() != "install_dist":
        return 0

    if args.pat is None:
        raise ValueError(f"Unknown 'pat' in args: {args}")

    if args.pkg_name is None:
        raise ValueError(f"Unknown 'pkg_name' in args: {args}")

    for fname in glob.glob(args.pat, recursive=True):
        subprocess.run(
            ["python", "-m", "pip", "uninstall", "-y", args.pkg_name])
        subprocess.run(["python", "-m", "pip", "install", fname])
        sys.exit(0)
    raise RuntimeError("Can't find whl file in dist folder!")


def extractall(args):
    if str(args.func).lower() != "extractall":
        return 0

    if args.pat is None:
        raise ValueError(f"Unknown 'pat' in args: {args}")

    if args.dest_to is None:
        raise ValueError(f"Unknown 'dest_to' in args: {args}")

    for fname in glob.glob(args.pat, recursive=True):
        print(f"Found: {fname}")
        if not os.path.isfile(fname):
            continue
        with zipfile.ZipFile(fname, "r") as zip_ref:
            print(f"Extract all from: {fname}")
            zip_ref.extractall(args.dest_to)


def copyfiles(args):
    if str(args.func).lower() != "copyfiles":
        return 0

    if args.pat is None:
        raise ValueError(f"Unknown 'pat' in args: {args}")

    if args.dest_to is None:
        raise ValueError(f"Unknown 'dest_to' in args: {args}")

    for fname in glob.glob(args.pat, recursive=True):
        print(f"Found: {fname}")
        if not os.path.isfile(fname):
            continue
        fname = os.path.abspath(fname)
        new_file = os.path.abspath(
            os.path.join(
                args.dest_to,
                os.path.basename(fname)))
        if not os.path.exists(os.path.dirname(new_file)):
            os.makedirs(os.path.dirname(new_file))
        print(f"Copy file from '{fname}' to '{new_file}'")
        shutil.copyfile(fname, new_file)


def showfiles(args):
    if str(args.func).lower() != "showfiles":
        return 0

    if args.pat is None:
        raise ValueError(f"Unknown 'pat' in args: {args}")

    print("#" * 100)
    for fname in glob.glob(args.pat, recursive=True):
        print(fname)


parser = argparse.ArgumentParser()
parser.add_argument("--func", help="function name")
parser.add_argument("--pat", help="pattern for function")
parser.add_argument("--dest_to", help="destination path")
parser.add_argument("--pkg_name", help="pkg_name for function")
args = parser.parse_args()

# util_funcs.py --func=extractall --pat=artifacts/** --dest_to=dist
extractall(args)
# util_funcs.py --func=install_dist --pat=dist/**/*.whl
# --pkg_name=actions_python_cpp_compiler
install_dist(args)
# util_funcs.py --func=copyfiles --pat=artifacts/**/*.whl --dest_to=dist
copyfiles(args)
# util_funcs.py --func=showfiles --pat=**/*.whl
showfiles(args)