ev car simulation in python

main.py
import math

class EVCar:
    def __init__(self, battery_capacity, max_speed, efficiency):
        self.battery_capacity = battery_capacity # kWh
        self.max_speed = max_speed # km/h
        self.efficiency = efficiency # km/kWh
        self.battery_charge = battery_capacity # kWh
        self.speed = 0 # km/h
        self.distance = 0 # km
        self.time = 0 # seconds
        
    def charge(self, amount):
        self.battery_charge += amount
        if self.battery_charge > self.battery_capacity:
            self.battery_charge = self.battery_capacity
        
    def drive(self, time):
        # calculate distance and battery usage
        distance = self.speed * time / 3600 # km
        battery_usage = distance / self.efficiency # kWh
        
        # check if there is enough battery to drive
        if battery_usage > self.battery_charge:
            print("Not enough battery to drive.")
            return False
        
        # update car status
        self.distance += distance
        self.battery_charge -= battery_usage
        self.time += time
        
        return True

# example usage
car = EVCar(60, 120, 5)
print(car.battery_charge) # 60
print(car.speed) # 0
print(car.distance) # 0
print(car.time) # 0

car.drive(3600) # drive for 1 hour
print(car.battery_charge) # 55
print(car.speed) # 120
print(car.distance) # 120
print(car.time) # 3600
1389 chars
47 lines