Removed reliance on strptime/strftime

The code now uses Boost's input and output facets for times and dates.
This ensures completely consistency regarding timezones and times, and
fixes the regression test that was broken while I was away coding in
London (where it was GMT-0 and I didn't notice the difference between
local and GMT).
This commit is contained in:
John Wiegley 2009-10-11 05:19:01 -04:00
parent 1a8e835bfe
commit f161aea8ce
9 changed files with 122 additions and 126 deletions

View file

@ -114,7 +114,7 @@ void item_t::set_tag(const string& tag,
assert(result.second);
}
void item_t::parse_tags(const char * p, int current_year)
void item_t::parse_tags(const char * p, optional<date_t::year_type> current_year)
{
if (const char * b = std::strchr(p, '[')) {
if (const char * e = std::strchr(p, ']')) {
@ -160,7 +160,7 @@ void item_t::parse_tags(const char * p, int current_year)
}
}
void item_t::append_note(const char * p, int current_year)
void item_t::append_note(const char * p, optional<date_t::year_type> current_year)
{
if (note) {
*note += '\n';

View file

@ -129,8 +129,10 @@ public:
virtual void set_tag(const string& tag,
const optional<string>& value = none);
virtual void parse_tags(const char * p, int current_year = -1);
virtual void append_note(const char * p, int current_year = -1);
virtual void parse_tags(const char * p,
optional<date_t::year_type> current_year = none);
virtual void append_note(const char * p,
optional<date_t::year_type> current_year = none);
static bool use_effective_date;

View file

@ -67,7 +67,7 @@ class session_t : public symbol_scope_t
public:
bool flush_on_next_data_file;
int current_year;
date_t::year_type current_year;
shared_ptr<commodity_pool_t> commodity_pool;
scoped_ptr<account_t> master;

View file

@ -53,6 +53,7 @@
#include <algorithm>
#include <exception>
#include <typeinfo>
#include <locale>
#include <stdexcept>
#include <iostream>
#include <streambuf>
@ -98,7 +99,6 @@ typedef std::ostream::pos_type ostream_pos_type;
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <ctime>
#include <csignal>
#if defined __FreeBSD__ && __FreeBSD__ <= 4
@ -144,6 +144,8 @@ typedef std::ostream::pos_type ostream_pos_type;
#include <boost/cast.hpp>
#include <boost/current_function.hpp>
#include <boost/date_time/posix_time/posix_time.hpp>
#include <boost/date_time/posix_time/posix_time_io.hpp>
#include <boost/date_time/gregorian/gregorian_io.hpp>
#include <boost/filesystem/convenience.hpp>
#include <boost/filesystem/exception.hpp>
#include <boost/filesystem/fstream.hpp>

View file

@ -65,7 +65,6 @@ namespace {
account_t * master;
const path * original_file;
accounts_map account_aliases;
int current_year;
bool strict;
path pathname;
@ -76,6 +75,8 @@ namespace {
std::size_t count;
std::size_t errors;
optional<date_t::year_type> current_year;
scoped_ptr<auto_xact_finalizer_t> auto_xact_finalizer;
instance_t(std::list<account_t *>& _account_stack,
@ -476,7 +477,7 @@ void instance_t::nomarket_directive(char * line)
void instance_t::year_directive(char * line)
{
current_year = lexical_cast<int>(skip_ws(line + 1));
current_year = lexical_cast<unsigned short>(skip_ws(line + 1));
}
void instance_t::option_directive(char * line)

View file

@ -41,64 +41,83 @@ std::string output_datetime_format = "%Y-%m-%d %H:%M:%S";
std::string output_date_format = "%Y-%m-%d";
namespace {
const char * formats[] = {
"%y/%m/%d",
"%Y/%m/%d",
"%m/%d",
"%Y/%m",
"%y.%m.%d",
"%Y.%m.%d",
"%m.%d",
"%Y.%m",
"%y-%m-%d",
"%Y-%m-%d",
"%m-%d",
"%Y-%m",
NULL
struct date_format_t {
const char * format;
bool has_year;
date_format_t(const char * _format, bool _has_year)
: format(_format), has_year(_has_year) {}
};
bool parse_date_mask(const char * date_str, std::tm& result)
const date_format_t formats[] = {
date_format_t("%m/%d", false),
date_format_t("%Y/%m/%d", true),
date_format_t("%Y/%m", true),
date_format_t("%y/%m/%d", true),
date_format_t("%m.%d", false),
date_format_t("%Y.%m.%d", true),
date_format_t("%Y.%m", true),
date_format_t("%y.%m.%d", true),
date_format_t("%m-%d", false),
date_format_t("%Y-%m-%d", true),
date_format_t("%Y-%m", true),
date_format_t("%y-%m-%d", true)
};
date_t parse_date_mask_routine(const char * date_str, const date_format_t& df,
optional<date_t::year_type> year, bool& saw_year)
{
std::string str(date_str);
gregorian::date_input_facet * facet(new gregorian::date_input_facet(df.format));
std::istringstream sstr(str);
sstr.imbue(std::locale(sstr.getloc(), facet));
date_t when;
sstr >> when;
if (! when.is_not_a_date()) {
if (sstr.good() && ! sstr.eof() && sstr.peek() != EOF)
return date_t();
DEBUG("times.parse", "Parsed date string: " << date_str);
DEBUG("times.parse", "Parsed result is: " << when);
DEBUG("times.parse", "Format used was: " << df.format);
if (! df.has_year) {
saw_year = false;
when = date_t(year ? *year : CURRENT_DATE().year(),
when.month(), when.day());
if (when.month() > CURRENT_DATE().month())
when -= gregorian::years(1);
} else {
saw_year = true;
}
}
return when;
}
date_t parse_date_mask(const char * date_str, optional<date_t::year_type> year,
bool& saw_year)
{
if (input_date_format) {
std::memset(&result, -1, sizeof(std::tm));
if (strptime(date_str, input_date_format->c_str(), &result))
return true;
}
for (const char ** f = formats; *f; f++) {
std::memset(&result, -1, sizeof(std::tm));
if (strptime(date_str, *f, &result))
return true;
}
return false;
date_format_t df(input_date_format->c_str(), true);
if (! icontains(*input_date_format, "%y"))
df.has_year = false;
date_t when = parse_date_mask_routine(date_str, df, year, saw_year);
if (! when.is_not_a_date())
return when;
}
bool quick_parse_date(const char * date_str, std::tm& result, const int year)
{
if (! parse_date_mask(date_str, result))
return false;
result.tm_hour = 0;
result.tm_min = 0;
result.tm_sec = 0;
if (result.tm_mday == -1)
result.tm_mday = 1;
if (result.tm_mon == -1) {
result.tm_mon = 0;
if (result.tm_mday > (CURRENT_DATE().day() - 1))
result.tm_mon = 11;
for (uint8_t i = 0; i < (sizeof(formats) / sizeof(date_format_t)); i++) {
date_t when = parse_date_mask_routine(date_str, formats[i], year,
saw_year);
if (! when.is_not_a_date())
return when;
}
if (result.tm_year == -1) {
result.tm_year = (year == -1 ? int(CURRENT_DATE().year()) : year) - 1900;
if (year == -1 && result.tm_mon > (CURRENT_DATE().month() - 1))
result.tm_year--;
}
return true;
return date_t();
}
}
@ -153,21 +172,24 @@ string_to_month_of_year(const std::string& str)
return none;
}
datetime_t parse_datetime(const char * str, int)
datetime_t parse_datetime(const char * str, optional<date_t::year_type>)
{
std::tm when;
std::memset(&when, -1, sizeof(std::tm));
if (strptime(str, "%Y/%m/%d %H:%M:%S", &when))
return posix_time::ptime_from_tm(when);
else
return datetime_t();
posix_time::time_input_facet * facet
(new posix_time::time_input_facet("%Y/%m/%d %H:%M:%S"));
std::string temp(str);
std::istringstream sstr(temp);
sstr.imbue(std::locale(sstr.getloc(), facet));
datetime_t when;
sstr >> when;
return when;
}
date_t parse_date(const char * str, int current_year)
date_t parse_date(const char * str, optional<date_t::year_type> current_year)
{
std::tm when;
quick_parse_date(str, when, current_year);
return gregorian::date_from_tm(when);
bool saw_year;
return parse_date_mask(str, current_year, saw_year);
}
date_t date_interval_t::add_duration(const date_t& date,
@ -436,52 +458,24 @@ namespace {
date_t * begin,
date_t * end)
{
struct std::tm when;
bool saw_year = true;
date_t when = parse_date_mask(word.c_str(), none, saw_year);
if (! parse_date_mask(word.c_str(), when))
if (when.is_not_a_date())
throw_(date_error, _("Could not parse date mask: %1") << word);
when.tm_hour = 0;
when.tm_min = 0;
when.tm_sec = 0;
when.tm_isdst = -1;
bool saw_year = true;
bool saw_mon = true;
bool saw_day = true;
if (when.tm_year == -1) {
when.tm_year = CURRENT_DATE().year() - 1900;
saw_year = false;
}
if (when.tm_mon == -1) {
when.tm_mon = 0;
saw_mon = false;
} else {
saw_year = false; // don't increment by year if month used
}
if (when.tm_mday == -1) {
when.tm_mday = 1;
saw_day = false;
} else {
saw_mon = false; // don't increment by month if day used
saw_year = false; // don't increment by year if day used
}
if (begin) {
*begin = gregorian::date_from_tm(when);
*begin = when;
if (end) {
if (saw_year)
*end = *begin + gregorian::years(1);
else if (saw_mon)
else
*end = *begin + gregorian::months(1);
else if (saw_day)
*end = *begin + gregorian::days(1);
}
}
else if (end) {
*end = gregorian::date_from_tm(when);
*end = when;
}
}

View file

@ -81,39 +81,34 @@ string_to_day_of_week(const std::string& str);
optional<date_time::months_of_year>
string_to_month_of_year(const std::string& str);
datetime_t parse_datetime(const char * str, int current_year = -1);
datetime_t parse_datetime(const char * str,
optional<date_t::year_type> current_year = none);
inline datetime_t parse_datetime(const std::string& str,
int current_year = -1) {
optional<date_t::year_type> current_year = none) {
return parse_datetime(str.c_str(), current_year);
}
date_t parse_date(const char * str, int current_year = -1);
date_t parse_date(const char * str,
optional<date_t::year_type> current_year = none);
inline date_t parse_date(const std::string& str, int current_year = -1) {
inline date_t parse_date(const std::string& str,
optional<date_t::year_type> current_year = none) {
return parse_date(str.c_str(), current_year);
}
inline std::time_t to_time_t(const ptime& t)
{
if( t == posix_time::neg_infin )
return 0;
else if( t == posix_time::pos_infin )
return LONG_MAX;
ptime start(date(1970,1,1));
return (t-start).total_seconds();
}
extern std::string output_datetime_format;
inline std::string format_datetime(const datetime_t& when,
const optional<std::string>& format = none)
{
char buf[256];
std::time_t moment = to_time_t(when);
std::strftime(buf, 255, format ? format->c_str() :
output_datetime_format.c_str(), std::localtime(&moment));
return buf;
posix_time::time_facet * facet
(new posix_time::time_facet(format ? format->c_str() :
output_datetime_format.c_str()));
std::ostringstream buf;
buf.imbue(std::locale(std::locale::classic(), facet));
buf << when;
return buf.str();
}
extern std::string output_date_format;
@ -121,11 +116,13 @@ extern std::string output_date_format;
inline std::string format_date(const date_t& when,
const optional<std::string>& format = none)
{
char buf[256];
std::tm moment = gregorian::to_tm(when);
std::strftime(buf, 255, format ? format->c_str() :
output_date_format.c_str(), &moment);
return buf;
gregorian::date_facet * facet
(new gregorian::date_facet(format ? format->c_str() :
output_date_format.c_str()));
std::ostringstream buf;
buf.imbue(std::locale(std::locale::classic(), facet));
buf << when;
return buf.str();
}
class date_interval_t : public equality_comparable<date_interval_t>

View file

@ -1,6 +1,6 @@
bal
<<<
2008/1/1 one
2008/01/01 one
test:a 1
test:b
>>>1

View file

@ -13,7 +13,7 @@ bal discover
liabilities:credit cards:discover 4462:interest $-28.17
assets:bank:wells fargo:checking
2008/3/1 * discover card payment
2008/03/01 * discover card payment
liabilities:credit cards:discover 4462 $1198.14
assets:bank:wells fargo:checking
>>>1