quicktype-core 23.0.170 → 23.0.171

package/dist/language/Python/PythonRenderer.js

@@ -99,7 +99,9 @@ class PythonRenderer extends ConvenienceRenderer_1.ConvenienceRenderer {
             const memberTypes = Array.from(nonNulls).map(m => this.pythonType(m));
             if (hasNull !== null) {
                 let rest = [];
-                if (!this.getAlphabetizeProperties() && this.pyOptions.features.dataClasses && _isRootTypeDef) {
+                if (!this.getAlphabetizeProperties() &&
+                    (this.pyOptions.features.dataClasses || this.pyOptions.pydanticBaseModel) &&
+                    _isRootTypeDef) {
                     // Only push "= None" if this is a root level type def
                     //   otherwise we may get type defs like List[Optional[int] = None]
                     //   which are invalid
@@ -134,6 +136,9 @@ class PythonRenderer extends ConvenienceRenderer_1.ConvenienceRenderer {
     }
     declarationLine(t) {
         if (t instanceof Type_1.ClassType) {
+            if (this.pyOptions.pydanticBaseModel) {
+                return ["class ", this.nameForNamedType(t), "(", this.withImport("pydantic", "BaseModel"), "):"];
+            }
             return ["class ", this.nameForNamedType(t), ":"];
         }
         if (t instanceof Type_1.EnumType) {
@@ -149,7 +154,7 @@ class PythonRenderer extends ConvenienceRenderer_1.ConvenienceRenderer {
         this.declaredTypes.add(t);
     }
     emitClassMembers(t) {
-        if (this.pyOptions.features.dataClasses)
+        if (this.pyOptions.features.dataClasses || this.pyOptions.pydanticBaseModel)
             return;
         const args = [];
         this.forEachClassProperty(t, "none", (name, _, cp) => {
@@ -179,7 +184,7 @@ class PythonRenderer extends ConvenienceRenderer_1.ConvenienceRenderer {
         return this.typeHint(" -> ", this.withTyping(type));
     }
     sortClassProperties(properties, propertyNames) {
-        if (this.pyOptions.features.dataClasses) {
+        if (this.pyOptions.features.dataClasses || this.pyOptions.pydanticBaseModel) {
             return (0, collection_utils_1.mapSortBy)(properties, (p) => {
                 return (p.type instanceof Type_1.UnionType && (0, TypeUtils_1.nullableFromUnion)(p.type) != null) || p.isOptional ? 1 : 0;
             });
@@ -189,7 +194,7 @@ class PythonRenderer extends ConvenienceRenderer_1.ConvenienceRenderer {
         }
     }
     emitClass(t) {
-        if (this.pyOptions.features.dataClasses) {
+        if (this.pyOptions.features.dataClasses && !this.pyOptions.pydanticBaseModel) {
             this.emitLine("@", this.withImport("dataclasses", "dataclass"));
         }
         this.declareType(t, () => {

package/dist/language/Python/language.d.ts

@@ -13,6 +13,7 @@ export declare const pythonOptions: {
     features: EnumOption<PythonFeatures>;
     justTypes: BooleanOption;
     nicePropertyNames: BooleanOption;
+    pydanticBaseModel: BooleanOption;
 };
 export declare class PythonTargetLanguage extends TargetLanguage {
     protected getOptions(): Array<Option<FixMeOptionsAnyType>>;

package/dist/language/Python/language.js

@@ -14,11 +14,17 @@ exports.pythonOptions = {
         ["3.7", { typeHints: true, dataClasses: true }]
     ], "3.6"),
     justTypes: new RendererOptions_1.BooleanOption("just-types", "Classes only", false),
-    nicePropertyNames: new RendererOptions_1.BooleanOption("nice-property-names", "Transform property names to be Pythonic", true)
+    nicePropertyNames: new RendererOptions_1.BooleanOption("nice-property-names", "Transform property names to be Pythonic", true),
+    pydanticBaseModel: new RendererOptions_1.BooleanOption("pydantic-base-model", "Uses pydantic BaseModel", false)
 };
 class PythonTargetLanguage extends TargetLanguage_1.TargetLanguage {
     getOptions() {
-        return [exports.pythonOptions.features, exports.pythonOptions.justTypes, exports.pythonOptions.nicePropertyNames];
+        return [
+            exports.pythonOptions.features,
+            exports.pythonOptions.justTypes,
+            exports.pythonOptions.nicePropertyNames,
+            exports.pythonOptions.pydanticBaseModel
+        ];
     }
     get stringTypeMapping() {
         const mapping = new Map();

package/package.json

@@ -1,6 +1,6 @@
 {
     "name": "quicktype-core",
-    "version": "23.0.170",
+    "version": "23.0.171",
     "description": "The quicktype engine as a library",
     "license": "Apache-2.0",
     "main": "dist/index.js",