From 86cc564d271110ceaa7825aba95a56cd4b1b816e Mon Sep 17 00:00:00 2001
From: Nam Tran <thnam.me@gmail.com>
Date: Mon, 14 Sep 2020 10:38:59 -0500
Subject: [PATCH] start database

---
 sites/WeatherStation/.gitignore                 |  1 +
 sites/WeatherStation/HomeWeather/admin.py       |  4 +++-
 sites/WeatherStation/HomeWeather/models.py      | 15 +++++++++++++++
 sites/WeatherStation/WeatherStation/settings.py |  1 +
 4 files changed, 20 insertions(+), 1 deletion(-)

diff --git a/sites/WeatherStation/.gitignore b/sites/WeatherStation/.gitignore
index a0491fb..6916bec 100644
--- a/sites/WeatherStation/.gitignore
+++ b/sites/WeatherStation/.gitignore
@@ -1,2 +1,3 @@
 db.sqlite3
 */__pycache__
+*/migrations
diff --git a/sites/WeatherStation/HomeWeather/admin.py b/sites/WeatherStation/HomeWeather/admin.py
index 8c38f3f..ffee64b 100644
--- a/sites/WeatherStation/HomeWeather/admin.py
+++ b/sites/WeatherStation/HomeWeather/admin.py
@@ -1,3 +1,5 @@
 from django.contrib import admin
-
+from .models import Sensor, Measurement
 # Register your models here.
+admin.site.register(Sensor)
+admin.site.register(Measurement)
diff --git a/sites/WeatherStation/HomeWeather/models.py b/sites/WeatherStation/HomeWeather/models.py
index 71a8362..a7a7f36 100644
--- a/sites/WeatherStation/HomeWeather/models.py
+++ b/sites/WeatherStation/HomeWeather/models.py
@@ -1,3 +1,18 @@
 from django.db import models
 
 # Create your models here.
+class Sensor(models.Model):
+    name = models.CharField(max_length=20, default="bedroom-bme280-01")
+    temperature_unit = models.CharField(max_length=5, default="C")
+    pressure_unit = models.CharField(max_length=5, default="hPa")
+    humidity_unit = models.CharField(max_length=5, default="%")
+    def __str__(self):
+        return f"Sensor {self.name}, units are C (temperature), hPa (pressure) and % (humidity)"
+
+class Measurement(models.Model):
+    time = models.DateTimeField('timestamp', auto_now=True)
+    temperature = models.FloatField(default=-400)
+    humidity = models.FloatField(default=-1)
+    pressure = models.FloatField(default=-1)
+    def __str__(self):
+        return f"{self.time}, {self.temperature} C, {self.pressure} hPa, {self.humidity} %"
diff --git a/sites/WeatherStation/WeatherStation/settings.py b/sites/WeatherStation/WeatherStation/settings.py
index b5d49a4..7b75fb1 100644
--- a/sites/WeatherStation/WeatherStation/settings.py
+++ b/sites/WeatherStation/WeatherStation/settings.py
@@ -31,6 +31,7 @@ ALLOWED_HOSTS = []
 # Application definition
 
 INSTALLED_APPS = [
+    'HomeWeather.apps.HomeweatherConfig',
     'django.contrib.admin',
     'django.contrib.auth',
     'django.contrib.contenttypes',