From 5feeb1e7f6ae24f096d5ae36ba7e6702412f7dcb Mon Sep 17 00:00:00 2001 From: Vassilii Khachaturov Date: Mon, 17 Feb 2014 15:54:04 +0200 Subject: [PATCH] 7117: disallow new year unless Julian or like cal raise an exception from Date.set for a meaningless newyear parameter Add Date.CALENDARS to reduce cut-and-paste here and with the unit test --- gramps/gen/lib/date.py | 17 ++++++++--------- 1 file changed, 8 insertions(+), 9 deletions(-) diff --git a/gramps/gen/lib/date.py b/gramps/gen/lib/date.py index 3de400eb6..8e9aa5ccb 100644 --- a/gramps/gen/lib/date.py +++ b/gramps/gen/lib/date.py @@ -4,7 +4,7 @@ # Copyright (C) 2000-2007 Donald N. Allingham # Copyright (C) 2009 Douglas S. Blank # Copyright (C) 2013 Paul Franklin -# Copyright (C) 2013 Vassilii Khachaturov +# Copyright (C) 2013-2014 Vassilii Khachaturov # # 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 @@ -21,8 +21,6 @@ # Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA # -# $Id$ - """Support for dates.""" #------------------------------------------------------------------------ @@ -540,6 +538,7 @@ class Date(object): CAL_PERSIAN = 4 CAL_ISLAMIC = 5 CAL_SWEDISH = 6 + CALENDARS = range(7) NEWYEAR_JAN1 = 0 # CODE NEWYEAR_MAR1 = 1 @@ -1169,8 +1168,7 @@ class Date(object): """ Set the calendar selected for the date. """ - if val not in (Date.CAL_GREGORIAN, Date.CAL_JULIAN, Date.CAL_HEBREW, - Date.CAL_FRENCH, Date.CAL_PERSIAN, Date.CAL_ISLAMIC, Date.CAL_SWEDISH): + if val not in Date.CALENDARS: raise DateError("Invalid calendar") self.calendar = val @@ -1583,11 +1581,12 @@ class Date(object): if quality not in (Date.QUAL_NONE, Date.QUAL_ESTIMATED, Date.QUAL_CALCULATED): raise DateError("Invalid quality") - if calendar not in (Date.CAL_GREGORIAN, Date.CAL_JULIAN, - Date.CAL_HEBREW, Date.CAL_FRENCH, - Date.CAL_PERSIAN, Date.CAL_ISLAMIC, - Date.CAL_SWEDISH): + if calendar not in Date.CALENDARS: raise DateError("Invalid calendar") + if newyear != 0 and calendar_has_fixed_newyear(calendar): + raise DateError( + "May not adjust newyear to {ny} for calendar {cal}".format( + ny = newyear, cal = calendar)) self.quality = quality self.modifier = modifier