122 lines
3.5 KiB
Python
122 lines
3.5 KiB
Python
import click
|
|
from flask.cli import with_appcontext
|
|
from datetime import datetime, timedelta
|
|
|
|
from app import db
|
|
from plugins.auth.models import User
|
|
from plugins.plant.models import (
|
|
Plant, PlantCommonName, PlantScientificName, PlantOwnershipLog,
|
|
PlantLineage
|
|
)
|
|
from plugins.growlog.models import PlantUpdate
|
|
from plugins.media.models import Media, ImageHeart, FeaturedImage
|
|
from plugins.submission.models import Submission, SubmissionImage
|
|
|
|
|
|
@click.command(name='preload-data') # 🔧 changed from preload_data
|
|
@with_appcontext
|
|
def preload_data_cli():
|
|
preload_data(auto=False)
|
|
|
|
def preload_data(auto=False):
|
|
if User.query.filter_by(email='admin@example.com').first():
|
|
if not auto:
|
|
click.echo("⚠️ Demo data already exists, skipping.")
|
|
return
|
|
|
|
if not auto:
|
|
click.echo("🌱 Seeding demo data...")
|
|
|
|
# USERS
|
|
admin = User(email='admin@example.com', role='admin', is_verified=True)
|
|
admin.set_password('password123')
|
|
|
|
user = User(email='user@example.com', role='user', is_verified=True)
|
|
user.set_password('password123')
|
|
|
|
db.session.add_all([admin, user])
|
|
db.session.commit()
|
|
|
|
# COMMON & SCIENTIFIC NAMES
|
|
monstera_common = PlantCommonName(name='Monstera')
|
|
deliciosa_sci = PlantScientificName(name='Monstera deliciosa')
|
|
aurea_sci = PlantScientificName(name='Monstera aurea')
|
|
db.session.add_all([monstera_common, deliciosa_sci, aurea_sci])
|
|
db.session.commit()
|
|
|
|
# PLANTS
|
|
parent_plant = Plant(
|
|
common_name_id=monstera_common.id,
|
|
scientific_name_id=deliciosa_sci.id,
|
|
created_by_user_id=admin.id
|
|
)
|
|
child_plant = Plant(
|
|
common_name_id=monstera_common.id,
|
|
scientific_name_id=aurea_sci.id,
|
|
created_by_user_id=user.id,
|
|
parent_id=1
|
|
)
|
|
db.session.add_all([parent_plant, child_plant])
|
|
db.session.flush()
|
|
|
|
# LINEAGE & OWNERSHIP
|
|
db.session.add(PlantLineage(parent_plant_id=parent_plant.id, child_plant_id=child_plant.id))
|
|
db.session.add(PlantOwnershipLog(
|
|
plant_id=child_plant.id,
|
|
user_id=user.id,
|
|
date_acquired=datetime.utcnow() - timedelta(days=20)
|
|
))
|
|
db.session.commit()
|
|
|
|
# UPDATE & MEDIA
|
|
update = PlantUpdate(
|
|
plant_id=child_plant.id,
|
|
update_type='Repotted',
|
|
description='Moved to a 6" pot with a new moss pole.',
|
|
)
|
|
db.session.add(update)
|
|
db.session.flush()
|
|
|
|
db.session.add(Media(
|
|
file_url='uploads/demo_plant_update.jpg',
|
|
update_id=update.id,
|
|
caption='Freshly repotted.'
|
|
))
|
|
db.session.commit()
|
|
|
|
# SUBMISSION & IMAGE
|
|
submission = Submission(
|
|
user_id=user.id,
|
|
plant_id=child_plant.id,
|
|
common_name='Monstera',
|
|
scientific_name='Monstera aurea',
|
|
price=120.00,
|
|
source='Etsy',
|
|
height=45,
|
|
width=30,
|
|
leaf_count=5,
|
|
potting_mix='2:1:1 bark:pumice:coco',
|
|
container_size='6"',
|
|
health_status='Healthy',
|
|
notes='Some minor yellowing on one leaf.'
|
|
)
|
|
db.session.add(submission)
|
|
db.session.flush()
|
|
|
|
image = SubmissionImage(
|
|
submission_id=submission.id,
|
|
file_path='uploads/demo_submission.jpg',
|
|
is_visible=True
|
|
)
|
|
db.session.add(image)
|
|
db.session.flush()
|
|
|
|
db.session.add_all([
|
|
ImageHeart(user_id=admin.id, submission_image_id=image.id),
|
|
FeaturedImage(submission_image_id=image.id, override_text='Gorgeous coloration', is_featured=True)
|
|
])
|
|
db.session.commit()
|
|
|
|
if not auto:
|
|
click.echo("🎉 Demo data seeded successfully.")
|