diff options
Diffstat (limited to '')
| -rw-r--r-- | README.md | 43 | ||||
| -rwxr-xr-x | magni.py | 17 |
2 files changed, 45 insertions, 15 deletions
@@ -2,9 +2,46 @@ magni takes a url and upscales all the images inside and returns a simple html page with the images embedded.</br> ```sh -HTTPS_PROXY=socks5h://127.0.0.1:9094 ./magni.py --url https://chapmanganato.com/manga-dt980702/chapter-184 +usage: magni.py [-h] [--url URL] [--method METHOD] [--port PORT] + +options: + -h, --help show this help message and exit + --url URL, -u URL the url to the page containing the images + --method METHOD, -m METHOD + the method to use. either fsrcnn or espcn + --port PORT, -p PORT the port to serve the images over +``` + +## Install and Run + +### Install +```sh +poetry install +``` + +### Run +```sh +poetry shell && HTTPS_PROXY=socks5h://127.0.0.1:9094 ./magni.py --url https://chapmanganato.com/manga-dt980702/chapter-184 +``` +you can obviously use `poetry run` as well: +```sh +HTTPS_PROXY=socks5h://127.0.0.1:9094 poetry run ./magni.py --url https://chapmanganato.com/manga-dt980702/chapter-184 ``` -## Notes -* `magni` can use a socks5 proxy as is displayed in the above example.</br> +## Env Vars +magni recognizes three environment variables:</br> + +### HTTPS_PROXY +You must specify a socks5 proxy here since magni uses `pysocks` to make the connections.</br> +If the env var is not defined magni will not use any proxy.</br> + +### MAGNI_MODEL_PATH +Path to the directory where magni will store the model.</br> +If the env var is not defined, magni will use `./models` as a default value.</br> + +### MAGNI_IMAGE_PATH +Path to the directory where magni will store the upscaled images.</br> +If the env var is not defined or empty magni will use `./images` as a default value.</br> + +## TODO * currently the models we are using are not as effective. I should either fine ones that are specifically trained on greyscale images or just train some myself.</br> @@ -1,6 +1,5 @@ #!/usr/bin/env python """Magni.""" -# HTTPS_PROXY=socks5h://127.0.0.1:9094 ./magni.py --url https://chapmanganato.com/manga-dt980702/chapter-184 import argparse import asyncio @@ -8,6 +7,7 @@ import concurrent.futures import http.server import os import random +import secrets import socketserver import sys import typing @@ -130,7 +130,7 @@ def get_user_agent() -> str: "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/108.0.0.0 Safari/537.36", ] - return user_agents[random.randint(0, len(user_agents) - 1)] + return user_agents[secrets.randbelow(len(user_agents))] def get_proxies() -> typing.Dict: @@ -192,9 +192,7 @@ def multi_get_tag( # flake8: noqa: E501 -async def model_downloader() -> typing.Optional[ - typing.List[typing.Tuple[str, str]] -]: +async def model_downloader() -> typing.Optional[typing.List[typing.Tuple[str, str]]]: """Download the models.""" down_list = [ "https://github.com/fannymonori/TF-ESPCN/raw/master/export/ESPCN_x3.pb", @@ -208,9 +206,7 @@ async def model_downloader() -> typing.Optional[ url_list: typing.List[str] = [] for model in down_list: if ( - os.path.exists( - get_model_path() + "/" + model[model.rfind("/") + 1 :] - ) + os.path.exists(get_model_path() + "/" + model[model.rfind("/") + 1 :]) is False ): url_list.append(model) @@ -223,10 +219,7 @@ async def model_downloader() -> typing.Optional[ ] = multi_get_tag(url_tag_list) model_path: str = os.getcwd() - if ( - "MAGNI_MODEL_PATH" in os.environ - and os.environ["MAGNI_MODEL_PATH"] != "" - ): + if "MAGNI_MODEL_PATH" in os.environ and os.environ["MAGNI_MODEL_PATH"] != "": model_path = os.environ["MAGNI_MODEL_PATH"] if response_list is None: |