83 lines
2.8 KiB
Python
83 lines
2.8 KiB
Python
from app.core.database import engine
|
|
from sqlalchemy import text
|
|
import uuid
|
|
|
|
def seed_taxis():
|
|
taxis_data = [
|
|
{
|
|
'id': str(uuid.uuid4()),
|
|
'owner_name': 'Taxi Volcán Express',
|
|
'phone_number': '+507 771-3344',
|
|
'license_plate': 'CHI-1234',
|
|
'cooperative': 'Cooperativa Boquete',
|
|
'corregimiento': 'Boquete',
|
|
'shift': 'dia',
|
|
'rating': 4.5,
|
|
'english_speaking': True,
|
|
'is_active': True
|
|
},
|
|
{
|
|
'id': str(uuid.uuid4()),
|
|
'owner_name': 'Taxi Taroe Boquete',
|
|
'phone_number': '+507 722-8890',
|
|
'license_plate': 'CHI-5678',
|
|
'cooperative': 'Cooperativa Boquete',
|
|
'corregimiento': 'David - Boquete',
|
|
'shift': 'tarde',
|
|
'rating': 5.0,
|
|
'english_speaking': True,
|
|
'is_active': True
|
|
},
|
|
{
|
|
'id': str(uuid.uuid4()),
|
|
'owner_name': 'Taxi Aeropuerto Express',
|
|
'phone_number': '+507 788-9900',
|
|
'license_plate': 'CHI-9012',
|
|
'cooperative': None,
|
|
'corregimiento': 'Aeropuerto - Boquete',
|
|
'shift': 'dia',
|
|
'rating': 4.8,
|
|
'english_speaking': True,
|
|
'is_active': True
|
|
},
|
|
{
|
|
'id': str(uuid.uuid4()),
|
|
'owner_name': 'Taxi Nocturno Boquete',
|
|
'phone_number': '+507 755-4433',
|
|
'license_plate': 'CHI-3456',
|
|
'cooperative': 'Cooperativa Boquete',
|
|
'corregimiento': 'Boquete - David',
|
|
'shift': 'noche',
|
|
'rating': 4.2,
|
|
'english_speaking': False,
|
|
'is_active': True
|
|
},
|
|
{
|
|
'id': str(uuid.uuid4()),
|
|
'owner_name': 'Taxi Local Boquete',
|
|
'phone_number': '+507 766-5544',
|
|
'license_plate': 'CHI-7890',
|
|
'cooperative': None,
|
|
'corregimiento': 'Boquete',
|
|
'shift': 'tarde',
|
|
'rating': 4.7,
|
|
'english_speaking': False,
|
|
'is_active': True
|
|
}
|
|
]
|
|
|
|
with engine.connect() as conn:
|
|
for taxi in taxis_data:
|
|
conn.execute(text("""
|
|
INSERT INTO taxis (id, owner_name, phone_number, license_plate, cooperative,
|
|
corregimiento, shift, rating, english_speaking, is_active)
|
|
VALUES (:id, :owner_name, :phone_number, :license_plate, :cooperative,
|
|
:corregimiento, :shift, :rating, :english_speaking, :is_active)
|
|
ON CONFLICT (license_plate) DO NOTHING
|
|
"""), taxi)
|
|
conn.commit()
|
|
print(f"✅ Inserted {len(taxis_data)} taxis successfully!")
|
|
|
|
if __name__ == "__main__":
|
|
seed_taxis()
|