File: database.py

package info (click to toggle)
python-werkzeug 1.0.1%2Bdfsg1-2%2Bdeb11u1
  • links: PTS, VCS
  • area: main
  • in suites: bullseye
  • size: 2,888 kB
  • sloc: python: 21,897; javascript: 173; makefile: 36; xml: 16
file content (80 lines) | stat: -rw-r--r-- 2,012 bytes parent folder | download
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
# -*- coding: utf-8 -*-
"""
    plnt.database
    ~~~~~~~~~~~~~

    The database definitions for the planet.

    :copyright: 2007 Pallets
    :license: BSD-3-Clause
"""
from sqlalchemy import Column
from sqlalchemy import DateTime
from sqlalchemy import ForeignKey
from sqlalchemy import Integer
from sqlalchemy import MetaData
from sqlalchemy import String
from sqlalchemy import Table
from sqlalchemy.orm import create_session
from sqlalchemy.orm import dynamic_loader
from sqlalchemy.orm import mapper
from sqlalchemy.orm import scoped_session

from .utils import application
from .utils import local_manager


def new_db_session():
    return create_session(application.database_engine, autoflush=True, autocommit=False)


metadata = MetaData()
session = scoped_session(new_db_session, local_manager.get_ident)


blog_table = Table(
    "blogs",
    metadata,
    Column("id", Integer, primary_key=True),
    Column("name", String(120)),
    Column("description", String),
    Column("url", String(200)),
    Column("feed_url", String(250)),
)

entry_table = Table(
    "entries",
    metadata,
    Column("id", Integer, primary_key=True),
    Column("blog_id", Integer, ForeignKey("blogs.id")),
    Column("guid", String(200), unique=True),
    Column("title", String(140)),
    Column("url", String(200)),
    Column("text", String),
    Column("pub_date", DateTime),
    Column("last_update", DateTime),
)


class Blog(object):
    query = session.query_property()

    def __init__(self, name, url, feed_url, description=u""):
        self.name = name
        self.url = url
        self.feed_url = feed_url
        self.description = description

    def __repr__(self):
        return "<%s %r>" % (self.__class__.__name__, self.url)


class Entry(object):
    query = session.query_property()

    def __repr__(self):
        return "<%s %r>" % (self.__class__.__name__, self.guid)


mapper(Entry, entry_table)
mapper(Blog, blog_table, properties=dict(entries=dynamic_loader(Entry, backref="blog")))