sqlite bug

build/lib/flask_wiz/cli.py

@@ -17,18 +17,14 @@ def cli():
 
 @cli.command()
 def new():
- # Prompt user for project name
  name = click.prompt('Enter project name')
 
- # Prompt user for database system
  db_options = ['mongodb', 'sqlite', 'mysql', 'postgresql']
  db = click.prompt('Select database system', type=click.Choice(db_options))
 
- # Additional setup code for the new project (e.g., creating files, folders, etc.)
  os.makedirs(name)
  os.chdir(name)
 
- # Create directories for templates and static files
  os.makedirs('templates')
  os.makedirs('static')
 
@@ -37,14 +33,15 @@ def new():
  gitignore_file.write("venv/\n")
  gitignore_file.write("__pycache__/\n")
  gitignore_file.write(".vscode/\n")
- # Add more entries as needed
 
  with open('.env', 'w') as env_file:
  env_file.write("# Default .env file for Flask project\n")
- # Add default environment variables if needed
 
  with open('app.py', 'w') as app_file:
+ db_module = None
+
  if db == 'mongodb':
+ db_module = 'pymongo'
  app_file.write(
  """from flask import Flask
 from pymongo import MongoClient
@@ -61,6 +58,7 @@ def index():
  app.run()
 """)
  elif db == 'sqlite':
+ db_module = 'sqlite3'
  app_file.write(
  """from flask import Flask
 import sqlite3
@@ -76,6 +74,7 @@ def index():
  app.run()
 """)
  elif db == 'mysql':
+ db_module = 'pymysql'
  app_file.write(
  """from flask import Flask
 import pymysql
@@ -91,6 +90,7 @@ def index():
  app.run()
 """)
  elif db == 'postgresql':
+ db_module = 'psycopg2'
  app_file.write(
  """from flask import Flask
 import psycopg2
@@ -106,7 +106,10 @@ def index():
  app.run()
 """)
 
+ if db_module:
+ os.system(f"pip install {db_module}") # Install the required database module
+
  click.echo(f'New Flask project "{name}" created successfully with {db} database!')
 
 if __name__ == '__main__':
- cli()
+ cli()

flask_wiz.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: flask-wiz
-Version: 1.6.3
+Version: 1.9.1
 Author: Krish Soni
 Description-Content-Type: text/markdown
 

flask_wiz.egg-info/requires.txt

@@ -1,7 +1,3 @@
 flask
 click
 Flask
-sqlite
-mysql-connector-python
-psycopg2
-pymongo

flask_wiz/cli.py

@@ -17,18 +17,14 @@ def cli():
 
 @cli.command()
 def new():
- # Prompt user for project name
  name = click.prompt('Enter project name')
 
- # Prompt user for database system
  db_options = ['mongodb', 'sqlite', 'mysql', 'postgresql']
  db = click.prompt('Select database system', type=click.Choice(db_options))
 
- # Additional setup code for the new project (e.g., creating files, folders, etc.)
  os.makedirs(name)
  os.chdir(name)
 
- # Create directories for templates and static files
  os.makedirs('templates')
  os.makedirs('static')
 
@@ -37,14 +33,15 @@ def new():
  gitignore_file.write("venv/\n")
  gitignore_file.write("__pycache__/\n")
  gitignore_file.write(".vscode/\n")
- # Add more entries as needed
 
  with open('.env', 'w') as env_file:
  env_file.write("# Default .env file for Flask project\n")
- # Add default environment variables if needed
 
  with open('app.py', 'w') as app_file:
+ db_module = None
+
  if db == 'mongodb':
+ db_module = 'pymongo'
  app_file.write(
  """from flask import Flask
 from pymongo import MongoClient
@@ -61,6 +58,7 @@ def index():
  app.run()
 """)
  elif db == 'sqlite':
+ db_module = 'sqlite3'
  app_file.write(
  """from flask import Flask
 import sqlite3
@@ -76,6 +74,7 @@ def index():
  app.run()
 """)
  elif db == 'mysql':
+ db_module = 'pymysql'
  app_file.write(
  """from flask import Flask
 import pymysql
@@ -91,6 +90,7 @@ def index():
  app.run()
 """)
  elif db == 'postgresql':
+ db_module = 'psycopg2'
  app_file.write(
  """from flask import Flask
 import psycopg2
@@ -106,7 +106,10 @@ def index():
  app.run()
 """)
 
+ if db_module:
+ os.system(f"pip install {db_module}") # Install the required database module
+
  click.echo(f'New Flask project "{name}" created successfully with {db} database!')
 
 if __name__ == '__main__':
- cli()
+ cli()

setup.py

@@ -7,15 +7,15 @@
  'flask',
  'click',
  'Flask', 
- 'sqlite',
- 'mysql-connector-python',
- 'psycopg2',
- 'pymongo',
+ # 'sqlite3',
+ # 'mysql-connector-python',
+ # 'psycopg2',
+ # 'pymongo',
 ]
 
 setup(
  name='flask-wiz',
- version='1.6.3',
+ version='1.9.1',
  author='Krish Soni',
  packages=find_packages(),
  include_package_data=True,
@@ -27,4 +27,4 @@
  },
  long_description=long_description,
  long_description_content_type="text/markdown",
-)
+)

wiz.py

@@ -58,7 +58,7 @@ def index():
  app.run()
 """)
  elif db == 'sqlite':
- db_module = 'sqlite3'
+ #db_module = 'sqlite3'
  app_file.write(
  """from flask import Flask
 import sqlite3
@@ -112,4 +112,4 @@ def index():
  click.echo(f'New Flask project "{name}" created successfully with {db} database!')
 
 if __name__ == '__main__':
- cli()
+ cli()