gramps/src/DisplayModels/_NoteModel.py

127 lines
4.2 KiB
Python
Raw Normal View History

2007-02-13 23:26:31 +05:30
#
# Gramps - a GTK+/GNOME based genealogy program
#
# Copyright (C) 2000-2007 Donald N. Allingham
2007-02-13 23:26:31 +05:30
#
# This program is free software; you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation; either version 2 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 General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program; if not, write to the Free Software
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
#
# $Id$
2007-02-13 23:26:31 +05:30
#-------------------------------------------------------------------------
#
# python modules
#
#-------------------------------------------------------------------------
import logging
_LOG = logging.getLogger(".DisplayModels.NoteModel")
2007-02-13 23:26:31 +05:30
#-------------------------------------------------------------------------
#
# GNOME/GTK modules
#
#-------------------------------------------------------------------------
import gtk
#-------------------------------------------------------------------------
#
# GRAMPS modules
#
#-------------------------------------------------------------------------
from _BaseModel import BaseModel
from gen.lib import (Note, NoteType, MarkerType, StyledText)
2007-02-13 23:26:31 +05:30
#-------------------------------------------------------------------------
#
# NoteModel
2007-02-13 23:26:31 +05:30
#
#-------------------------------------------------------------------------
class NoteModel(BaseModel):
"""
"""
def __init__(self, db, scol=0, order=gtk.SORT_ASCENDING, search=None,
2007-02-13 23:26:31 +05:30
skip=set(), sort_map=None):
"""Setup initial values for instance variables."""
self.gen_cursor = db.get_note_cursor
self.map = db.get_raw_note_data
2007-02-13 23:26:31 +05:30
self.fmap = [
self.column_preview,
2007-02-13 23:26:31 +05:30
self.column_id,
self.column_type,
self.column_marker,
2007-02-13 23:26:31 +05:30
self.column_handle,
self.column_marker_color
]
2007-02-13 23:26:31 +05:30
self.smap = [
self.column_preview,
2007-02-13 23:26:31 +05:30
self.column_id,
self.column_type,
self.column_marker,
2007-02-13 23:26:31 +05:30
self.column_handle,
self.column_marker_color
]
self.marker_color_column = 5
BaseModel.__init__(self, db, scol, order, search=search,
skip=skip, sort_map=sort_map)
2007-02-13 23:26:31 +05:30
def on_get_n_columns(self):
"""Return the column number of the Note tab."""
return len(self.fmap) + 1
2007-02-13 23:26:31 +05:30
def column_handle(self, data):
"""Return the handle of the Note."""
return data[Note.POS_HANDLE]
2007-02-13 23:26:31 +05:30
def column_id(self, data):
"""Return the id of the Note."""
return unicode(data[Note.POS_ID])
2007-02-13 23:26:31 +05:30
def column_type(self, data):
"""Return the type of the Note in readable format."""
temp = NoteType()
temp.set(data[Note.POS_TYPE])
return unicode(str(temp))
2007-02-13 23:26:31 +05:30
def column_marker(self, data):
"""Return the marker type of the Note in readable format."""
temp = MarkerType()
temp.set(data[Note.POS_MARKER])
return unicode(str(temp))
def column_preview(self, data):
"""Return a shortend version of the Note's text."""
#data is the encoding in the database, make it a unicode object
#for universal work
note = unicode(data[Note.POS_TEXT][StyledText.POS_TEXT])
note = " ".join(note.split())
if len(note) > 80:
return note[:80] + "..."
else:
return note
2007-02-13 23:26:31 +05:30
def column_marker_color(self, data):
"""Return the color of the Note's marker type if exist."""
try:
col = data[Note.POS_MARKER][MarkerType.POS_VALUE]
if col == MarkerType.COMPLETE:
return self.complete_color
elif col == MarkerType.TODO_TYPE:
return self.todo_color
elif col == MarkerType.CUSTOM:
return self.custom_color
else:
return None
except IndexError:
return None