org.neo4art.importer.wikipedia.parser.util.WikipediaDateTimeInfoboxParserUtils.java Source code

Introduction

Here is the source code for org.neo4art.importer.wikipedia.parser.util.WikipediaDateTimeInfoboxParserUtils.java

Source

/**
 * Copyright 2015 the original author or authors.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *      http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package org.neo4art.importer.wikipedia.parser.util;

import java.text.SimpleDateFormat;
import java.util.Calendar;
import java.util.Date;
import java.util.regex.Matcher;
import java.util.regex.Pattern;

import org.apache.commons.lang3.StringUtils;
import org.apache.commons.logging.Log;
import org.apache.commons.logging.LogFactory;

/**
 * Parser for
 * 
 * <a href="https://en.wikipedia.org/wiki/Template:Birth_date">Template:Birth_date</a>,
 * <a href="https://en.wikipedia.org/wiki/Template:Birth_date_and_age">Template:Birth_date_and_age</a>,
 * <a href="https://en.wikipedia.org/wiki/Template:Birth_year_and_age">Template:Birth_year_and_age</a>,
 * 
 * <a href="https://en.wikipedia.org/wiki/Template:Death_date_and_age">Template:Death_date_and_age</a>,
 * 
 * @author Lorenzo Speranzoni
 * @since 20 Oct 2015
 */
public class WikipediaDateTimeInfoboxParserUtils {

    private static Log logger = LogFactory.getLog(WikipediaDateTimeInfoboxParserUtils.class);

    /**
     * Known pattern:
     * 
     * {{Birth date|1993|2|4|df=yes}}
     * {{Birth date|1993|2|4|mf=yes}}
     * {{Birth date|1993|2|4}}
     * {{Birth date|df=yes|1756|03|04}}
     *
     * {{Birth date and age|1993|2|4|df=yes}}
     * {{Birth date and age|1993|2|4|mf=yes}}
     * {{Birth date and age|1993|2|4}}
     *
     * {{Death date and age|1993|2|24|1941|4|12|df=yes}}
     * {{Death date and age|1993|2|24|1941|4|df=yes}}
     * {{Death date and age|1993|2|24|1941|mf=yes}}
     * {{Death date and age|df=yes|1890|7|29|1853|3|30}}
     *
     * 30 March 1853
     * 
     * @param dateAsString
     * @return
     */
    public static Date parseAsDate(String dateAsString) {

        if (StringUtils.isBlank(dateAsString)) {

            return null;
        }

        final String PATTERNS[] = { "[\\{]{2}[Bb]irth date[\\s]*\\|([\\d]+)\\|([\\d]+)\\|([\\d]+)\\|[\\S]+[\\}]{2}",
                "[\\{]{2}[Bb]irth date[\\s]*\\|([\\d]+)\\|([\\d]+)\\|([\\d]+)[\\}]{2}",
                "[\\{]{2}[Bb]irth date[\\s]*\\|[\\D]+\\|([\\d]+)\\|([\\d]+)\\|([\\d]+)[\\}]{2}",

                "[\\{]{2}[Bb]irth date and age[\\s]*\\|([\\d]+)\\|([\\d]+)\\|([\\d]+)\\|[\\S]*",
                "[\\{]{2}[Bb]irth date and age[\\s]*\\|([\\d]+)\\|([\\d]+)\\|([\\d]+)[\\}]{2}",

                "[\\{]{2}[Dd]eath date and age[\\s]*\\|([\\d]+)\\|([\\d]+)\\|([\\d]+)\\|[\\S]*",
                "[\\{]{2}[Dd]eath date and age[\\s]*\\|[\\D]+\\|([\\d]+)\\|([\\d]+)\\|([\\d]+)\\|[\\S]*" };

        for (String patternRegex : PATTERNS) {

            Pattern patter = Pattern.compile(patternRegex);
            Matcher matcher = patter.matcher(dateAsString);

            if (matcher.matches()) {

                return toDate(matcher.group(1), matcher.group(2), matcher.group(3));
            }
        }

        try {

            return new SimpleDateFormat("dd MMM yyyy").parse(dateAsString);
        } catch (Exception e) {
        }

        try {

            Calendar year = Calendar.getInstance();
            year.set(Integer.parseInt(dateAsString), Calendar.JANUARY, 1, 0, 0, 0);
            year.set(Calendar.MILLISECOND, 0);

            return year.getTime();
        } catch (Exception e) {
        }

        logger.trace("Unable to parse " + dateAsString + " as Date.");

        return null;
    }

    /**
     * 
     * @param year
     * @param month
     * @param day
     * @return
     */
    public static Date toDate(String year, String month, String day) {

        try {

            Calendar calendar = Calendar.getInstance();

            calendar.set(Integer.parseInt(StringUtils.trim(year)), Integer.parseInt(StringUtils.trim(month)) - 1,
                    Integer.parseInt(StringUtils.trim(day)), 0, 0, 0);
            calendar.set(Calendar.MILLISECOND, 0);

            return calendar.getTime();
        } catch (Exception e) {
        }

        return null;
    }
}