Fix a bug for # field (#558)

koxudaxi · Sep 30, 2021 · c015850 · c015850
1 parent 2329203
commit c015850
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 1 deletion.
diff --git a/datamodel_code_generator/reference.py b/datamodel_code_generator/reference.py
@@ -146,7 +146,7 @@ def get_valid_name(
         if not name:
             name = self.empty_field_name
         if name[0] == '#':
-            name = name[1:]
+            name = name[1:] or self.empty_field_name
         # TODO: when first character is a number
         name = re.sub(r'\W', '_', name)
         if name[0].isnumeric():

diff --git a/tests/data/expected/main/main_jsonschema_special_field_name/output.py b/tests/data/expected/main/main_jsonschema_special_field_name/output.py
@@ -15,3 +15,4 @@ class SpecialField(BaseModel):
     class_: Optional[int] = Field(None, alias='class')
     class_s: Optional[int] = Field(None, alias="class's")
     class_s_1: Optional[str] = Field(None, alias='class-s')
+    _: Optional[str] = Field(None, alias='#')
diff --git a/tests/data/jsonschema/special_field_name.json b/tests/data/jsonschema/special_field_name.json
@@ -17,6 +17,9 @@
     },
     "class-s": {
       "type": "string"
+    },
+    "#": {
+      "type": "string"
     }
   }
 }