1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
|
from django.template import Template, Context
import os.path
from drf_generators.templates.serializer import SERIALIZER
from drf_generators.templates.apiview import API_URL, API_VIEW
from drf_generators.templates.viewset import VIEW_SET_URL, VIEW_SET_VIEW
from drf_generators.templates.function import FUNCTION_URL, FUNCTION_VIEW
from drf_generators.templates.modelviewset import MODEL_URL, MODEL_VIEW
__all__ = ['BaseGenerator', 'APIViewGenerator', 'ViewSetGenerator',
'FunctionViewGenerator', 'ModelViewSetGenerator']
class BaseGenerator(object):
def __init__(self, app_config, force):
self.app_config = app_config
self.force = force
self.app = app_config.models_module
self.name = app_config.name
self.serializer_template = Template(SERIALIZER)
self.models = self.get_model_names()
self.serializers = self.get_serializer_names()
def generate_serializers(self, depth):
content = self.serializer_content(depth)
filename = 'serializers.py'
if self.write_file(content, filename):
return ' - writing %s' % filename
else:
return 'Serializer generation cancelled'
def generate_views(self):
content = self.view_content()
filename = 'views.py'
if self.write_file(content, filename):
return ' - writing %s' % filename
else:
return 'View generation cancelled'
def generate_urls(self):
content = self.url_content()
filename = 'urls.py'
if self.write_file(content, filename):
return ' - writing %s' % filename
else:
return 'Url generation cancelled'
def serializer_content(self, depth):
context = Context({'app': self.name, 'models': self.models,
'depth': depth})
return self.serializer_template.render(context)
def view_content(self):
context = Context({'app': self.name, 'models': self.models,
'serializers': self.serializers})
return self.view_template.render(context)
def url_content(self):
context = Context({'app': self.name, 'models': self.models})
return self.url_template.render(context)
def get_model_names(self):
return [m.__name__ for m in self.app_config.get_models()]
def get_serializer_names(self):
return [m + 'Serializer' for m in self.models]
def write_file(self, content, filename):
name = os.path.join(os.path.dirname(self.app.__file__), filename)
if os.path.exists(name) and not self.force:
msg = "Are you sure you want to overwrite %s? (y/n): " % filename
prompt = input # python3
response = prompt(msg)
if response != "y":
return False
new_file = open(name, 'w+')
new_file.write(content)
new_file.close()
return True
class APIViewGenerator(BaseGenerator):
def __init__(self, app_config, force):
super(APIViewGenerator, self).__init__(app_config, force)
self.view_template = Template(API_VIEW)
self.url_template = Template(API_URL)
class ViewSetGenerator(BaseGenerator):
def __init__(self, app_config, force):
super(ViewSetGenerator, self).__init__(app_config, force)
self.view_template = Template(VIEW_SET_VIEW)
self.url_template = Template(VIEW_SET_URL)
class FunctionViewGenerator(BaseGenerator):
def __init__(self, app_config, force):
super(FunctionViewGenerator, self).__init__(app_config, force)
self.view_template = Template(FUNCTION_VIEW)
self.url_template = Template(FUNCTION_URL)
class ModelViewSetGenerator(BaseGenerator):
def __init__(self, app_config, force):
super(ModelViewSetGenerator, self).__init__(app_config, force)
self.view_template = Template(MODEL_VIEW)
self.url_template = Template(MODEL_URL)
|
