Make it work for any country

elemoine · Apr 12, 2017 · c8a622f · c8a622f
1 parent 41bf983
commit c8a622f
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 5 deletions.
diff --git a/README.md b/README.md
@@ -24,10 +24,10 @@ Activate virtual env if `workalendar` was installed in a virtual env:
 SELECT workalendar.workon('/patch/to/virtual/env');
 ```
 
-Get the french holidays for a given year:
+Get the french holidays for 2017:
 
 ```sql
-SELECT * from workalendar.holidays(2017);
+SELECT * from workalendar.holidays(2017, 'europe', 'France');
     date    |             name
 ------------+------------------------------
  2017-01-01 | New year

diff --git a/workalendar--1.0.0.sql b/workalendar--1.0.0.sql
@@ -15,12 +15,13 @@ CREATE TYPE holiday AS (
     name text
 );
 
-CREATE OR REPLACE FUNCTION holidays(year int)
+CREATE OR REPLACE FUNCTION holidays(year int, continent text, country text)
   RETURNS SETOF holiday AS
 $BODY$
     from datetime import date
-    from workalendar.europe import France
-    cal = France()
+    from importlib import import_module
+    imp = import_module('workalendar.' + continent)
+    cal = getattr(imp, country)()
     for holiday in cal.holidays(year):
         yield holiday
 $BODY$