Skip to content

Commit af3326b

Browse files
committed
Added migrations for (0008) whitelisting any users who have whitelisted projects, and (0009) removing the whitelisted field from the Project model
1 parent 4e5e9ce commit af3326b

File tree

2 files changed

+248
-0
lines changed

2 files changed

+248
-0
lines changed
+135
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,135 @@
1+
# encoding: utf-8
2+
import datetime
3+
from south.db import db
4+
from south.v2 import DataMigration
5+
from django.db import models
6+
from projects.models import Project
7+
from core.models import UserProfile
8+
from django.contrib.auth.models import User
9+
from django.core.exceptions import ObjectDoesNotExist
10+
11+
class Migration(DataMigration):
12+
13+
def forwards(self, orm):
14+
"Whitelist any user who has whitelisted projects."
15+
for project in Project.objects.all():
16+
if project.whitelisted:
17+
# If user doesn't have a profile, create one
18+
try:
19+
profile = project.user.get_profile()
20+
except ObjectDoesNotExist:
21+
profile = UserProfile(user=project.user)
22+
# Whitelist the user's profile
23+
profile.whitelisted = True
24+
profile.save()
25+
26+
27+
def backwards(self, orm):
28+
"Whitelist all projects owned by whitelisted users."
29+
for project in Project.objects.all():
30+
# If user doesn't have a profile, skip
31+
# (assume they are not whitelisted)
32+
try:
33+
profile = project.user.get_profile()
34+
except ObjectDoesNotExist:
35+
pass
36+
else:
37+
# Whitelist the user's projects
38+
if profile.whitelisted:
39+
project.whitelisted = True
40+
project.save()
41+
42+
43+
models = {
44+
'auth.group': {
45+
'Meta': {'object_name': 'Group'},
46+
'id': ('django.db.models.fields.AutoField', [], {'primary_key': 'True'}),
47+
'name': ('django.db.models.fields.CharField', [], {'unique': 'True', 'max_length': '80'}),
48+
'permissions': ('django.db.models.fields.related.ManyToManyField', [], {'to': "orm['auth.Permission']", 'symmetrical': 'False', 'blank': 'True'})
49+
},
50+
'auth.permission': {
51+
'Meta': {'ordering': "('content_type__app_label', 'content_type__model', 'codename')", 'unique_together': "(('content_type', 'codename'),)", 'object_name': 'Permission'},
52+
'codename': ('django.db.models.fields.CharField', [], {'max_length': '100'}),
53+
'content_type': ('django.db.models.fields.related.ForeignKey', [], {'to': "orm['contenttypes.ContentType']"}),
54+
'id': ('django.db.models.fields.AutoField', [], {'primary_key': 'True'}),
55+
'name': ('django.db.models.fields.CharField', [], {'max_length': '50'})
56+
},
57+
'auth.user': {
58+
'Meta': {'object_name': 'User'},
59+
'date_joined': ('django.db.models.fields.DateTimeField', [], {'default': 'datetime.datetime.now'}),
60+
'email': ('django.db.models.fields.EmailField', [], {'max_length': '75', 'blank': 'True'}),
61+
'first_name': ('django.db.models.fields.CharField', [], {'max_length': '30', 'blank': 'True'}),
62+
'groups': ('django.db.models.fields.related.ManyToManyField', [], {'to': "orm['auth.Group']", 'symmetrical': 'False', 'blank': 'True'}),
63+
'id': ('django.db.models.fields.AutoField', [], {'primary_key': 'True'}),
64+
'is_active': ('django.db.models.fields.BooleanField', [], {'default': 'True'}),
65+
'is_staff': ('django.db.models.fields.BooleanField', [], {'default': 'False'}),
66+
'is_superuser': ('django.db.models.fields.BooleanField', [], {'default': 'False'}),
67+
'last_login': ('django.db.models.fields.DateTimeField', [], {'default': 'datetime.datetime.now'}),
68+
'last_name': ('django.db.models.fields.CharField', [], {'max_length': '30', 'blank': 'True'}),
69+
'password': ('django.db.models.fields.CharField', [], {'max_length': '128'}),
70+
'user_permissions': ('django.db.models.fields.related.ManyToManyField', [], {'to': "orm['auth.Permission']", 'symmetrical': 'False', 'blank': 'True'}),
71+
'username': ('django.db.models.fields.CharField', [], {'unique': 'True', 'max_length': '30'})
72+
},
73+
'contenttypes.contenttype': {
74+
'Meta': {'ordering': "('name',)", 'unique_together': "(('app_label', 'model'),)", 'object_name': 'ContentType', 'db_table': "'django_content_type'"},
75+
'app_label': ('django.db.models.fields.CharField', [], {'max_length': '100'}),
76+
'id': ('django.db.models.fields.AutoField', [], {'primary_key': 'True'}),
77+
'model': ('django.db.models.fields.CharField', [], {'max_length': '100'}),
78+
'name': ('django.db.models.fields.CharField', [], {'max_length': '100'})
79+
},
80+
'projects.file': {
81+
'Meta': {'ordering': "('denormalized_path',)", 'object_name': 'File'},
82+
'content': ('django.db.models.fields.TextField', [], {}),
83+
'denormalized_path': ('django.db.models.fields.CharField', [], {'max_length': '255'}),
84+
'heading': ('django.db.models.fields.CharField', [], {'max_length': '255'}),
85+
'id': ('django.db.models.fields.AutoField', [], {'primary_key': 'True'}),
86+
'ordering': ('django.db.models.fields.PositiveSmallIntegerField', [], {'default': '1'}),
87+
'parent': ('django.db.models.fields.related.ForeignKey', [], {'blank': 'True', 'related_name': "'children'", 'null': 'True', 'to': "orm['projects.File']"}),
88+
'project': ('django.db.models.fields.related.ForeignKey', [], {'related_name': "'files'", 'to': "orm['projects.Project']"}),
89+
'slug': ('django.db.models.fields.SlugField', [], {'max_length': '50', 'db_index': 'True'}),
90+
'status': ('django.db.models.fields.PositiveSmallIntegerField', [], {'default': '1'})
91+
},
92+
'projects.filerevision': {
93+
'Meta': {'ordering': "('-revision_number',)", 'object_name': 'FileRevision'},
94+
'comment': ('django.db.models.fields.TextField', [], {'blank': 'True'}),
95+
'created_date': ('django.db.models.fields.DateTimeField', [], {'auto_now_add': 'True', 'blank': 'True'}),
96+
'diff': ('django.db.models.fields.TextField', [], {'blank': 'True'}),
97+
'file': ('django.db.models.fields.related.ForeignKey', [], {'related_name': "'revisions'", 'to': "orm['projects.File']"}),
98+
'id': ('django.db.models.fields.AutoField', [], {'primary_key': 'True'}),
99+
'is_reverted': ('django.db.models.fields.BooleanField', [], {'default': 'False'}),
100+
'revision_number': ('django.db.models.fields.IntegerField', [], {})
101+
},
102+
'projects.importedfile': {
103+
'Meta': {'object_name': 'ImportedFile'},
104+
'id': ('django.db.models.fields.AutoField', [], {'primary_key': 'True'}),
105+
'md5': ('django.db.models.fields.CharField', [], {'max_length': '255'}),
106+
'name': ('django.db.models.fields.CharField', [], {'max_length': '255'}),
107+
'path': ('django.db.models.fields.CharField', [], {'max_length': '255'}),
108+
'project': ('django.db.models.fields.related.ForeignKey', [], {'related_name': "'imported_files'", 'to': "orm['projects.Project']"}),
109+
'slug': ('django.db.models.fields.SlugField', [], {'max_length': '50', 'db_index': 'True'})
110+
},
111+
'projects.project': {
112+
'Meta': {'ordering': "('slug',)", 'object_name': 'Project'},
113+
'copyright': ('django.db.models.fields.CharField', [], {'max_length': '255', 'blank': 'True'}),
114+
'description': ('django.db.models.fields.TextField', [], {'blank': 'True'}),
115+
'docs_directory': ('django.db.models.fields.CharField', [], {'max_length': '255', 'blank': 'True'}),
116+
'extensions': ('django.db.models.fields.CharField', [], {'default': "''", 'max_length': '255'}),
117+
'id': ('django.db.models.fields.AutoField', [], {'primary_key': 'True'}),
118+
'modified_date': ('django.db.models.fields.DateTimeField', [], {'auto_now': 'True', 'blank': 'True'}),
119+
'name': ('django.db.models.fields.CharField', [], {'max_length': '255'}),
120+
'path': ('django.db.models.fields.CharField', [], {'max_length': '255'}),
121+
'project_url': ('django.db.models.fields.URLField', [], {'max_length': '200', 'blank': 'True'}),
122+
'pub_date': ('django.db.models.fields.DateTimeField', [], {'auto_now_add': 'True', 'blank': 'True'}),
123+
'repo': ('django.db.models.fields.CharField', [], {'max_length': '100', 'blank': 'True'}),
124+
'repo_type': ('django.db.models.fields.CharField', [], {'default': "'git'", 'max_length': '10'}),
125+
'skip': ('django.db.models.fields.BooleanField', [], {'default': 'False'}),
126+
'slug': ('django.db.models.fields.SlugField', [], {'max_length': '50', 'db_index': 'True'}),
127+
'status': ('django.db.models.fields.PositiveSmallIntegerField', [], {'default': '1'}),
128+
'suffix': ('django.db.models.fields.CharField', [], {'default': "'.rst'", 'max_length': '10'}),
129+
'theme': ('django.db.models.fields.CharField', [], {'default': "'default'", 'max_length': '20'}),
130+
'user': ('django.db.models.fields.related.ForeignKey', [], {'related_name': "'projects'", 'to': "orm['auth.User']"}),
131+
'version': ('django.db.models.fields.CharField', [], {'max_length': '100', 'blank': 'True'})
132+
}
133+
}
134+
135+
complete_apps = ['projects']
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,113 @@
1+
# encoding: utf-8
2+
import datetime
3+
from south.db import db
4+
from south.v2 import SchemaMigration
5+
from django.db import models
6+
7+
class Migration(SchemaMigration):
8+
9+
def forwards(self, orm):
10+
11+
# Deleting field 'Project.whitelisted'
12+
db.delete_column('projects_project', 'whitelisted')
13+
14+
15+
def backwards(self, orm):
16+
17+
# Adding field 'Project.whitelisted'
18+
db.add_column('projects_project', 'whitelisted', self.gf('django.db.models.fields.BooleanField')(default=False), keep_default=False)
19+
20+
21+
models = {
22+
'auth.group': {
23+
'Meta': {'object_name': 'Group'},
24+
'id': ('django.db.models.fields.AutoField', [], {'primary_key': 'True'}),
25+
'name': ('django.db.models.fields.CharField', [], {'unique': 'True', 'max_length': '80'}),
26+
'permissions': ('django.db.models.fields.related.ManyToManyField', [], {'to': "orm['auth.Permission']", 'symmetrical': 'False', 'blank': 'True'})
27+
},
28+
'auth.permission': {
29+
'Meta': {'ordering': "('content_type__app_label', 'content_type__model', 'codename')", 'unique_together': "(('content_type', 'codename'),)", 'object_name': 'Permission'},
30+
'codename': ('django.db.models.fields.CharField', [], {'max_length': '100'}),
31+
'content_type': ('django.db.models.fields.related.ForeignKey', [], {'to': "orm['contenttypes.ContentType']"}),
32+
'id': ('django.db.models.fields.AutoField', [], {'primary_key': 'True'}),
33+
'name': ('django.db.models.fields.CharField', [], {'max_length': '50'})
34+
},
35+
'auth.user': {
36+
'Meta': {'object_name': 'User'},
37+
'date_joined': ('django.db.models.fields.DateTimeField', [], {'default': 'datetime.datetime.now'}),
38+
'email': ('django.db.models.fields.EmailField', [], {'max_length': '75', 'blank': 'True'}),
39+
'first_name': ('django.db.models.fields.CharField', [], {'max_length': '30', 'blank': 'True'}),
40+
'groups': ('django.db.models.fields.related.ManyToManyField', [], {'to': "orm['auth.Group']", 'symmetrical': 'False', 'blank': 'True'}),
41+
'id': ('django.db.models.fields.AutoField', [], {'primary_key': 'True'}),
42+
'is_active': ('django.db.models.fields.BooleanField', [], {'default': 'True'}),
43+
'is_staff': ('django.db.models.fields.BooleanField', [], {'default': 'False'}),
44+
'is_superuser': ('django.db.models.fields.BooleanField', [], {'default': 'False'}),
45+
'last_login': ('django.db.models.fields.DateTimeField', [], {'default': 'datetime.datetime.now'}),
46+
'last_name': ('django.db.models.fields.CharField', [], {'max_length': '30', 'blank': 'True'}),
47+
'password': ('django.db.models.fields.CharField', [], {'max_length': '128'}),
48+
'user_permissions': ('django.db.models.fields.related.ManyToManyField', [], {'to': "orm['auth.Permission']", 'symmetrical': 'False', 'blank': 'True'}),
49+
'username': ('django.db.models.fields.CharField', [], {'unique': 'True', 'max_length': '30'})
50+
},
51+
'contenttypes.contenttype': {
52+
'Meta': {'ordering': "('name',)", 'unique_together': "(('app_label', 'model'),)", 'object_name': 'ContentType', 'db_table': "'django_content_type'"},
53+
'app_label': ('django.db.models.fields.CharField', [], {'max_length': '100'}),
54+
'id': ('django.db.models.fields.AutoField', [], {'primary_key': 'True'}),
55+
'model': ('django.db.models.fields.CharField', [], {'max_length': '100'}),
56+
'name': ('django.db.models.fields.CharField', [], {'max_length': '100'})
57+
},
58+
'projects.file': {
59+
'Meta': {'ordering': "('denormalized_path',)", 'object_name': 'File'},
60+
'content': ('django.db.models.fields.TextField', [], {}),
61+
'denormalized_path': ('django.db.models.fields.CharField', [], {'max_length': '255'}),
62+
'heading': ('django.db.models.fields.CharField', [], {'max_length': '255'}),
63+
'id': ('django.db.models.fields.AutoField', [], {'primary_key': 'True'}),
64+
'ordering': ('django.db.models.fields.PositiveSmallIntegerField', [], {'default': '1'}),
65+
'parent': ('django.db.models.fields.related.ForeignKey', [], {'blank': 'True', 'related_name': "'children'", 'null': 'True', 'to': "orm['projects.File']"}),
66+
'project': ('django.db.models.fields.related.ForeignKey', [], {'related_name': "'files'", 'to': "orm['projects.Project']"}),
67+
'slug': ('django.db.models.fields.SlugField', [], {'max_length': '50', 'db_index': 'True'}),
68+
'status': ('django.db.models.fields.PositiveSmallIntegerField', [], {'default': '1'})
69+
},
70+
'projects.filerevision': {
71+
'Meta': {'ordering': "('-revision_number',)", 'object_name': 'FileRevision'},
72+
'comment': ('django.db.models.fields.TextField', [], {'blank': 'True'}),
73+
'created_date': ('django.db.models.fields.DateTimeField', [], {'auto_now_add': 'True', 'blank': 'True'}),
74+
'diff': ('django.db.models.fields.TextField', [], {'blank': 'True'}),
75+
'file': ('django.db.models.fields.related.ForeignKey', [], {'related_name': "'revisions'", 'to': "orm['projects.File']"}),
76+
'id': ('django.db.models.fields.AutoField', [], {'primary_key': 'True'}),
77+
'is_reverted': ('django.db.models.fields.BooleanField', [], {'default': 'False'}),
78+
'revision_number': ('django.db.models.fields.IntegerField', [], {})
79+
},
80+
'projects.importedfile': {
81+
'Meta': {'object_name': 'ImportedFile'},
82+
'id': ('django.db.models.fields.AutoField', [], {'primary_key': 'True'}),
83+
'md5': ('django.db.models.fields.CharField', [], {'max_length': '255'}),
84+
'name': ('django.db.models.fields.CharField', [], {'max_length': '255'}),
85+
'path': ('django.db.models.fields.CharField', [], {'max_length': '255'}),
86+
'project': ('django.db.models.fields.related.ForeignKey', [], {'related_name': "'imported_files'", 'to': "orm['projects.Project']"}),
87+
'slug': ('django.db.models.fields.SlugField', [], {'max_length': '50', 'db_index': 'True'})
88+
},
89+
'projects.project': {
90+
'Meta': {'ordering': "('slug',)", 'object_name': 'Project'},
91+
'copyright': ('django.db.models.fields.CharField', [], {'max_length': '255', 'blank': 'True'}),
92+
'description': ('django.db.models.fields.TextField', [], {'blank': 'True'}),
93+
'docs_directory': ('django.db.models.fields.CharField', [], {'max_length': '255', 'blank': 'True'}),
94+
'extensions': ('django.db.models.fields.CharField', [], {'default': "''", 'max_length': '255'}),
95+
'id': ('django.db.models.fields.AutoField', [], {'primary_key': 'True'}),
96+
'modified_date': ('django.db.models.fields.DateTimeField', [], {'auto_now': 'True', 'blank': 'True'}),
97+
'name': ('django.db.models.fields.CharField', [], {'max_length': '255'}),
98+
'path': ('django.db.models.fields.CharField', [], {'max_length': '255'}),
99+
'project_url': ('django.db.models.fields.URLField', [], {'max_length': '200', 'blank': 'True'}),
100+
'pub_date': ('django.db.models.fields.DateTimeField', [], {'auto_now_add': 'True', 'blank': 'True'}),
101+
'repo': ('django.db.models.fields.CharField', [], {'max_length': '100', 'blank': 'True'}),
102+
'repo_type': ('django.db.models.fields.CharField', [], {'default': "'git'", 'max_length': '10'}),
103+
'skip': ('django.db.models.fields.BooleanField', [], {'default': 'False'}),
104+
'slug': ('django.db.models.fields.SlugField', [], {'max_length': '50', 'db_index': 'True'}),
105+
'status': ('django.db.models.fields.PositiveSmallIntegerField', [], {'default': '1'}),
106+
'suffix': ('django.db.models.fields.CharField', [], {'default': "'.rst'", 'max_length': '10'}),
107+
'theme': ('django.db.models.fields.CharField', [], {'default': "'default'", 'max_length': '20'}),
108+
'user': ('django.db.models.fields.related.ForeignKey', [], {'related_name': "'projects'", 'to': "orm['auth.User']"}),
109+
'version': ('django.db.models.fields.CharField', [], {'max_length': '100', 'blank': 'True'})
110+
}
111+
}
112+
113+
complete_apps = ['projects']

0 commit comments

Comments
 (0)