mirror of
https://github.com/ulbwazhine/MockupEngineer.git
synced 2024-11-10 19:26:34 +03:00
85 lines
3.1 KiB
Python
85 lines
3.1 KiB
Python
import datetime
|
|
import os
|
|
import shutil
|
|
|
|
from FluentGenerator import FluentImage
|
|
from PIL import Image
|
|
from jinja2 import Template
|
|
|
|
from MockupEngineer import MockupEngineerInstance, generated_path
|
|
|
|
|
|
def create_examples():
|
|
mockup = MockupEngineerInstance()
|
|
|
|
max_size = 1000
|
|
|
|
for template in mockup.templates:
|
|
screenshot_path = FluentImage(
|
|
colors=10, width=template.image.width, height=template.image.height, background='white').generate()
|
|
mockup_path = mockup.generate(template_id=template.id, screenshot_path=screenshot_path)
|
|
os.remove(screenshot_path)
|
|
shutil.move(mockup_path, template.preview)
|
|
|
|
mockup_img = Image.open(template.preview)
|
|
background_img = Image.new('RGBA', mockup_img.size, (255, 255, 255, 255))
|
|
background_img.paste(mockup_img, (0, 0), mockup_img)
|
|
|
|
if background_img.size[0] > max_size or background_img.size[1] > max_size:
|
|
if background_img.size[0] > background_img.size[1]:
|
|
width = max_size
|
|
height = int(background_img.size[1] * (max_size / background_img.size[0]))
|
|
else:
|
|
width = int(background_img.size[0] * (max_size / background_img.size[1]))
|
|
height = max_size
|
|
|
|
background_img = background_img.resize((width, height), Image.ANTIALIAS)
|
|
background_img.save(template.preview)
|
|
|
|
shutil.rmtree(generated_path)
|
|
|
|
|
|
def create_readme():
|
|
mockup = MockupEngineerInstance()
|
|
|
|
with open(os.path.join(os.path.dirname(os.path.abspath(__file__)), 'readme-template.md'), 'r', encoding='utf8') as f:
|
|
data = Template(f.read()).render(templates=mockup.get_templates(), os=os, enumerate=enumerate)
|
|
|
|
with open(os.path.join(os.path.dirname(os.path.abspath(__file__)), 'readme.md'), 'w', encoding='utf8') as f:
|
|
f.write(data)
|
|
|
|
|
|
def bump_version():
|
|
with open(os.path.join(os.path.dirname(os.path.abspath(__file__)), 'setup.py'), 'r') as f:
|
|
data = f.read()
|
|
|
|
setup_version = data.split('version=\'')[1].split('\',')[0]
|
|
setup_date = datetime.datetime.strptime(".".join(setup_version.split('.')[:-1]), '%Y.%m.%d')
|
|
setup_variant = int(setup_version.split('.')[-1])
|
|
current_date = datetime.datetime.utcnow()
|
|
setup_variant = 1 if (current_date.day > setup_date.day or
|
|
current_date.month > setup_date.month or
|
|
current_date.year > setup_date.year) else setup_variant + 1
|
|
setup_date = current_date.strftime('%Y.%m.%d')
|
|
bumped = data.replace(setup_version, '{}.{}'.format(setup_date, setup_variant))
|
|
|
|
with open(os.path.join(os.path.dirname(os.path.abspath(__file__)), 'setup.py'), 'w') as f:
|
|
f.write(bumped)
|
|
|
|
|
|
def run_pypi():
|
|
print(f'cd "{os.path.dirname(os.path.abspath(__file__))}"')
|
|
print('python setup.py sdist')
|
|
print('twine upload dist/*')
|
|
input()
|
|
os.chdir(os.path.dirname(os.path.abspath(__file__)))
|
|
os.remove('dist')
|
|
os.remove('MockupEngineer.egg-info')
|
|
|
|
|
|
if __name__ == '__main__':
|
|
create_readme()
|
|
# create_examples()
|
|
bump_version()
|
|
run_pypi()
|