63e799695d88575c41d67359ae28b0482ccc5f8d
[librarian.git] / src / librarian / util.py
1 # Functions to convert between integers and Roman numerals.
2 # by Paul Winkler
3 # http://code.activestate.com/recipes/81611-roman-numerals/
4 # PSFL (GPL compatible)
5 from __future__ import print_function, unicode_literals
6
7 import os
8 import six
9
10
11 def int_to_roman(input):
12     """
13     Convert an integer to Roman numerals.
14
15     Examples:
16     >>> int_to_roman(0)
17     Traceback (most recent call last):
18     ValueError: Argument must be between 1 and 3999
19
20     >>> int_to_roman(-1)
21     Traceback (most recent call last):
22     ValueError: Argument must be between 1 and 3999
23
24     >>> int_to_roman(1.5)  # doctest: +IGNORE_EXCEPTION_DETAIL
25     Traceback (most recent call last):
26     TypeError: expected integer, got <type 'float'>
27
28     >>> for i in range(1, 21): print(int_to_roman(i))
29     ...
30     I
31     II
32     III
33     IV
34     V
35     VI
36     VII
37     VIII
38     IX
39     X
40     XI
41     XII
42     XIII
43     XIV
44     XV
45     XVI
46     XVII
47     XVIII
48     XIX
49     XX
50     >>> print(int_to_roman(2000))
51     MM
52     >>> print(int_to_roman(1999))
53     MCMXCIX
54     """
55     if not isinstance(input, int):
56         raise TypeError("expected integer, got %s" % type(input))
57     if not 0 < input < 4000:
58         raise ValueError("Argument must be between 1 and 3999")
59     ints = (1000, 900,  500, 400, 100,  90, 50,  40, 10,  9,    5,  4,    1)
60     nums = ('M', 'CM', 'D', 'CD', 'C', 'XC', 'L', 'XL', 'X', 'IX', 'V', 'IV',
61             'I')
62     result = ""
63     for i in range(len(ints)):
64         count = int(input / ints[i])
65         result += nums[i] * count
66         input -= ints[i] * count
67     return result
68
69
70 def roman_to_int(input):
71     """
72     Convert a roman numeral to an integer.
73
74     >>> r = list(range(1, 4000))
75     >>> nums = [int_to_roman(i) for i in r]
76     >>> ints = [roman_to_int(n) for n in nums]
77     >>> print(r == ints)
78     1
79
80     >>> roman_to_int('VVVIV')
81     Traceback (most recent call last):
82      ...
83     ValueError: input is not a valid roman numeral: VVVIV
84     >>> roman_to_int(1)  # doctest: +IGNORE_EXCEPTION_DETAIL
85     Traceback (most recent call last):
86      ...
87     TypeError: expected string, got <type 'int'>
88     >>> roman_to_int('a')
89     Traceback (most recent call last):
90      ...
91     ValueError: input is not a valid roman numeral: A
92     >>> roman_to_int('IL')
93     Traceback (most recent call last):
94      ...
95     ValueError: input is not a valid roman numeral: IL
96     """
97     if not isinstance(input, six.text_type):
98         raise TypeError("expected string, got %s" % type(input))
99     input = input.upper()
100     nums = ['M', 'D', 'C', 'L', 'X', 'V', 'I']
101     ints = [1000, 500, 100, 50,  10,  5,    1]
102     places = []
103     for c in input:
104         if c not in nums:
105             raise ValueError("input is not a valid roman numeral: %s" % input)
106     for i in range(len(input)):
107         c = input[i]
108         value = ints[nums.index(c)]
109         # If the next place holds a larger number, this value is negative.
110         try:
111             nextvalue = ints[nums.index(input[i + 1])]
112             if nextvalue > value:
113                 value *= -1
114         except IndexError:
115             # there is no next place.
116             pass
117         places.append(value)
118     sum = 0
119     for n in places:
120         sum += n
121     # Easiest test for validity...
122     if int_to_roman(sum) == input:
123         return sum
124     else:
125         raise ValueError('input is not a valid roman numeral: %s' % input)
126
127
128 def makedirs(path):
129     if not os.path.isdir(path):
130         os.makedirs(path)
131
132
133 def get_translation(language):
134     import gettext
135     from .functions import lang_code_3to2
136
137     return gettext.translation(
138         'messages',
139         localedir=os.path.join(os.path.dirname(__file__), 'locale'),
140         languages=[lang_code_3to2(language), 'pl'],
141     )