From 89994b13070e66a9325b4b70ae37eb91c5879e46 Mon Sep 17 00:00:00 2001
From: Michiel Scholten <michiel@diginaut.net>
Date: Thu, 28 Feb 2019 16:07:04 +0100
Subject: [PATCH] Fixes for themes; import models

---
 digimarks/marks.py  | 15 +++++++++------
 digimarks/models.py |  2 +-
 2 files changed, 10 insertions(+), 7 deletions(-)

diff --git a/digimarks/marks.py b/digimarks/marks.py
index 195b9ce..b07fdd1 100644
--- a/digimarks/marks.py
+++ b/digimarks/marks.py
@@ -8,7 +8,10 @@ from flask import (Flask, abort, jsonify, redirect, render_template, request,
                    url_for)
 from werkzeug.contrib.atom import AtomFeed
 
-DIGIMARKS_USER_AGENT = 'digimarks/1.2.0-dev'
+from . import themes
+from .models import Bookmark, PublicTag, User, urljoin, get_tags_for_user
+
+DIGIMARKS_USER_AGENT = 'digimarks/2.0.0-dev'
 
 try:
     import settings
@@ -58,9 +61,9 @@ def get_cached_tags(userkey):
 def get_theme(userkey):
     try:
         usertheme = usersettings[userkey]['theme']
-        return themes[usertheme]
+        return themes.themes[usertheme]
     except KeyError:
-        return themes[DEFAULT_THEME]  # default
+        return themes.themes[themes.DEFAULT_THEME]  # default
 
 
 def make_external(url):
@@ -81,14 +84,14 @@ def _find_bookmarks(userkey, filter_text):
 
 @app.errorhandler(404)
 def page_not_found(e):
-    theme = themes[DEFAULT_THEME]
+    theme = themes.themes[themes.DEFAULT_THEME]
     return render_template('404.html', error=e, theme=theme), 404
 
 
 @app.route('/')
 def index():
     """ Homepage, point visitors to project page """
-    theme = themes[DEFAULT_THEME]
+    theme = themes.themes[themes.DEFAULT_THEME]
     return render_template('index.html', theme=theme)
 
 
@@ -487,7 +490,7 @@ def publictag_page(tagkey):
     #this_tag = get_object_or_404(PublicTag.select().where(PublicTag.tagkey == tagkey))
     try:
         this_tag, bookmarks = get_publictag(tagkey)
-        theme = themes[DEFAULT_THEME]
+        theme = themes.themes[themes.DEFAULT_THEME]
         return render_template(
             'publicbookmarks.html',
             bookmarks=bookmarks,
diff --git a/digimarks/models.py b/digimarks/models.py
index 9c123ce..424988b 100644
--- a/digimarks/models.py
+++ b/digimarks/models.py
@@ -87,7 +87,7 @@ class User(BaseModel):
     """ User account """
     username = CharField()
     key = CharField()
-    theme = CharField(default=DEFAULT_THEME)
+    theme = CharField(default=themes.DEFAULT_THEME)
     created_date = DateTimeField(default=datetime.datetime.now)
 
     def generate_key(self):