Price Function added

Author: Bobur Yusupov

requirements.txt

@@ -11,6 +11,8 @@ django-allauth==0.54.0
 django-cors-headers==4.1.0
 djangorestframework==3.14.0
 drf-spectacular==0.26.3
+geographiclib==2.0
+geopy==2.3.0
 idna==3.4
 inflection==0.5.1
 jsonschema==4.17.3
@@ -26,6 +28,7 @@ PyYAML==6.0
 requests==2.31.0
 requests-oauthlib==1.3.1
 sqlparse==0.4.4
+typing_extensions==4.7.0
 tzdata==2023.3
 uritemplate==4.1.1
 urllib3==2.0.3

taxi/functions/__init__.py

@@ -0,0 +1,34 @@
+from geopy import (
+    geocoders,
+    distance,
+    )
+
+geolocator = geocoders.Nominatim(user_agent="Taxi")
+
+class Distance:
+    """Returns distance between to cities."""
+    def __init__(self, first_city, second_city):
+        self.first_city = first_city
+        self.second_city = second_city
+
+    def city_coordinates(self, city_name):
+        location = geolocator.geocode(city_name)
+        coordinates = (location.latitude, location.longitude)
+
+        return coordinates
+
+    def distance(self, first_city, second_city):
+        first_city = self.city_coordinates(first_city)
+        second_city = self.city_coordinates(second_city)
+
+        distance_between_cities = distance.distance(first_city, second_city).kilometers
+
+        return distance_between_cities
+    
+    def price(self):
+        distance1 = self.distance(self.first_city, self.second_city)
+
+        oil_price = 4
+        price = distance1 / 10 * oil_price + 100 + 100
+
+        return int(price)

taxi/functions/measure_length.py

@@ -6,6 +6,8 @@
 from django.db import models
 from django.conf import settings
 
+from taxi.functions.measure_length import Distance
+
 class OrderTaxi(models.Model):
     id = models.UUIDField(primary_key=True, default=uuid4, editable=False)
     user = models.ForeignKey(
@@ -15,5 +17,10 @@ class OrderTaxi(models.Model):
     from_location = models.CharField(max_length=254)
     to_location = models.CharField(max_length=254)
 
+    @property
+    def price(self):
+        price = Distance(self.from_location, self.to_location)
+        return price.price
+
     def __str__(self):
-        return f"{self.from_location}-{self.to_location}"
+        return f"{self.from_location}-{self.to_location}"

taxi/models/models.py

@@ -10,7 +10,7 @@ class OrderTaxiSerializer(ModelSerializer):
     """OrderTaxi Serializer."""
     class Meta:
         model = OrderTaxi
-        fields = ['id', 'from_location', 'to_location']
+        fields = ['id', 'from_location', 'to_location', 'price']
         read_only_fields = ['id']
 
     def create_taxi_order(self, validated_data):