| 123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687 |
- import os
- import django
- import random
- from datetime import datetime, timedelta
- # Set up Django environment
- os.environ.setdefault('DJANGO_SETTINGS_MODULE', 'recycle.settings')
- django.setup()
- from tracking.models import Vehicle, Trip, MaintenanceLog, Appointment, GPSData, FuelTransaction
- from django.contrib.auth.models import User
- from userprofile.models import UserProfile
- # Create mock data for Users and UserProfiles
- users_data = [
- {'username': 'driver1', 'password': 'password123', 'email': 'driver1@example.com', 'phone_number': '1234567890', 'address': '123 Main St', 'date_of_birth': '1990-01-01', 'idcard_no': '123456789', 'geolocation': '12.34,56.78', 'business_type': 'Logistics'},
- {'username': 'driver2', 'password': 'password123', 'email': 'driver2@example.com', 'phone_number': '0987654321', 'address': '456 Elm St', 'date_of_birth': '1985-05-05', 'idcard_no': '987654321', 'geolocation': '98.76,54.32', 'business_type': 'Transport'},
- ]
- users = []
- for data in users_data:
- user, created = User.objects.get_or_create(username=data['username'], defaults={'email': data['email'], 'password': data['password']})
- if created:
- user.set_password(data['password']) # Ensure the password is set correctly
- user.save()
- user_profile, created = UserProfile.objects.get_or_create(user=user, defaults={
- 'phone_number': data['phone_number'],
- 'address': data['address'],
- 'date_of_birth': data['date_of_birth'],
- 'idcard_no': data['idcard_no'],
- 'geolocation': data['geolocation'],
- 'business_type': data['business_type']
- })
- users.append(user)
- # Create mock data for Vehicles
- vehicle_data = [
- {'user': users[0], 'make': 'Toyota', 'model': 'Hilux', 'year': 2018, 'license_plate': 'ABC1234', 'capacity': 1000, 'status': 'available'},
- {'user': users[1], 'make': 'Ford', 'model': 'Ranger', 'year': 2019, 'license_plate': 'XYZ5678', 'capacity': 1200, 'status': 'in use'},
- {'user': users[0], 'make': 'Isuzu', 'model': 'D-Max', 'year': 2020, 'license_plate': 'LMN9101', 'capacity': 1500, 'status': 'under maintenance'},
- ]
- vehicles = [Vehicle.objects.create(**data) for data in vehicle_data]
- # Create mock data for Trips
- trip_data = [
- {'vehicle': vehicles[0], 'driver': users[0], 'start_location': 'Location A', 'end_location': 'Location B', 'start_time': datetime.now() - timedelta(days=1), 'end_time': datetime.now(), 'mileage': 100, 'fuel_consumption': 10},
- {'vehicle': vehicles[1], 'driver': users[1], 'start_location': 'Location C', 'end_location': 'Location D', 'start_time': datetime.now() - timedelta(days=2), 'end_time': datetime.now() - timedelta(days=1), 'mileage': 150, 'fuel_consumption': 15},
- ]
- trips = [Trip.objects.create(**data) for data in trip_data]
- # Create mock data for MaintenanceLogs
- maintenance_data = [
- {'vehicle': vehicles[2], 'date': datetime.now() - timedelta(days=10), 'description': 'Engine oil change', 'mileage': 20000},
- {'vehicle': vehicles[0], 'date': datetime.now() - timedelta(days=20), 'description': 'Brake pad replacement', 'mileage': 30000},
- ]
- maintenance_logs = [MaintenanceLog.objects.create(**data) for data in maintenance_data]
- # Create mock data for Appointments
- appointment_data = [
- {'vehicle': vehicles[0], 'start_time': datetime.now() + timedelta(days=1), 'end_time': datetime.now() + timedelta(days=1, hours=2), 'purpose': 'Pickup recycling materials'},
- {'vehicle': vehicles[1], 'start_time': datetime.now() + timedelta(days=2), 'end_time': datetime.now() + timedelta(days=2, hours=3), 'purpose': 'Deliver recycled goods'},
- ]
- appointments = [Appointment.objects.create(**data) for data in appointment_data]
- # Create mock data for GPSData
- gps_data = [
- {'vehicle': vehicles[0], 'latitude': 13.7563, 'longitude': 100.5018},
- {'vehicle': vehicles[1], 'latitude': 13.736717, 'longitude': 100.523186},
- ]
- gps_records = [GPSData.objects.create(**data) for data in gps_data]
- # Create mock data for FuelTransactions
- fuel_transaction_data = [
- {'vehicle': vehicles[0], 'date': datetime.now() - timedelta(days=1), 'amount': 50, 'price_per_liter': 1.2, 'total_cost': 60, 'fuel_card_number': '1234567890'},
- {'vehicle': vehicles[1], 'date': datetime.now() - timedelta(days=2), 'amount': 60, 'price_per_liter': 1.3, 'total_cost': 78, 'fuel_card_number': '0987654321'},
- ]
- fuel_transactions = [FuelTransaction.objects.create(**data) for data in fuel_transaction_data]
- print("Mock data created successfully!")
|