mirror of
https://github.com/Crocmagnon/plant-badger.git
synced 2024-11-22 01:18:02 +01:00
Add script to download and prepare image
This commit is contained in:
parent
b691341c16
commit
bb55abeb02
4 changed files with 36 additions and 0 deletions
|
@ -2,3 +2,5 @@ mpremote
|
||||||
invoke>=2.0.0
|
invoke>=2.0.0
|
||||||
black
|
black
|
||||||
pre-commit
|
pre-commit
|
||||||
|
requests
|
||||||
|
Pillow
|
||||||
|
|
0
src/images/.gitkeep
Normal file
0
src/images/.gitkeep
Normal file
Binary file not shown.
Before Width: | Height: | Size: 12 KiB |
34
tasks.py
34
tasks.py
|
@ -1,6 +1,7 @@
|
||||||
import time
|
import time
|
||||||
from pathlib import Path
|
from pathlib import Path
|
||||||
|
|
||||||
|
from PIL import Image
|
||||||
from invoke import task, Context
|
from invoke import task, Context
|
||||||
|
|
||||||
BASE_DIR = Path(__file__).parent.resolve(strict=True)
|
BASE_DIR = Path(__file__).parent.resolve(strict=True)
|
||||||
|
@ -39,8 +40,41 @@ def list(c: Context):
|
||||||
|
|
||||||
|
|
||||||
@task
|
@task
|
||||||
|
def download_image(c: Context):
|
||||||
|
import requests
|
||||||
|
import sys
|
||||||
|
|
||||||
|
sys.path.insert(0, str(SRC_DIR))
|
||||||
|
from secrets import HA_ACCESS_TOKEN, HA_BASE_URL, HA_PLANT_ID
|
||||||
|
|
||||||
|
url = HA_BASE_URL + "/states/" + HA_PLANT_ID
|
||||||
|
headers = {"Authorization": "Bearer " + HA_ACCESS_TOKEN}
|
||||||
|
res = requests.get(url, headers=headers)
|
||||||
|
data = res.json()
|
||||||
|
image_url = data["attributes"]["entity_picture"]
|
||||||
|
image_path = SRC_DIR / "images" / "plant.jpg"
|
||||||
|
c.run(f"curl -o {image_path} {image_url}", pty=True, echo=True)
|
||||||
|
|
||||||
|
# resize image_path to 128x128 with Pillow
|
||||||
|
image = Image.open(image_path)
|
||||||
|
image = image.resize((128, 128))
|
||||||
|
|
||||||
|
# crop image to 104x128, centered
|
||||||
|
left = int((image.width - 104) / 2)
|
||||||
|
top = 0
|
||||||
|
right = left + 104
|
||||||
|
bottom = top + 128
|
||||||
|
image = image.crop((left, top, right, bottom))
|
||||||
|
|
||||||
|
# convert image to grayscale
|
||||||
|
image = image.convert("L")
|
||||||
|
image.save(image_path)
|
||||||
|
|
||||||
|
|
||||||
|
@task(pre=[download_image])
|
||||||
def initial_setup(c: Context, board_id: str):
|
def initial_setup(c: Context, board_id: str):
|
||||||
"""Install dependencies and copy project files to the board."""
|
"""Install dependencies and copy project files to the board."""
|
||||||
|
wipe(c, board_id)
|
||||||
with c.cd(SRC_DIR):
|
with c.cd(SRC_DIR):
|
||||||
if MICROPYTHON_DEPENDENCIES:
|
if MICROPYTHON_DEPENDENCIES:
|
||||||
deps = " ".join(MICROPYTHON_DEPENDENCIES)
|
deps = " ".join(MICROPYTHON_DEPENDENCIES)
|
||||||
|
|
Loading…
Reference in a new issue