diff options
Diffstat (limited to 'src/web/models')
| -rw-r--r-- | src/web/models/__init__.py | 8 | ||||
| -rw-r--r-- | src/web/models/article.py | 10 | ||||
| -rw-r--r-- | src/web/models/bookmark.py | 75 | ||||
| -rw-r--r-- | src/web/models/tag.py | 30 | ||||
| -rw-r--r-- | src/web/models/user.py | 4 |
5 files changed, 13 insertions, 114 deletions
diff --git a/src/web/models/__init__.py b/src/web/models/__init__.py index 229a99e5..1fc0c3e0 100644 --- a/src/web/models/__init__.py +++ b/src/web/models/__init__.py @@ -32,13 +32,9 @@ from .user import User from .article import Article from .icon import Icon from .category import Category -#from .tag import Tag -from .tag import BookmarkTag -from .tag import ArticleTag -from .bookmark import Bookmark +from .tag import Tag -__all__ = ['Feed', 'Role', 'User', 'Article', 'Icon', 'Category', 'Tag', - 'Bookmark', 'ArticleTag', 'BookmarkTag'] +__all__ = ['Feed', 'Role', 'User', 'Article', 'Icon', 'Category', 'Tag'] import os diff --git a/src/web/models/article.py b/src/web/models/article.py index 6245b10e..87ad1a9a 100644 --- a/src/web/models/article.py +++ b/src/web/models/article.py @@ -34,8 +34,6 @@ from sqlalchemy.ext.associationproxy import association_proxy from web.models.right_mixin import RightMixin - - class Article(db.Model, RightMixin): "Represent an article from a feed." id = db.Column(db.Integer(), primary_key=True) @@ -54,10 +52,10 @@ class Article(db.Model, RightMixin): category_id = db.Column(db.Integer(), db.ForeignKey('category.id')) # relationships - tag_objs = db.relationship('ArticleTag', back_populates='article', - cascade='all,delete-orphan', - lazy=False, - foreign_keys='[ArticleTag.article_id]') + tag_objs = db.relationship('Tag', back_populates='article', + cascade='all,delete-orphan', + lazy=False, + foreign_keys='[Tag.article_id]') tags = association_proxy('tag_objs', 'text') # index diff --git a/src/web/models/bookmark.py b/src/web/models/bookmark.py deleted file mode 100644 index 6101a5f0..00000000 --- a/src/web/models/bookmark.py +++ /dev/null @@ -1,75 +0,0 @@ -#! /usr/bin/env python -# -*- coding: utf-8 -*- - -# Newspipe - A Web based news aggregator. -# Copyright (C) 2010-2016 Cédric Bonhomme - https://www.cedricbonhomme.org -# -# For more information : https://github.com/newspipe/newspipe -# -# This program is free software: you can redistribute it and/or modify -# it under the terms of the GNU Affero General Public License as -# published by the Free Software Foundation, either version 3 of the -# License, or (at your option) any later version. -# -# This program is distributed in the hope that it will be useful, -# but WITHOUT ANY WARRANTY; without even the implied warranty of -# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the -# GNU Affero General Public License for more details. -# -# You should have received a copy of the GNU Affero General Public License -# along with this program. If not, see <http://www.gnu.org/licenses/>. - -__author__ = "Cedric Bonhomme" -__version__ = "$Revision: 0.1 $" -__date__ = "$Date: 2016/12/07 $" -__revision__ = "$Date: 2016/12/07 $" -__copyright__ = "Copyright (c) Cedric Bonhomme" -__license__ = "GPLv3" - -from bootstrap import db -from datetime import datetime -from sqlalchemy.orm import validates -from sqlalchemy.ext.associationproxy import association_proxy - -from web.models.right_mixin import RightMixin - - -bookmarktags_table = db.Table('bookmarktags', db.metadata, - db.Column('bookmark_id', db.Integer, db.ForeignKey("bookmark.id"), - primary_key=True), - db.Column('tag_text', db.String, db.ForeignKey("BookmarkTag.text"), - primary_key=True) -) - -class Bookmark(db.Model, RightMixin): - """ - Represent a bookmark. - """ - id = db.Column(db.Integer(), primary_key=True) - href = db.Column(db.String(), default="") - title = db.Column(db.String(), default="") - description = db.Column(db.String(), default="FR") - shared = db.Column(db.Boolean(), default=False) - to_read = db.Column(db.Boolean(), default=False) - time = db.Column(db.DateTime(), default=datetime.utcnow) - user_id = db.Column(db.Integer(), db.ForeignKey('user.id')) - - # relationships - tag_objs = db.relationship("BookmarkTag", secondary=lambda: bookmarktags_table) - # tag_objs = db.relationship('BookmarkTag', back_populates='bookmark', - # cascade='all,delete-orphan', - # lazy=False, - # foreign_keys='[BookmarkTag.bookmark_id]') - tags = association_proxy('tag_objs', 'text') - - - @validates('description') - def validates_title(self, key, value): - return str(value).strip() - - @validates('extended') - def validates_description(self, key, value): - return str(value).strip() - - def __repr__(self): - return '<Bookmark %r>' % (self.href) diff --git a/src/web/models/tag.py b/src/web/models/tag.py index c23d15ea..8d7fe4d4 100644 --- a/src/web/models/tag.py +++ b/src/web/models/tag.py @@ -1,38 +1,22 @@ #! /usr/bin/env python # -*- coding: utf-8 -*- +from sqlalchemy import Column, ForeignKey, Integer, String +from sqlalchemy.orm import relationship + from bootstrap import db -class ArticleTag(db.Model): - text = db.Column(db.String, primary_key=True, unique=False) +class Tag(db.Model): + text = Column(String, primary_key=True, unique=False) # foreign keys - article_id = db.Column(db.Integer, db.ForeignKey('article.id', ondelete='CASCADE'), + article_id = Column(Integer, ForeignKey('article.id', ondelete='CASCADE'), primary_key=True) # relationships - article = db.relationship('Article', back_populates='tag_objs', + article = relationship('Article', back_populates='tag_objs', foreign_keys=[article_id]) def __init__(self, text): self.text = text - - -class BookmarkTag(db.Model): - __tablename__ = 'BookmarkTag' - #id = db.Column(db.Integer, primary_key=True) - text = db.Column(db.String, primary_key=True) - - # foreign keys - # bookmark_id = db.Column(db.Integer, db.ForeignKey('bookmark.id', ondelete='CASCADE'), - # primary_key=True) - - # relationships - # bookmark = db.relationship('Bookmark', back_populates='tag_objs', - # single_parent=True, - # cascade="all, delete-orphan", - # foreign_keys=[bookmark_id]) - - def __init__(self, text): - self.text = text diff --git a/src/web/models/user.py b/src/web/models/user.py index d2b3c52a..460958e0 100644 --- a/src/web/models/user.py +++ b/src/web/models/user.py @@ -38,7 +38,6 @@ from bootstrap import db from web.models.right_mixin import RightMixin from web.models.category import Category from web.models.feed import Feed -from web.models.bookmark import Bookmark class User(db.Model, UserMixin, RightMixin): @@ -72,9 +71,6 @@ class User(db.Model, UserMixin, RightMixin): feeds = db.relationship('Feed', backref='user', cascade='all, delete-orphan', foreign_keys=[Feed.user_id]) - bookmarks = db.relationship('Bookmark', backref='user', - cascade='all, delete-orphan', - foreign_keys=[Bookmark.user_id]) @staticmethod def _fields_base_write(): |