summaryrefslogtreecommitdiff
path: root/src/backend/download.py
blob: 7e2d4ec7f94226a94318ed0ab900e76d13ea3c9a (plain)
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
#!/usr/bin/python3

import atexit
import io
import os
import subprocess
import sys
import pickle
import sisyphus.getfs
import sisyphus.killemerge


def start(redir_out=False, dl_world=False):
    dl_list = []

    if dl_world:
        file_path = os.path.join(
            sisyphus.getfs.p_mtd_dir, "sisyphus_worlddeps.pickle")
    else:
        file_path = os.path.join(
            sisyphus.getfs.p_mtd_dir, "sisyphus_pkgdeps.pickle")

    with open(file_path, "rb") as f:
        bin_list, src_list, need_cfg = pickle.load(f)

    dl_list = [f'={package}' for package in bin_list]

    if redir_out:
        p_exe = subprocess.Popen(['emerge', '--nodeps', '--quiet', '--verbose', '--getbinpkg', '--fetchonly', '--rebuilt-binaries',
                                 '--with-bdeps=y', '--misspell-suggestion=n', '--fuzzy-search=n'] + list(dl_list), stdout=subprocess.PIPE, stderr=subprocess.PIPE)
        # kill portage if the program dies or it's terminated by the user
        atexit.register(sisyphus.killemerge.start, p_exe)

        for p_out in io.TextIOWrapper(p_exe.stdout, encoding="utf-8"):
            print(p_out.rstrip())

        p_exe.wait()
    else:
        p_exe = subprocess.Popen(['emerge', '--nodeps', '--quiet', '--verbose', '--getbinpkg', '--fetchonly',
                                 '--rebuilt-binaries', '--with-bdeps=y', '--misspell-suggestion=n', '--fuzzy-search=n'] + list(dl_list))
        try:
            p_exe.wait()
        except KeyboardInterrupt:
            p_exe.terminate()
            try:
                p_exe.wait(1)
            except subprocess.TimeoutExpired:
                p_exe.kill()
            sys.exit()