Package Gnumed :: Package timelinelib :: Package repositories :: Module dbwrapper |
[frames] | no frames]

# Source Code for Module Gnumed.timelinelib.repositories.dbwrapper

     1  # Copyright (C) 2009, 2010, 2011  Rickard Lindberg, Roger Lindberg
     2  #
     3  # This file is part of Timeline.
     4  #
     5  # Timeline is free software: you can redistribute it and/or modify
     6  # it under the terms of the GNU General Public License as published by
     7  # the Free Software Foundation, either version 3 of the License, or
     8  # (at your option) any later version.
     9  #
    10  # Timeline is distributed in the hope that it will be useful,
    11  # but WITHOUT ANY WARRANTY; without even the implied warranty of
    12  # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
    13  # GNU General Public License for more details.
    14  #
    15  # You should have received a copy of the GNU General Public License
    16  # along with Timeline.  If not, see <http://www.gnu.org/licenses/>.
    17
    18
    19  from timelinelib.repositories.interface import CategoryRepository
    20  from timelinelib.repositories.interface import EventRepository
    21  from timelinelib.wxgui.utils import category_tree
    22
    23
    24 -class DbWrapperCategoryRepository(CategoryRepository):
    25
    26 -    def __init__(self, db):
    27          self.db = db
    28
    29 -    def get_all(self):
    30          return self.db.get_categories()
    31
    32 -    def get_tree(self, remove):
    33          return category_tree(self.get_all(), remove=remove)
    34
    35 -    def save(self, category):
    36          self.db.save_category(category)
    37
    38
    39 -class DbWrapperEventRepository(EventRepository):
    40
    41 -    def __init__(self, db):
    42          self.db = db
    43
    44 -    def save(self, event):
    45          self.db.save_event(event)
    46