Commits

George Notaras committed 275b7d2

Added support for forbidden database names

Comments (0)

Files changed (3)

src/PostgreSQL_manager/admin.py

 from django.db.utils import DatabaseError, IntegrityError
 from django.contrib import messages
 
-from PostgreSQL_manager.forms import PgUserModelForm
+from PostgreSQL_manager.forms import PgUserModelForm, PgDatabaseModelForm
 
 
 
 
 class PgDatabaseAdmin(admin.ModelAdmin):
 
-    readonly_fields = ['name', 'owner', 'date_created', 'date_modified']
+    form = PgDatabaseModelForm
     list_display = ('name', 'owner', 'connlimit', 'date_created')
     
     def get_form(self, request, obj=None, **kwargs):

src/PostgreSQL_manager/forms.py

         return password2
 
 
+
+class PgDatabaseModelForm(forms.ModelForm):
+
+    class Meta:
+        model = cache.get_model('PostgreSQL_manager', 'PgDatabase')
+    
+    def clean_name(self):
+        name = self.cleaned_data.get('name')
+        if name.lower() in settings.PGMANAGER_FORBIDDEN_DATABASE_NAMES:
+            self._errors['name'] = self.error_class(['This name is reserved for internal use.'])
+        return name
+
+

src/PostgreSQL_manager/settings.py

     )
 PGMANAGER_FORBIDDEN_USER_NAMES = getattr(settings, 'PGMANAGER_FORBIDDEN_USER_NAMES', _PGMANAGER_FORBIDDEN_USER_NAMES)
 
+_PGMANAGER_FORBIDDEN_DATABASE_NAMES = (
+    'postgres',
+    'template0',
+    'template1',
+    )
+PGMANAGER_FORBIDDEN_DATABASE_NAMES = getattr(settings, 'PGMANAGER_FORBIDDEN_DATABASE_NAMES', _PGMANAGER_FORBIDDEN_DATABASE_NAMES)
+