From c1dd38e7204dd90b356aa4874b57d3c304f6811d Mon Sep 17 00:00:00 2001 From: Lucas Cimon Date: Sun, 3 Dec 2017 01:03:01 +0100 Subject: [PATCH] Fixing tests --- gen_imgs_from_mds.py | 23 +++++++---------------- run.sh | 2 +- 2 files changed, 8 insertions(+), 17 deletions(-) diff --git a/gen_imgs_from_mds.py b/gen_imgs_from_mds.py index fb27f80..62f3a8d 100755 --- a/gen_imgs_from_mds.py +++ b/gen_imgs_from_mds.py @@ -2,11 +2,10 @@ import html5lib, os, sys from markdown import markdown -SMALLEST_JPG = b'\xff\xd8\xff\xdb\x00C\x00\x03\x02\x02\x02\x02\x02\x03\x02\x02\x02\x03\x03\x03\x03\x04\x06\x04\x04\x04\x04\x04\x08\x06\x06\x05\x06\t\x08\n\n\t\x08\t\t\n\x0c\x0f\x0c\n\x0b\x0e\x0b\t\t\r\x11\r\x0e\x0f\x10\x10\x11\x10\n\x0c\x12\x13\x12\x10\x13\x0f\x10\x10\x10\xff\xc9\x00\x0b\x08\x00\x01\x00\x01\x01\x01\x11\x00\xff\xcc\x00\x06\x00\x10\x10\x05\xff\xda\x00\x08\x01\x01\x00\x00?\x00\xd2\xcf \xff\xd9' - -SMALLEST_PNG = b'\x89PNG\r\n\x1a\n\x00\x00\x00\rIHDR\x00\x00\x00\x01\x00\x00\x00\x01\x08\x06\x00\x00\x00\x1f\x15\xc4\x89\x00\x00\x00\nIDATx\x9cc\x00\x01\x00\x00\x05\x00\x01\r\n-\xb4\x00\x00\x00\x00IEND\xaeB`\x82' - -SMALLEST_GIF = b'GIF89a\x01\x00\x01\x00\x80\x00\x00\xff\xff\xff\x00\x00\x00!\xf9\x04\x01\x00\x00\x00\x00,\x00\x00\x00\x00\x01\x00\x01\x00\x00\x02\x02D\x01\x00;' +SMALLEST = {} +SMALLEST['jpg'] = SMALLEST['jpeg'] = b'\xff\xd8\xff\xdb\x00C\x00\x03\x02\x02\x02\x02\x02\x03\x02\x02\x02\x03\x03\x03\x03\x04\x06\x04\x04\x04\x04\x04\x08\x06\x06\x05\x06\t\x08\n\n\t\x08\t\t\n\x0c\x0f\x0c\n\x0b\x0e\x0b\t\t\r\x11\r\x0e\x0f\x10\x10\x11\x10\n\x0c\x12\x13\x12\x10\x13\x0f\x10\x10\x10\xff\xc9\x00\x0b\x08\x00\x01\x00\x01\x01\x01\x11\x00\xff\xcc\x00\x06\x00\x10\x10\x05\xff\xda\x00\x08\x01\x01\x00\x00?\x00\xd2\xcf \xff\xd9' +SMALLEST['png'] = b'\x89PNG\r\n\x1a\n\x00\x00\x00\rIHDR\x00\x00\x00\x01\x00\x00\x00\x01\x08\x06\x00\x00\x00\x1f\x15\xc4\x89\x00\x00\x00\nIDATx\x9cc\x00\x01\x00\x00\x05\x00\x01\r\n-\xb4\x00\x00\x00\x00IEND\xaeB`\x82' +SMALLEST['gif'] = b'GIF89a\x01\x00\x01\x00\x80\x00\x00\xff\xff\xff\x00\x00\x00!\xf9\x04\x01\x00\x00\x00\x00,\x00\x00\x00\x00\x01\x00\x01\x00\x00\x02\x02D\x01\x00;' for md_file_path in sys.argv[1:]: with open(md_file_path) as md_file: @@ -18,14 +17,6 @@ for md_file_path in sys.argv[1:]: if img_url.startswith('http'): continue os.makedirs(os.path.dirname(img_url), exist_ok=True) - if img_url.lower().endswith('jpg') or img_url.lower().endswith('jpeg'): - with open(img_url, 'wb') as img_file: - img_file.write(SMALLEST_JPG) - elif img_url.lower().endswith('png'): - with open(img_url, 'wb') as img_file: - img_file.write(SMALLEST_PNG) - elif img_url.lower().endswith('gif'): - with open(img_url, 'wb') as img_file: - img_file.write(SMALLEST_GIF) - else: - print('Unknown image type:', img_url) + ext = img_url.split('.')[-1].lower() + with open(img_url, 'wb') as img_file: + img_file.write(SMALLEST[ext]) diff --git a/run.sh b/run.sh index 2698dca..57b7161 100755 --- a/run.sh +++ b/run.sh @@ -21,7 +21,7 @@ test_ludochaordic () { cd ludochaordic pip install pelican markdown beautifulsoup4 pillow html5lib - ../gen_imgs_from_mds.py *.md + ../pelican-mg/gen_imgs_from_mds.py content/*.md make DEBUG=1 OUTPUTDIR=output html csslint --ignore=order-alphabetical output/theme/css/main.css