test.py

import json
import pandas as pd
import requests
from backend import read_from_s3, write_to_s3, read_local_plant_data
from datetime import datetime, timedelta


def trefle_find_plant(query: str):
    call = f"https://trefle.io/api/v1/plants/search?q={query}&token={TREFLE_TOKEN}"
    print(f"Running trefle API search:\n{call}")
    response = requests.get(call)

    if response.status_code == 200:
        data = response.json()

        filename = "trefle_jsons/" + query.replace(" ", "_") + "_plant.json"
        with open(filename, 'w') as file:
            json.dump(data, file, indent=4)
    else:
        print(f"Failed to retrieve plants data, Status Code: {response.status_code}")
def trefle_find_species(query: str):
    call = f"https://trefle.io/api/v1/species/search?q={query}&token={TREFLE_TOKEN}"
    print(f"Running trefle API search:\n{call}")
    response = requests.get(call)

    if response.status_code == 200:
        data = response.json()

        filename = "trefle_jsons/" + query.replace(" ", "_") + "_species.json"
        with open(filename, 'w') as file:
            json.dump(data, file, indent=4)
    else:
        print(f"Failed to retrieve species data, Status Code: {response.status_code}")
def trefle_pull_request(query: str):
    response = requests.get(query)

    if response.status_code == 200:
        data = response.json()

        filename = "trefle_jsons/" + "request.json"
        with open(filename, 'w') as file:
            json.dump(data, file, indent=4)
    else:
        print(f"Failed to retrieve request, Status Code: {response.status_code}")
def trefle_pull_plant(slug: str):
    call = f"https://trefle.io/api/v1/plants/{slug}?token={TREFLE_TOKEN}"
    print(f"Running trefle API search:\n{call}")
    response = requests.get(call)

    if response.status_code == 200:
        data = response.json()
        
        filename = f"trefle_jsons/{slug}_plant.json"
        with open(filename, 'w') as file:
            json.dump(data, file, indent=4)
    else:
        print(f"Failed to retrieve plants data, Status Code: {response.status_code}")
def trefle_pull_species(slug: str):
    call = f"https://trefle.io/api/v1/species/{slug}?token={TREFLE_TOKEN}"
    print(f"Running trefle API search:\n{call}")
    response = requests.get(call)

    if response.status_code == 200:
        data = response.json()

        filename = f"trefle_jsons/{slug}_species.json"
        with open(filename, 'w') as file:
            json.dump(data, file, indent=4)
    else:
        print(f"Failed to retrieve species data, Status Code: {response.status_code}")
def trefle_pull_plant_id(id: int):
    id_str = str(id)
    print(f"Finding plant id: {id_str}")
    call = f"https://trefle.io/api/v1/plants/{id_str}?token={TREFLE_TOKEN}"
    print(f"Running trefle API search:\n{call}")
    response = requests.get(call)

    if response.status_code == 200:
        data = response.json()
        
        filename = f"trefle_jsons/{id_str}_plant.json"
        with open(filename, 'w') as file:
            json.dump(data, file, indent=4)
    else:
        print(f"Failed to retrieve plants data, Status Code: {response.status_code}")
def trefle_pull_species_id(id: int):
    id_str = str(id)
    print(f"Finding species id: {id_str}")
    call = f"https://trefle.io/api/v1/species/{id_str}?token={TREFLE_TOKEN}"
    print(f"Running trefle API search:\n{call}")
    response = requests.get(call)

    if response.status_code == 200:
        data = response.json()

        filename = f"trefle_jsons/{id_str}_species.json"
        with open(filename, 'w') as file:
            json.dump(data, file, indent=4)
    else:
        print(f"Failed to retrieve species data, Status Code: {response.status_code}")


def perenual_pull_species_list(page=1):
    call = f"https://perenual.com/api/species-list?key={PERENUAL_TOKEN}&page={page}"
    print(f"Running perenual API search:\n{call}")
    response = requests.get(call)

    if response.status_code == 200:
        data = response.json()

        filename = "perenual_jsons/plant_list.json"
        if page != 1:
            filename = f"perenual_jsons/plant_list_page_{page}.json"
        with open(filename, 'w') as file:
            json.dump(data, file, indent=4)
    else:
        print(f"Failed to retrieve plants data, Status Code: {response.status_code}")

def perenual_query_plants(query: str):
    call = f"https://perenual.com/api/species-list?key={PERENUAL_TOKEN}&q={query}"
    print(f"Running perenual API search:\n{call}")
    response = requests.get(call)

    if response.status_code == 200:
        data = response.json()

        filename = f"perenual_jsons/query_{query.replace(' ', '_')}.json"
        with open(filename, 'w') as file:
            json.dump(data, file, indent=4)
    else:
        print(f"Failed to retrieve plants data, Status Code: {response.status_code}")


def read_plant_data(filename='plants.json'):
    with open(filename, 'r') as file:
        data = json.load(file)
    return data['plants']

def write_plants_data(plants_data):
    with open('plants_TEST.json', 'w') as file:
        json.dump(plants_data, file, indent=4)

#data = read_from_s3()
#write_plants_data(data)
#write_to_s3(data)


#new_data = {
#    'water_schedule': 5,
#    'position': 1
#    }

#plants = read_plant_data("plants.json")

#breakpoint() # REMOVE

#for plant in plants:
#    if plant['id'] == 0:
#        plant.update(new_data)



#with open("perenual_jsons/query_philodendron_short_no_indent.json", 'w') as file:
#    json.dump(data, file)

#perenual_query_plants("philodendron")
#data = read_plant_data("perenual_jsons/query_hong_kong_orchid.json")

#print(len(data))

#for plant in data:
#    print(f"common_name: {plant['common_name']}")
#    print(f"watering:    {plant['watering']}")