first commit
This commit is contained in:
commit
3e4574dcdd
163 files changed
+3559
No files matched your search
@@ -0,0 +1,70 @@
|
||||
from django.shortcuts import render
|
||||
from rest_framework.views import APIView
|
||||
from rest_framework.response import Response
|
||||
from rest_framework import serializers
|
||||
import pandas as pd
|
||||
from apps.appsetting.models import City as CityModel
|
||||
from apps.appsetting.selector import get_city_list
|
||||
# Create your views here.
|
||||
|
||||
|
||||
class CityListView(APIView):
|
||||
class OutputSerializer(serializers.ModelSerializer):
|
||||
class Meta:
|
||||
model = CityModel
|
||||
fields = '__all__'
|
||||
|
||||
def get(self, request):
|
||||
city_list = get_city_list()
|
||||
output = self.OutputSerializer(city_list, many=True)
|
||||
return Response(data=output.data)
|
||||
|
||||
|
||||
class ImportExcelOstanView(APIView):
|
||||
|
||||
def get(self, request):
|
||||
|
||||
df = pd.read_excel('ostan.xlsx')
|
||||
for index, row in df.iterrows():
|
||||
ostan_name = row['name']
|
||||
pv_id = row['pv_id']
|
||||
|
||||
try:
|
||||
ostan = CityModel.objects.filter(pv_id=pv_id, is_ostan=True).exists()
|
||||
except Exception as exp:
|
||||
print(ostan_name)
|
||||
print(pv_id)
|
||||
|
||||
if not ostan:
|
||||
ostan_obj = CityModel()
|
||||
ostan_obj.name = ostan_name
|
||||
ostan_obj.pv_id = pv_id
|
||||
ostan_obj.is_ostan = True
|
||||
ostan_obj.save()
|
||||
|
||||
return Response(data={"result": 'ok'})
|
||||
|
||||
|
||||
class ImportExcelCitysView(APIView):
|
||||
|
||||
def get(self, request):
|
||||
|
||||
|
||||
df = pd.read_excel('cities.xlsx')
|
||||
for index, row in df.iterrows():
|
||||
city_name = row['name']
|
||||
pv_id = row['province_id']
|
||||
|
||||
try:
|
||||
ostan = CityModel.objects.get(pv_id=pv_id, is_ostan=True)
|
||||
except Exception as exp:
|
||||
print(city_name)
|
||||
print(pv_id)
|
||||
|
||||
city_obj = CityModel()
|
||||
city_obj.name = city_name
|
||||
city_obj.is_ostan = False
|
||||
city_obj.ostan = ostan
|
||||
city_obj.save()
|
||||
|
||||
return Response(data={"result": 'ok'})
|
||||
Reference in new issue
Block a user