django-simplecms/cms/__main__.py

59 wiersze
2.4 KiB
Python

import os, re, argparse, shutil, example, cms
from pip._internal.operations import freeze as pip
def main():
parser = argparse.ArgumentParser(description='SimpleCMS')
parser.add_argument('project_name', nargs='?', default='.')
project_name = parser.parse_args().project_name
if project_name == '.':
project_dir = os.getcwd()
project_name = os.path.basename(project_dir)
else:
project_dir = os.path.join(os.getcwd(), project_name)
if re.match(r'^\w+$', project_name):
if input(f'Do you want to create a new project in `{project_dir}` ?\033[1D') in 'Yy':
create_project(project_name, project_dir)
else:
print(f'Invalid project name: {project_name}')
def create_project(project_name, project_dir):
os.makedirs(project_dir, exist_ok=True)
with open(os.path.join(project_dir, 'requirements.txt'), 'w') as f:
for line in pip.freeze():
if 'django_simplecms' in line:
line = f'django-simplecms=={cms.__version__}'
print(line, file=f)
shutil.copytree(os.path.dirname(example.__file__),os.path.join(project_dir, project_name), dirs_exist_ok=True)
with open(os.open(os.path.join(project_dir, 'manage.py'), os.O_CREAT|os.O_WRONLY, 0o755), 'w') as f:
print('#!/usr/bin/env python',
'import os, sys',
f"os.environ.setdefault('DJANGO_SETTINGS_MODULE', '{project_name}.settings')",
'from django.core.management import execute_from_command_line',
'execute_from_command_line(sys.argv)',
sep='\n', file=f)
with open(os.path.join(project_dir, project_name, 'wsgi.py'), 'w') as f:
print('import os',
'from django.core.wsgi import get_wsgi_application',
f"os.environ.setdefault('DJANGO_SETTINGS_MODULE', '{project_name}.settings')",
'application = get_wsgi_application()',
sep='\n', file=f)
with open(os.path.join(project_dir, '.gitignore'), 'w') as f:
print('*.pyc\n__pycache__/', file=f)
print(f'Successfully created project "{project_name}"',
'',
'Things to do next:',
'- create a database',
'- ./manage.py makemigrations',
'- ./manage.py migrate',
'- ./manage.py createsuperuser',
'- ./manage.py runserver',
sep='\n')
if __name__ == '__main__':
main()