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
101
102
103
104
105
106
107
108
109
|
#
# Copyright (C) 2016-2019 by Razvan Radulescu, Nathan Lovato, and contributors
#
# This file is part of Power Sequencer.
#
# Power Sequencer is free software: you can redistribute it and/or modify it under the terms of the
# GNU General Public License as published by the Free Software Foundation, either version 3 of the
# License, or (at your option) any later version.
#
# Power Sequencer is distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY;
# without even the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License along with Power Sequencer. If
# not, see <https://www.gnu.org/licenses/>.
#
"""
Collection of utility functions, class-independent
"""
import os.path as osp
from collections import deque
from shutil import which
class ToolError(Exception):
"""Raised if external dependencies aren't found on system.
"""
pass
def checktools(tools):
tools = [(t, which(t) or "") for t in tools]
check = {"tools": tools, "test": all(map(lambda x: x[1], tools))}
if not check["test"]:
msg = ["BPSProxy couldn't find external dependencies:"]
msg += [
"[{check}] {tool}: {path}".format(
check="v" if path is not "" else "X", tool=tool, path=path or "NOT FOUND"
)
for tool, path in check["tools"]
]
msg += [
(
"Check if you have them properly installed and available in the PATH"
" environemnt variable."
)
]
raise ToolError("\n".join(msg))
def get_path_video(cfg, clargs, path, **kwargs):
return osp.join(
osp.dirname(path), cfg["proxy_directory"], osp.basename(path), "proxy_{size}.avi"
)
def get_path_image(cfg, clargs, path, **kwargs):
return osp.join(
osp.dirname(path),
cfg["proxy_directory"],
"images",
"{size}",
"{file}_proxy.jpg".format(file=osp.basename(path)),
)
def get_path(cfg, clargs, path, **kwargs):
get_path_f = {"video": get_path_video, "image": get_path_image}
what = what_vi(cfg, clargs, path, **kwargs)
return get_path_f[what](cfg, clargs, path, **kwargs)
def get_dir_video(cfg, clargs, path, **kwargs):
return iter((osp.join(osp.dirname(path), cfg["proxy_directory"], osp.basename(path)),))
def get_dir_image(cfg, clargs, path, **kwargs):
ps = osp.join(osp.dirname(path), cfg["proxy_directory"], "images", "{size}")
return map(lambda s: ps.format(size=s), clargs.sizes)
def get_dir(cfg, clargs, path, **kwargs):
get_dir_f = {"video": get_dir_video, "image": get_dir_image}
what = what_vi(cfg, clargs, path, **kwargs)
return get_dir_f[what](cfg, clargs, path, **kwargs)
def what_vi(cfg, clargs, p, **kwargs):
return "video" if osp.splitext(p)[1].lower() in cfg["extensions"]["video"] else "image"
def kickstart(it):
deque(it, maxlen=0)
def printw(cfg, text, s="\n", e="...", p="", **kwargs):
p = p or cfg["pre"]["work"]
print("{s}{p} {}{e}".format(text, s=s, e=e, p=p), **kwargs)
def printd(cfg, text, s="", e=".", p="", **kwargs):
p = p or cfg["pre"]["done"]
printw(cfg, text, s=s, e=e, p=p, **kwargs)
def prints(cfg, text, s="", e=".", p="", **kwargs):
p = p or cfg["pre"]["skip"]
printw(cfg, text, s=s, e=e, p=p, **kwargs)
|
