Fix generating table names from uppercase model names (#256)

* Fix generating table names from uppercase model names

* Some tests

Author: droserasprout

CHANGELOG.md

@@ -2,6 +2,12 @@
 
 Please use [this](https://docs.gitlab.com/ee/development/changelog.html) document as guidelines to keep a changelog.
 
+## [unreleased]
+
+### Fixed
+
+* database: Fixed generating table names from uppercase model names.
+
 ## 4.2.5 - 2022-02-21
 
 ### Fixed

src/dipdup/utils/__init__.py

@@ -61,13 +61,20 @@ async def slowdown(seconds: int):
 
 def snake_to_pascal(value: str) -> str:
     """humps wrapper for Python imports"""
-    return humps.pascalize(value.replace('.', '_'))
+    value = value.replace('.', '_')
+    # NOTE: Special case, humps returns uppercase otherwise
+    if value.isupper():
+        value = value.lower()
+    return humps.pascalize(value)
 
 
 def pascal_to_snake(value: str, strip_dots: bool = True) -> str:
     """humps wrapper for Python imports"""
     if strip_dots:
         value = value.replace('.', '_')
+    # NOTE: Special case, humps returns uppercase otherwise
+    if value.isupper():
+        value = value.lower()
     return humps.depascalize(value).replace('__', '_')
 
 

tests/test_dipdup/test_utils.py

@@ -5,6 +5,8 @@
 
 from dipdup.models import Index
 from dipdup.models import IndexType
+from dipdup.utils import pascal_to_snake
+from dipdup.utils import snake_to_pascal
 from dipdup.utils.database import in_global_transaction
 from dipdup.utils.database import tortoise_wrapper
 
@@ -39,3 +41,20 @@ async def test_in_global_transaction(self):
                     raise Exception
             count = await Index.filter().count()
             self.assertEqual(3, count)
+
+    async def test_humps_helpers(self) -> None:
+        self.assertEqual('foo_bar', pascal_to_snake('foo_bar', True))
+        self.assertEqual('foo_bar', pascal_to_snake('FooBar', True))
+        self.assertEqual('foo_bar', pascal_to_snake('Foo.Bar', True))
+        self.assertEqual('foobar', pascal_to_snake('FOOBAR', True))
+
+        self.assertEqual('foo_bar', pascal_to_snake('foo_bar', False))
+        self.assertEqual('foo_bar', pascal_to_snake('FooBar', False))
+        self.assertEqual('foo._bar', pascal_to_snake('Foo.Bar', False))
+        self.assertEqual('foobar', pascal_to_snake('FOOBAR', False))
+
+        self.assertEqual('FooBar', snake_to_pascal('fooBar'))
+        self.assertEqual('FooBar', snake_to_pascal('FooBar'))
+        self.assertEqual('Foobar', snake_to_pascal('foobar'))
+        self.assertEqual('FooBar', snake_to_pascal('foo__bar'))
+        self.assertEqual('Foobar', snake_to_pascal('FOOBAR'))