Upper enumerant names

Close #18
dahlia · Jul 24, 2024 · 8cb9c58 · 8cb9c58
1 parent e211deb
commit 8cb9c58
Show file tree

Hide file tree

Showing 3 changed files with 27 additions and 21 deletions.
diff --git a/README.rst b/README.rst
@@ -10,24 +10,25 @@ This Python package contains `ISO 4217`_ currency data, represented as
 enum_ module which was introduced in 3.4.
 
 >>> from iso4217 import Currency
->>> Currency.usd
-<Currency.usd: 'USD'>
->>> Currency.usd.code
+>>> Currency.USD
+<Currency.USD: 'USD'>
+>>> Currency.USD.code
 'USD'
->>> Currency.usd.currency_name
+>>> Currency.USD.currency_name
 'US Dollar'
->>> Currency.usd.exponent  # USD has cents
+>>> Currency.USD.exponent  # USD has cents
 2
->>> Currency.jpy
-<Currency.jpy: 'JPY'>
->>> Currency.jpy.currency_name
+>>> Currency.JPY
+<Currency.JPY: 'JPY'>
+>>> Currency.JPY.currency_name
 'Yen'
->>> Currency.jpy.exponent  # JPY has no minor units
+>>> Currency.JPY.exponent  # JPY has no minor units
 0
 >>> Currency('KRW')  # Get by the code string
-<Currency.krw: 'KRW'>
->>> Currency.krw is Currency('KRW')
+<Currency.KRW: 'KRW'>
+>>> Currency.KRW is Currency('KRW')
 True
+>>> Currency.krw is Currency.KRW  # Lower enumerants are also available
 
 Written by `Hong Minhee`_.  Distributed under Public Domain.
 

diff --git a/iso4217/__init__.py b/iso4217/__init__.py
@@ -100,6 +100,7 @@ def update_enum_dict(locals_, raw_table):
             lcode += '_'
         enumerants[lcode] = code
     for code, enumerant in enumerants.items():
+        locals_[enumerant] = enumerant
         locals_[code] = enumerant
 
 

diff --git a/iso4217/test.py b/iso4217/test.py
@@ -2,18 +2,22 @@
 
 
 def test():
-    assert Currency.usd.code == 'USD'
-    assert Currency.usd.number == 840
-    assert Currency.usd.currency_name == 'US Dollar'
-    assert Currency.usd.country_names.issuperset(
+    assert Currency('USD') is Currency.USD
+    assert Currency.USD.code == 'USD'
+    assert Currency.USD.number == 840
+    assert Currency.USD.currency_name == 'US Dollar'
+    assert Currency.USD.country_names.issuperset(
         frozenset(['GUAM', 'UNITED STATES OF AMERICA (THE)'])
     )
-    assert Currency.usd.exponent == 2
-    assert Currency.jpy.code == 'JPY'
-    assert Currency.jpy.number == 392
-    assert Currency.jpy.currency_name == 'Yen'
-    assert Currency.jpy.country_names == frozenset(['JAPAN'])
-    assert Currency.jpy.exponent == 0
+    assert Currency.USD.exponent == 2
+    assert Currency.usd is Currency.USD
+    assert Currency('JPY') is Currency.JPY
+    assert Currency.JPY.code == 'JPY'
+    assert Currency.JPY.number == 392
+    assert Currency.JPY.currency_name == 'Yen'
+    assert Currency.JPY.country_names == frozenset(['JAPAN'])
+    assert Currency.JPY.exponent == 0
+    assert Currency.jpy is Currency.JPY
 
 
 if __name__ == '__main__':