wip graphql hn clone

Author: Brian Caffey

backend/apps/accounts/schema.py

@@ -0,0 +1,46 @@
+from django.contrib.auth import get_user_model
+
+import graphene
+from graphene_django import DjangoObjectType
+
+
+class UserType(DjangoObjectType):
+    class Meta:
+        model = get_user_model()
+
+
+class Query(graphene.ObjectType):
+    current_user = graphene.Field(UserType)
+    users = graphene.List(UserType)
+
+    def resolve_users(self, info):
+        return get_user_model().objects.all()
+
+    def resolve_current_user(self, info):
+        user = info.context.user
+        print(dir(info.context))
+        if user.is_anonymous:
+            raise Exception("Not logged in...")
+
+        return user
+
+
+class CreateUser(graphene.Mutation):
+    user = graphene.Field(UserType)
+
+    class Arguments:
+        password = graphene.String(required=True)
+        email = graphene.String(required=True)
+
+    def mutate(self, info, password, email):
+        user = get_user_model()(
+            email=email
+        )
+        user.set_password(password)
+        user.save()
+
+        return CreateUser(user=user)
+
+
+class Mutation(graphene.ObjectType):
+    create_user = CreateUser.Field()

backend/apps/hn/__init__.py

@@ -0,0 +1,6 @@
+from django.contrib import admin
+
+# Register your models here.
+from .models import Link
+
+admin.site.register(Link)

backend/apps/hn/admin.py

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

backend/apps/hn/apps.py

@@ -0,0 +1,22 @@
+# Generated by Django 2.2 on 2020-02-11 04:02
+
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+
+    initial = True
+
+    dependencies = [
+    ]
+
+    operations = [
+        migrations.CreateModel(
+            name='Link',
+            fields=[
+                ('id', models.AutoField(auto_created=True, primary_key=True, serialize=False, verbose_name='ID')),
+                ('url', models.URLField()),
+                ('description', models.TextField(blank=True)),
+            ],
+        ),
+    ]

backend/apps/hn/migrations/0001_initial.py

@@ -0,0 +1,21 @@
+# Generated by Django 2.2 on 2020-02-11 17:06
+
+from django.conf import settings
+from django.db import migrations, models
+import django.db.models.deletion
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        migrations.swappable_dependency(settings.AUTH_USER_MODEL),
+        ('hn', '0001_initial'),
+    ]
+
+    operations = [
+        migrations.AddField(
+            model_name='link',
+            name='posted_by',
+            field=models.ForeignKey(null=True, on_delete=django.db.models.deletion.CASCADE, to=settings.AUTH_USER_MODEL),
+        ),
+    ]

backend/apps/hn/migrations/0002_link_posted_by.py

@@ -0,0 +1,24 @@
+# Generated by Django 2.2 on 2020-02-11 18:51
+
+from django.conf import settings
+from django.db import migrations, models
+import django.db.models.deletion
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        migrations.swappable_dependency(settings.AUTH_USER_MODEL),
+        ('hn', '0002_link_posted_by'),
+    ]
+
+    operations = [
+        migrations.CreateModel(
+            name='Vote',
+            fields=[
+                ('id', models.AutoField(auto_created=True, primary_key=True, serialize=False, verbose_name='ID')),
+                ('link', models.ForeignKey(on_delete=django.db.models.deletion.CASCADE, related_name='votes', to='hn.Link')),
+                ('user', models.ForeignKey(on_delete=django.db.models.deletion.CASCADE, to=settings.AUTH_USER_MODEL)),
+            ],
+        ),
+    ]

backend/apps/hn/migrations/0003_vote.py

@@ -0,0 +1,25 @@
+from django.conf import settings
+from django.db import models
+
+# Create your models here.
+
+class Link(models.Model):
+    url = models.URLField()
+    description = models.TextField(blank=True)
+    posted_by = models.ForeignKey(
+        settings.AUTH_USER_MODEL,
+        null=True,
+        on_delete=models.CASCADE
+    )
+
+
+class Vote(models.Model):
+    user = models.ForeignKey(
+        settings.AUTH_USER_MODEL,
+        on_delete=models.CASCADE
+    )
+    link = models.ForeignKey(
+        Link,
+        related_name="votes",
+        on_delete=models.CASCADE
+    )

backend/apps/hn/migrations/__init__.py

@@ -0,0 +1,101 @@
+import graphene
+from django.db.models import Q
+from graphene_django import DjangoObjectType
+
+from apps.accounts.schema import UserType
+
+from .models import Link, Vote
+
+class LinkType(DjangoObjectType):
+    class Meta:
+        model = Link
+
+class VoteType(DjangoObjectType):
+    class Meta:
+        model = Vote
+
+
+class Query(graphene.ObjectType):
+    links = graphene.List(
+        LinkType,
+        search=graphene.String(),
+        first=graphene.Int(),
+        skip=graphene.Int()
+    )
+    votes = graphene.List(VoteType)
+
+    def resolve_links(self, info, search=None, first=None, skip=None, **kwargs):
+        qs = Link.objects.all()
+        if search:
+            filter = (
+                Q(url__icontains=search) |
+                Q(description__icontains=search)
+            )
+            qs = qs.filter(filter)
+
+        if skip:
+            qs = qs[skip:]
+
+        if first:
+            qs = qs[:first]
+
+        return qs
+
+    def resolve_votes(self, info, **kwargs):
+        return Vote.objects.all()
+
+
+class CreateLink(graphene.Mutation):
+    id = graphene.Int()
+    url = graphene.String()
+    description = graphene.String()
+    posted_by = graphene.Field(UserType)
+
+    class Arguments:
+        url = graphene.String()
+        description = graphene.String()
+
+    def mutate(self, info, url, description):
+        user = info.context.user or None
+        link = Link(
+            url=url,
+            description=description,
+            posted_by=user
+        )
+
+        link.save()
+
+        return CreateLink(
+            id=link.id,
+            url=link.url,
+            description=link.description,
+            posted_by=link.posted_by
+        )
+
+class CreateVote(graphene.Mutation):
+    user = graphene.Field(UserType)
+    link = graphene.Field(LinkType)
+
+    class Arguments:
+        link_id = graphene.Int()
+
+    def mutate(self, info, link_id):
+        user = info.context.user
+        if user.is_anonymous:
+            raise Exception("Must be logged in to vote")
+
+        link = Link.objects.filter(id=link_id).first()
+
+        if not link:
+            raise Exception("Invalid link")
+
+        Vote.objects.create(
+            user=user,
+            link=link
+        )
+
+        return CreateVote(user=user, link=link)
+
+class Mutation(graphene.ObjectType):
+    create_link = CreateLink.Field()
+    create_vote = CreateVote.Field()