Correctly import sql driver according to project generation

martenframework · Mar 7, 2024 · 2eef269 · 2eef269
1 parent 88954f7
commit 2eef269
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/src/marten/cli/manage/command/new/templates/project/src/project.cr.ecr b/src/marten/cli/manage/command/new/templates/project/src/project.cr.ecr
@@ -3,7 +3,13 @@ require "marten"
 <%- if context.targets_auth? -%>
 require "marten_auth"
 <%- end -%>
+<%- if context.database == "sqlite3" -%>
 require "sqlite3"
+<%- elsif context.database == "mysql" -%>
+require "mysql"
+<%- elsif context.database == "postgresql" -%>
+require "pg"
+<%- end -%>
 
 # Configuration requirements.
 require "../config/settings/base"