Add a scheduler that updates UI when ...

authors in the database is changed. Also add
a REST method to get number of authors (to check
if they are changed) and a management command
to add a new author.

Author: spapas

books/__init__.py books/management/__init__.pybooks/__init__.py renamed to books/management/__init__.py

@@ -0,0 +1,14 @@
+from django.core.management.base import BaseCommand, CommandError
+from books.models import Author
+import random, string
+
+class Command(BaseCommand):
+    help = 'Closes the specified poll for voting'
+
+
+    def handle(self, *args, **options):
+        author = Author.objects.create(
+            first_name=random.choice(string.ascii_uppercase) + ''.join(random.choice(string.ascii_lowercase) for _ in range(7)),
+            last_name=random.choice(string.ascii_uppercase) + ''.join(random.choice(string.ascii_lowercase) for _ in range(7)),
+        )
+        print "Author {} {} added ".format(author.first_name, author.last_name)

books/management/commands/__init__.py

@@ -1,7 +1,9 @@
 from django.shortcuts import render
 from django.views.generic import TemplateView
 from rest_framework import viewsets, filters
+from rest_framework.decorators import detail_route, list_route
 from rest_framework.pagination import PageNumberPagination
+from rest_framework.response import Response
 from books.models import Book, Category, SubCategory, Author
 from books.serializers import BookSerializer, CategorySerializer, SubCategorySerializer, AuthorSerializer
 
@@ -35,6 +37,12 @@ class AuthorViewSet(viewsets.ModelViewSet):
     search_fields = ('last_name', 'first_name', )
     ordering_fields = ('id', 'last_name', )
     ordering = ('last_name',)
+    
+    @list_route()
+    def get_author_number(self, request):
+        return Response(Author.objects.all().count())
+    
+    
 
 class SubCategoryViewSet(viewsets.ModelViewSet):
     queryset = SubCategory.objects.all()

books/management/commands/add_author.py

@@ -7,6 +7,7 @@ import LoadingContainer from './loading';
 import StatPanel from './StatPanel'
 import { bindActionCreators } from 'redux'
 
+
 class App extends React.Component {
 
     render() {

books/views.py

@@ -5,6 +5,7 @@ import { hideNotification } from '../actions'
 
 import * as colors from '../util/colors'
 
+
 const NotificationContainer = (props) => {
     let { message, notification_type } = props.notification;
     let { onHide } = props;

srcjs/components/app.js

@@ -13,6 +13,10 @@ import AuthorPanel from './components/AuthorPanel';
 import BookForm from './components/BookForm';
 import AuthorForm from './components/AuthorForm';
 
+import schedule from './scheduler';
+
+
+
 const About = () => {
     return <div>
         <h2>About</h2>
@@ -48,4 +52,4 @@ render((
 )
 
 
-
+schedule();

srcjs/components/notification.js

@@ -0,0 +1,19 @@
+import store from './store'
+import {loadAuthors, showSuccessNotification} from './actions';
+
+
+export default () => {
+    console.log("Starting scheduler")
+    window.setInterval( () => {
+        
+        let url = 'http://127.0.0.1:8000/api/authors/get_author_number/';
+        $.get(url, realAuthorNumber => {
+            let authorNumber = store.getState().authors.rows.length;
+            if(authorNumber!=realAuthorNumber) {
+                store.dispatch(showSuccessNotification("Authors have changed - reloading ..."));
+                store.dispatch(loadAuthors());
+            }
+        });
+        
+    }, 2000);
+};