📦 Add: World API starter code with City schema

Author: Nisha Chaube

worldapi/__init__.py

@@ -0,0 +1,3 @@
+from django.contrib import admin
+
+# Register your models here.

worldapi/__pycache__/__init__.cpython-38.pyc

@@ -0,0 +1,5 @@
+from django.apps import AppConfig
+
+
+class WorldapiConfig(AppConfig):
+    name = 'worldapi'

worldapi/__pycache__/admin.cpython-38.pyc

@@ -0,0 +1,49 @@
+# Generated by Django 2.1.7 on 2020-10-06 00:23
+
+from django.db import migrations, models
+import django.db.models.deletion
+
+
+class Migration(migrations.Migration):
+
+    initial = True
+
+    dependencies = [
+    ]
+
+    operations = [
+        migrations.CreateModel(
+            name='City',
+            fields=[
+                ('city_id', models.AutoField(primary_key=True, serialize=False)),
+                ('city_name', models.CharField(max_length=100)),
+                ('country_code', models.CharField(max_length=100)),
+                ('city_district', models.CharField(max_length=100)),
+                ('city_population', models.IntegerField()),
+            ],
+        ),
+        migrations.CreateModel(
+            name='Country',
+            fields=[
+                ('code', models.CharField(max_length=3, primary_key=True, serialize=False)),
+                ('name', models.CharField(max_length=100)),
+                ('continent', models.CharField(max_length=100)),
+                ('region', models.CharField(max_length=100)),
+                ('indepyear', models.IntegerField()),
+                ('population', models.IntegerField()),
+                ('localname', models.CharField(max_length=100)),
+                ('governmentform', models.CharField(max_length=100)),
+                ('headofstate', models.TextField()),
+                ('capital', models.IntegerField()),
+                ('code2', models.CharField(max_length=2)),
+            ],
+        ),
+        migrations.CreateModel(
+            name='CountryLanguage',
+            fields=[
+                ('id', models.AutoField(auto_created=True, primary_key=True, serialize=False, verbose_name='ID')),
+                ('country_language', models.CharField(max_length=50)),
+                ('country_code', models.ForeignKey(on_delete=django.db.models.deletion.CASCADE, to='worldapi.City')),
+            ],
+        ),
+    ]

worldapi/__pycache__/models.cpython-38.pyc

@@ -0,0 +1,38 @@
+from django.db import models
+from django.conf import settings
+
+
+class Country(models.Model):
+    code = models.CharField(primary_key=True, max_length=3)
+    name = models.CharField(max_length=100)
+    continent = models.CharField(max_length=100)
+    region = models.CharField(max_length=100)
+    # surfacearea real  
+    indepyear = models.IntegerField()
+    population = models.IntegerField()
+    # lifeexpectancy real,
+    # gnp numeric(10,2),
+    # gnpold numeric(10,2),
+    localname = models.CharField(max_length=100)
+    governmentform = models.CharField(max_length=100)
+    headofstate = models.TextField()
+    capital = models.IntegerField()
+    code2 = models.CharField(max_length=2)
+
+
+class City(models.Model):
+    city_id = models.AutoField(primary_key=True)
+    city_name = models.CharField(max_length=100)
+    # country_code = models.ForeignKey(Country, on_delete=models.CASCADE)
+    country_code = models.CharField(max_length=100)
+    city_district = models.CharField(max_length=100)
+    city_population = models.IntegerField()
+
+class CountryLanguage(models.Model):
+    country_code = models.ForeignKey(City, on_delete=models.CASCADE)
+    country_language = models.CharField(max_length=50)
+
+    # is_official boolean NOT NULL,
+    # percentage real NOT NULL
+
+

worldapi/__pycache__/schema.cpython-38.pyc

@@ -0,0 +1,100 @@
+import graphene
+from graphene_django import DjangoObjectType
+from django.db.models import Q
+from graphene import ObjectType
+from .models import City
+
+class CityType(DjangoObjectType):
+    class Meta:
+        model = City
+
+class Query(graphene.ObjectType):
+    cities = graphene.List(
+        CityType,
+        city_id=graphene.Int(),
+        first=graphene.Int(),
+        jump=graphene.Int(),
+    )
+
+    def resolve_cities(self, info, city_id, first=None, jump=None, **kwargs):
+        all_cities = City.objects.all()
+        if city_id:
+            filter = Q(city_id__icontains=city_id)
+            filtered = all_cities.filter(filter)
+
+            if jump:
+                filtered = filtered[jump:]
+            if first:
+                filtered = filtered[:first]
+            else: 
+                filtered = City.objects.all()
+
+        return filtered        
+
+class AddCity(graphene.Mutation):
+    addCity = graphene.Field(CityType)
+
+    class Arguments:
+        city_id = graphene.Int(required=True)
+        city_name = graphene.String(required=True)
+        country_code = graphene.String(required=True)
+        city_district = graphene.String(required=True)
+        city_population = graphene.Int(required=True)
+
+    def mutate(
+        self,
+        info,
+        city_id,
+        city_name,
+        country_code,
+        city_district,
+        city_population,
+        **kwargs
+    ):
+
+        # user = info.context.user
+        # if user.is_anonymous:
+        #     raise Exception("Not logged in!!")
+
+        city = City(
+            city_id = city_id,
+            city_name = city_name,
+            country_code = country_code,
+            city_district = city_district,
+            city_population = city_population
+        )
+
+        city.save()
+
+        return AddCity(addCity=city)
+
+class DeleteCity(graphene.Mutation):
+    deleteCity = graphene.Field(CityType)
+
+    class Arguments:
+        city_id = graphene.Int(required=True)
+
+    def mutate(
+        self,
+        info,
+        city_id,
+        **kwargs
+    ):
+
+        # user = info.context.user
+        # if user.is_anonymous:
+        #     raise Exception("Not logged in!!")
+
+        city = City(
+            city_id = city_id
+        )
+
+        print(city.delete())
+
+        return True
+
+class Mutation(graphene.ObjectType):
+    add_city = AddCity.Field()
+    delete_city = DeleteCity.Field()
+
+schema = graphene.Schema(query=Query, mutation=Mutation)

worldapi/admin.py

@@ -0,0 +1,3 @@
+from django.test import TestCase
+
+# Create your tests here.

worldapi/apps.py

@@ -0,0 +1,3 @@
+from django.shortcuts import render
+
+# Create your views here.