download.main() - Code Metrics - Inspection of ":memo: Architecture graphs" - Pincer-org/Pincer - Measure and Improve Code Quality continuously with Scrutinizer

Passed

Pull Request — main (#273)

unknown

created 2021-11-29 23:41 UTC

download.main() A

↳ Parent: download

Complexity

Conditions

Size

Total Lines	30
Code Lines	23

Duplication

Lines	0
Ratio	0 %

Importance

Changes

Metric	Value
eloc	23
dl	0
loc	30
rs	9.328
c	0
b	0
f	0
cc	3
nop	0

# Copyright Pincer 2021-Present
# Full MIT License can be found in `LICENSE` at the project root.

import aiohttp
import argparse
import asyncio
import base64
import glob
import io
import os
from PIL import Image


async def download_img(base64_string):
    async with aiohttp.ClientSession() as session:
        async with session.get(
            'https://mermaid.ink/img/' + base64_string.decode("ascii")
        ) as resp:
            return Image.open(io.BytesIO(await resp.content.read()))


async def download_file(file):
    with open(file) as f:
        contents = f.read()
        graphbytes = contents.encode("ascii")
        base64_bytes = base64.b64encode(graphbytes)
        image = await download_img(base64_bytes)
        image.save(os.path.splitext(file)[0] + ".png")


async def download_all(directory):
    if not directory:
        directory = ""

    await asyncio.gather(*(
        download_file(file)

        for file in glob.glob(directory + "*.mmd")
    ))


def main():
    parser = argparse.ArgumentParser(
        description='Download rendered mermaid files.'
    )
    parser.add_argument(
        '--file',
        type=str,
        nargs='?',
        help='A file to download. If not specified all files are downloaded.',
        default=None
    )
    parser.add_argument(
        '--dir',
        type=str,
        nargs='?',
        help='Directory to search. Does not work with --file.',
        default=None
    )

    args = parser.parse_args()

    if "help" in args:
        print(args)
        return

    if args.file:
        asyncio.run(download_file(args.file))
        return

    asyncio.run(download_all(args.dir))


if __name__ == "__main__":
    main()


1			# Copyright Pincer 2021-Present
2			# Full MIT License can be found in `LICENSE` at the project root.
3
4			import aiohttp
5			import argparse
6			import asyncio
7			import base64
8			import glob
9			import io
10			import os
11			from PIL import Image
12
13
14			async def download_img(base64_string):
15			async with aiohttp.ClientSession() as session:
16			async with session.get(
17			'https://mermaid.ink/img/' + base64_string.decode("ascii")
18			) as resp:
19			return Image.open(io.BytesIO(await resp.content.read()))
20
21
22			async def download_file(file):
23			with open(file) as f:
24			contents = f.read()
25			graphbytes = contents.encode("ascii")
26			base64_bytes = base64.b64encode(graphbytes)
27			image = await download_img(base64_bytes)
28			image.save(os.path.splitext(file)[0] + ".png")
29
30
31			async def download_all(directory):
32			if not directory:
33			directory = ""
34
35			await asyncio.gather(*(
36			download_file(file)
			0 ignored issues – show Comprehensibility Best Practice introduced 2021-11-29 23:43 UTC by Report Bug Copy Issue Report The variable `file` does not seem to be defined. Loading history...
37			for file in glob.glob(directory + "*.mmd")
38			))
39
40
41			def main():
42			parser = argparse.ArgumentParser(
43			description='Download rendered mermaid files.'
44			)
45			parser.add_argument(
46			'--file',
47			type=str,
48			nargs='?',
49			help='A file to download. If not specified all files are downloaded.',
50			default=None
51			)
52			parser.add_argument(
53			'--dir',
54			type=str,
55			nargs='?',
56			help='Directory to search. Does not work with --file.',
57			default=None
58			)
59
60			args = parser.parse_args()
61
62			if "help" in args:
63			print(args)
64			return
65
66			if args.file:
67			asyncio.run(download_file(args.file))
68			return
69
70			asyncio.run(download_all(args.dir))
71
72
73			if __name__ == "__main__":
74			main()
75

Pincer-org / Pincer

Pull Request — main (#273)

download.main() A

Complexity

Size

Duplication

Importance

Duplication Side-by-Side

Filter issues like