mirror of
https://github.com/unshackle-dl/unshackle.git
synced 2026-05-17 14:29:27 +00:00
Compare commits
7 Commits
a6494d9b54
...
b16610ac63
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
b16610ac63 | ||
|
|
5fae23eb99 | ||
|
|
6186ff764b | ||
|
|
207756c090 | ||
|
|
9e194f4868 | ||
|
|
5b50a6cd79 | ||
|
|
289c8a3b23 |
@@ -47,6 +47,7 @@ class WindscribeVPN(Proxy):
|
|||||||
|
|
||||||
Supports:
|
Supports:
|
||||||
- Country code: "us", "ca", "gb"
|
- Country code: "us", "ca", "gb"
|
||||||
|
- Specific server: "sg007", "us150"
|
||||||
- City selection: "us:seattle", "ca:toronto"
|
- City selection: "us:seattle", "ca:toronto"
|
||||||
"""
|
"""
|
||||||
query = query.lower()
|
query = query.lower()
|
||||||
@@ -64,7 +65,17 @@ class WindscribeVPN(Proxy):
|
|||||||
elif query in self.server_map and not city:
|
elif query in self.server_map and not city:
|
||||||
hostname = self.server_map[query]
|
hostname = self.server_map[query]
|
||||||
else:
|
else:
|
||||||
if re.match(r"^[a-z]+$", query):
|
server_match = re.match(r"^([a-z]{2})(\d+)$", query)
|
||||||
|
if server_match:
|
||||||
|
# Specific server selection, e.g., sg007, us150
|
||||||
|
country_code, server_num = server_match.groups()
|
||||||
|
hostname = self.get_specific_server(country_code, server_num)
|
||||||
|
if not hostname:
|
||||||
|
raise ValueError(
|
||||||
|
f"No WindscribeVPN server found matching '{query}'. "
|
||||||
|
f"Check the server number or use just '{country_code}' for a random server."
|
||||||
|
)
|
||||||
|
elif re.match(r"^[a-z]+$", query):
|
||||||
hostname = self.get_random_server(query, city)
|
hostname = self.get_random_server(query, city)
|
||||||
else:
|
else:
|
||||||
raise ValueError(f"The query provided is unsupported and unrecognized: {query}")
|
raise ValueError(f"The query provided is unsupported and unrecognized: {query}")
|
||||||
@@ -75,6 +86,38 @@ class WindscribeVPN(Proxy):
|
|||||||
hostname = hostname.split(':')[0]
|
hostname = hostname.split(':')[0]
|
||||||
return f"https://{self.username}:{self.password}@{hostname}:443"
|
return f"https://{self.username}:{self.password}@{hostname}:443"
|
||||||
|
|
||||||
|
def get_specific_server(self, country_code: str, server_num: str) -> Optional[str]:
|
||||||
|
"""
|
||||||
|
Find a specific server by country code and server number.
|
||||||
|
|
||||||
|
Matches against hostnames like "sg-007.totallyacdn.com" for query "sg007".
|
||||||
|
Tries both the raw number and zero-padded variants.
|
||||||
|
|
||||||
|
Args:
|
||||||
|
country_code: Two-letter country code (e.g., "sg", "us")
|
||||||
|
server_num: Server number as string (e.g., "007", "7", "150")
|
||||||
|
|
||||||
|
Returns:
|
||||||
|
The matching hostname, or None if not found.
|
||||||
|
"""
|
||||||
|
num_stripped = server_num.lstrip("0") or "0"
|
||||||
|
candidates = {
|
||||||
|
f"{country_code}-{server_num}.",
|
||||||
|
f"{country_code}-{num_stripped}.",
|
||||||
|
f"{country_code}-{server_num.zfill(3)}.",
|
||||||
|
}
|
||||||
|
|
||||||
|
for location in self.countries:
|
||||||
|
if location.get("country_code", "").lower() != country_code:
|
||||||
|
continue
|
||||||
|
for group in location.get("groups", []):
|
||||||
|
for host in group.get("hosts", []):
|
||||||
|
hostname = host.get("hostname", "")
|
||||||
|
if any(hostname.startswith(prefix) for prefix in candidates):
|
||||||
|
return hostname
|
||||||
|
|
||||||
|
return None
|
||||||
|
|
||||||
def get_random_server(self, country_code: str, city: Optional[str] = None) -> Optional[str]:
|
def get_random_server(self, country_code: str, city: Optional[str] = None) -> Optional[str]:
|
||||||
"""
|
"""
|
||||||
Get a random server hostname for a country, optionally filtered by city.
|
Get a random server hostname for a country, optionally filtered by city.
|
||||||
|
|||||||
@@ -218,6 +218,8 @@ class Episode(Title):
|
|||||||
for indicator in ["HDR10", "SMPTE ST 2086"]
|
for indicator in ["HDR10", "SMPTE ST 2086"]
|
||||||
):
|
):
|
||||||
name += " HDR"
|
name += " HDR"
|
||||||
|
elif "HDR Vivid" in hdr_format:
|
||||||
|
name += " HDR"
|
||||||
else:
|
else:
|
||||||
name += f" {DYNAMIC_RANGE_MAP.get(hdr_format)} "
|
name += f" {DYNAMIC_RANGE_MAP.get(hdr_format)} "
|
||||||
elif "HLG" in trc or "Hybrid Log-Gamma" in trc or "ARIB STD-B67" in trc or "arib-std-b67" in trc.lower():
|
elif "HLG" in trc or "Hybrid Log-Gamma" in trc or "ARIB STD-B67" in trc or "arib-std-b67" in trc.lower():
|
||||||
|
|||||||
@@ -153,6 +153,8 @@ class Movie(Title):
|
|||||||
for indicator in ["HDR10", "SMPTE ST 2086"]
|
for indicator in ["HDR10", "SMPTE ST 2086"]
|
||||||
):
|
):
|
||||||
name += " HDR"
|
name += " HDR"
|
||||||
|
elif "HDR Vivid" in hdr_format:
|
||||||
|
name += " HDR"
|
||||||
else:
|
else:
|
||||||
name += f" {DYNAMIC_RANGE_MAP.get(hdr_format)} "
|
name += f" {DYNAMIC_RANGE_MAP.get(hdr_format)} "
|
||||||
elif "HLG" in trc or "Hybrid Log-Gamma" in trc or "ARIB STD-B67" in trc or "arib-std-b67" in trc.lower():
|
elif "HLG" in trc or "Hybrid Log-Gamma" in trc or "ARIB STD-B67" in trc or "arib-std-b67" in trc.lower():
|
||||||
|
|||||||
@@ -95,7 +95,7 @@ class Tracks:
|
|||||||
|
|
||||||
return rep
|
return rep
|
||||||
|
|
||||||
def tree(self, add_progress: bool = False) -> tuple[Tree, list[partial]]:
|
def tree(self, add_progress: bool = False) -> tuple[Tree, list[Callable[..., None]]]:
|
||||||
all_tracks = [*list(self), *self.chapters, *self.attachments]
|
all_tracks = [*list(self), *self.chapters, *self.attachments]
|
||||||
|
|
||||||
progress_callables = []
|
progress_callables = []
|
||||||
@@ -121,7 +121,29 @@ class Tracks:
|
|||||||
speed_estimate_period=10,
|
speed_estimate_period=10,
|
||||||
)
|
)
|
||||||
task = progress.add_task("", downloaded="-")
|
task = progress.add_task("", downloaded="-")
|
||||||
progress_callables.append(partial(progress.update, task_id=task))
|
state = {"total": 100.0}
|
||||||
|
|
||||||
|
def update_track_progress(
|
||||||
|
task_id: int = task,
|
||||||
|
_state: dict[str, float] = state,
|
||||||
|
_progress: Progress = progress,
|
||||||
|
**kwargs,
|
||||||
|
) -> None:
|
||||||
|
"""
|
||||||
|
Ensure terminal status states render as a fully completed bar.
|
||||||
|
|
||||||
|
Some downloaders can report completed slightly below total
|
||||||
|
before emitting the final "Downloaded" state.
|
||||||
|
"""
|
||||||
|
if "total" in kwargs and kwargs["total"] is not None:
|
||||||
|
_state["total"] = kwargs["total"]
|
||||||
|
|
||||||
|
downloaded_state = kwargs.get("downloaded")
|
||||||
|
if downloaded_state in {"Downloaded", "Decrypted", "[yellow]SKIPPED"}:
|
||||||
|
kwargs["completed"] = _state["total"]
|
||||||
|
_progress.update(task_id=task_id, **kwargs)
|
||||||
|
|
||||||
|
progress_callables.append(update_track_progress)
|
||||||
track_table = Table.grid()
|
track_table = Table.grid()
|
||||||
track_table.add_row(str(track)[6:], style="text2")
|
track_table.add_row(str(track)[6:], style="text2")
|
||||||
track_table.add_row(progress)
|
track_table.add_row(progress)
|
||||||
|
|||||||
Reference in New Issue
Block a user