Minor changes

2 files changed, 19 insertions, 4 deletions

diff --git a/tropical/db.py b/tropical/db.py
index 15e083b..431fce8 100644
--- a/tropical/db.py
+++ b/tropical/db.py
@@ -1,6 +1,5 @@
 import sqlite3, time, socket
 
-
 class CalDB:
     __DBVERSION__="1"
     
@@ -21,9 +20,9 @@ class CalDB:
         self.cur.execute('INSERT INTO infos VALUES("creation", "'+str(time.time())+'")')
         self.cur.execute('INSERT INTO infos VALUES("created_on", "'+socket.gethostname()+'")')
         # Calendars table
-        self.cur.execute("CREATE TABLE calendars(id INTEGER PRIMARY KEY, name TEXT,  description TEXT, color TEXT)")
+        self.cur.execute("CREATE TABLE calendars(id INTEGER PRIMARY KEY, name TEXT,  description TEXT, color TEXT, type INTEGER)")
         # Events table
-        self.cur.execute("CREATE TABLE events(id INTEGER PRIMARY KEY, name TEXT, calendar INTEGER, description TEXT, start REAL, end REAL, repeat TEXT, frequency INTEGER, FOREIGN KEY(calendar) REFERENCES calendars(id))")
+        self.cur.execute("CREATE TABLE events(id INTEGER PRIMARY KEY, name TEXT, calendar INTEGER, description TEXT, start REAL, end REAL, repeat INTEGER, frequency INTEGER, location INTEGER, FOREIGN KEY(calendar) REFERENCES calendars(id))")
         self.con.commit()
 
     def keyExists(self, db, key):
diff --git a/tropical/env.py b/tropical/env.py
index 3c2bc3b..e2484fd 100644
--- a/tropical/env.py
+++ b/tropical/env.py
@@ -1,8 +1,24 @@
 
-
 from pathlib import Path
 from db import CalDB
 import configparser
+from enum import Enum
+
+class EvtRepeat(Enum):
+    DAYLY = 1
+    WEEKLY = 2
+    MONTHLY = 3
+    YEARLY = 4
+
+class EvtLocation(Enum):
+    LOCAL = 1
+    REMOTE = 2
+    LOCAL_AND_REMOTE = 3
+
+class CalType(Enum):
+    TROPICAL = 1
+    CALDAV = 2
+
 
 class Env: