Fixed errors

app/admin/views.py

@@ -6,7 +6,7 @@ from datetime import datetime
 from app.classes.pricing import (
                                  Pricing_Oil_Oil,
                                  Pricing_Oil_Oil_schema)
-
+from app.classes.admin import Admin_Company, Admin_Company_schema
 
 @admin.route("/oil/create", methods=["POST"])
 def create_oil_price():
@@ -36,53 +36,6 @@ def create_oil_price():
     }), 200
 
 
-@admin.route("/service/create", methods=["POST"])
-def create_service_price():
-    """
-    Changes general labor rates prices
-    """
-    now = datetime.utcnow()
-
-    price_service_hour = request.json["price_service_hour"]
-    price_emergency_service_hourly_rate = request.json["price_emergency_service_hour"]
-    price_emergency_call = request.json["price_emergency_call"]
-    price_out_of_oil = request.json["price_out_of_oil"]
-    price_prime = request.json["price_prime"]
-    price_cleaning = request.json["price_cleaning"]
-
-
-    price_service = Pricing_Service_General(
-        price_service_hour=price_service_hour,
-        price_out_of_oil=price_out_of_oil,
-        price_emergency_service_hour=price_emergency_service_hourly_rate,
-        price_prime=price_prime,
-        price_emergency_call=price_emergency_call,
-        price_cleaning=price_cleaning,
-        date=now,
-    )
-
-    db.session.add(price_service)
-    db.session.commit()
-
-    return jsonify({
-        "ok": True,
-        'price': price_service.id,
-    }), 200
-
-
-@admin.route("/service/get", methods=["GET"])
-def get_service_price():
-    """
-    gets service prices
-    """
-    get_service_prices = (db.session
-                          .query(Pricing_Service_General)
-                          .order_by(Pricing_Service_General.date.desc())
-                          .first())
-
-    price_schema = Pricing_Service_General_schema(many=False)
-    return jsonify(price_schema.dump(get_service_prices))
-
 
 @admin.route("/oil/get", methods=["GET"])
 def get_oil_price():
@@ -95,3 +48,13 @@ def get_oil_price():
                       .first())
     price_schema = Pricing_Oil_Oil_schema(many=False)
     return jsonify(price_schema.dump(get_oil_prices))
+
+
+@admin.route("/company/<int:company_id>", methods=["GET"])
+def get_company(company_id):
+    get_data_company = (db.session
+                        .query(Admin_Company)
+                        .first())
+
+    company_schema = Admin_Company_schema(many=False)
+    return jsonify(company_schema.dump(get_data_company))