bloomhunt/mobile/views.py

from django.http import HttpResponse, HttpResponseRedirect
from django.shortcuts import render

from datetime import datetime
import json
import os

import pandas as pd
import matplotlib.pyplot as plt
import numpy as np
import seaborn as sns
import re as re
from .predict import predict_blooming


def index(request):
    soon_blooms = predict_blooming('20.05.2019')
    return render(request, 'mobile/index.html', {'prediction': soon_blooms})


def scan(request):
    question = q1
    if request.POST:
        data = request.POST
        print("data")
        if data["question"]:
            print("response to question", data["question"])
            for q in questions:
                if not q.key == data["question"]:
                    continue
                for opt in q.options:
                    if not opt.text == data["chosen_option"]:
                        continue
                    if opt.question:
                        question = opt.question
                        break
                    else:
                        return scanned(request, opt.result)
    # first question
    context = {'question': question}
    return render(request, 'mobile/scan.html', context)


def scanned(request, tree):
    context = {'tree': {'name': tree}}
    return render(request, 'mobile/scanned.html', context)


def addtree(request):
    if not request.POST:
        return HttpResponseRedirect("/mobile/scan")
    name = request.POST["name"]
    tree = request.POST["name"]
    longitude = request.POST["longitude"]
    latitude = request.POST["latitude"]
    bloom_start = datetime.now().timetuple().tm_yday
    if not os.path.isfile('/home/areion/tmp/data.json'):
        data = {'trees': []}
    else:
        with open('/home/areion/tmp/data.json', 'r') as f:
            data = json.load(f)

    with open('/home/areion/tmp/data.json', 'w+') as f:
        data["trees"].append(
            {'name': name, 'species': tree,
             'longitude': longitude, 'latitude': latitude,
             'bloom_start': bloom_start }
        )
        json.dump(data, f, indent=4)
    return render(request, "mobile/thankyou.html")


class Option:
    def __init__(self, text, question=None, result=None):
        self.text = text
        self.question = question
        self.result = result


class Question:
    def __init__(self, key, text, options):
        self.key = key
        self.text = text
        self.options = options


q3 = Question("alive", "Is your tree alive?",
              [Option("yes", result="Poor Apple Tree"),
               Option("no", result="Dead Apple Tree")])
q2 = Question("red", "Is your tree red?",
              [Option("yes", result="Red Apple Tree"),
               Option("no", result="Green Apple Tree")])
q1 = Question("leaves", "Does your tree have leaves?",
              [Option("yes", question=q2), Option("no", question=q3)])

questions = [q1, q2, q3]