Revision 775d5ad7

/dev/null
1
# encoding: utf-8
2

  
3
from south.v2 import DataMigration
4

  
5
from astakos.im.settings import SERVICES
6

  
7
import logging
8

  
9
logger = logging.getLogger(__name__)
10

  
11
class Migration(DataMigration):
12

  
13
    def forwards(self, orm):
14
        "Write your forwards methods here."
15

  
16
        try:
17
            default = orm.AstakosGroup.objects.get(name='default')
18
        except orm.AstakosGroup.DoesNotExist:
19
            return
20

  
21
        def create_policies(args):
22
            sn, dict = args
23
            url = dict.get('url')
24
            resources = dict.get('resources') or ()
25
            s, created = orm.Service.objects.get_or_create(
26
                name=sn,
27
                defaults={'url': url}
28
            )
29
            
30
            for r in resources:
31
                try:
32
                    rn = r.pop('name', '')
33
                    uplimit = r.pop('uplimit', None)
34
                    r, created = orm.Resource.objects.get_or_create(
35
                        service=s,
36
                        name=rn,
37
                        defaults=r)
38
                except Exception, e:
39
                    print "Cannot create resource ", rn
40
                    continue
41
                else:
42
                    q, created = orm.AstakosGroupQuota.objects.get_or_create(
43
                        group=default,
44
                        resource=r,
45
                        defaults={
46
                            'uplimit':uplimit,
47
                        }
48
                    )
49
        map(create_policies, SERVICES.iteritems())
50

  
51
    def backwards(self, orm):
52
        try:
53
            default = orm.AstakosGroup.objects.get(name='default')
54
        except orm.AstakosGroup.DoesNotExist:
55
            return
56

  
57
        def destroy_policies(args):
58
            sn, dict = args
59
            url = dict.get('url')
60
            resources = dict.get('resources') or ()
61
            for r in resources:
62
                rn = r.get('name', '')
63
                try:
64
                    q = orm.AstakosGroupQuota.objects.get(
65
                        group=default,
66
                        resource__name=rn)
67
                    q.delete()
68
                    q = orm.Resource.objects.get(service__name=sn, name=rn)
69
                    q.delete()
70
                except Exception, e:
71
                    print "Cannot create resource ", rn
72
                    continue
73

  
74
        map(destroy_policies, SERVICES.iteritems())
75

  
76
    models = {
77
        'auth.group': {
78
            'Meta': {'object_name': 'Group'},
79
            'id': ('django.db.models.fields.AutoField', [], {'primary_key': 'True'}),
80
            'name': ('django.db.models.fields.CharField', [], {'unique': 'True', 'max_length': '80'}),
81
            'permissions': ('django.db.models.fields.related.ManyToManyField', [], {'to': "orm['auth.Permission']", 'symmetrical': 'False', 'blank': 'True'})
82
        },
83
        'auth.permission': {
84
            'Meta': {'ordering': "('content_type__app_label', 'content_type__model', 'codename')", 'unique_together': "(('content_type', 'codename'),)", 'object_name': 'Permission'},
85
            'codename': ('django.db.models.fields.CharField', [], {'max_length': '100'}),
86
            'content_type': ('django.db.models.fields.related.ForeignKey', [], {'to': "orm['contenttypes.ContentType']"}),
87
            'id': ('django.db.models.fields.AutoField', [], {'primary_key': 'True'}),
88
            'name': ('django.db.models.fields.CharField', [], {'max_length': '50'})
89
        },
90
        'auth.user': {
91
            'Meta': {'object_name': 'User'},
92
            'date_joined': ('django.db.models.fields.DateTimeField', [], {'default': 'datetime.datetime.now'}),
93
            'email': ('django.db.models.fields.EmailField', [], {'max_length': '75', 'blank': 'True'}),
94
            'first_name': ('django.db.models.fields.CharField', [], {'max_length': '30', 'blank': 'True'}),
95
            'groups': ('django.db.models.fields.related.ManyToManyField', [], {'to': "orm['auth.Group']", 'symmetrical': 'False', 'blank': 'True'}),
96
            'id': ('django.db.models.fields.AutoField', [], {'primary_key': 'True'}),
97
            'is_active': ('django.db.models.fields.BooleanField', [], {'default': 'True'}),
98
            'is_staff': ('django.db.models.fields.BooleanField', [], {'default': 'False'}),
99
            'is_superuser': ('django.db.models.fields.BooleanField', [], {'default': 'False'}),
100
            'last_login': ('django.db.models.fields.DateTimeField', [], {'default': 'datetime.datetime.now'}),
101
            'last_name': ('django.db.models.fields.CharField', [], {'max_length': '30', 'blank': 'True'}),
102
            'password': ('django.db.models.fields.CharField', [], {'max_length': '128'}),
103
            'user_permissions': ('django.db.models.fields.related.ManyToManyField', [], {'to': "orm['auth.Permission']", 'symmetrical': 'False', 'blank': 'True'}),
104
            'username': ('django.db.models.fields.CharField', [], {'unique': 'True', 'max_length': '30'})
105
        },
106
        'contenttypes.contenttype': {
107
            'Meta': {'ordering': "('name',)", 'unique_together': "(('app_label', 'model'),)", 'object_name': 'ContentType', 'db_table': "'django_content_type'"},
108
            'app_label': ('django.db.models.fields.CharField', [], {'max_length': '100'}),
109
            'id': ('django.db.models.fields.AutoField', [], {'primary_key': 'True'}),
110
            'model': ('django.db.models.fields.CharField', [], {'max_length': '100'}),
111
            'name': ('django.db.models.fields.CharField', [], {'max_length': '100'})
112
        },
113
        'im.additionalmail': {
114
            'Meta': {'object_name': 'AdditionalMail'},
115
            'email': ('django.db.models.fields.EmailField', [], {'max_length': '75'}),
116
            'id': ('django.db.models.fields.AutoField', [], {'primary_key': 'True'}),
117
            'owner': ('django.db.models.fields.related.ForeignKey', [], {'to': "orm['im.AstakosUser']"})
118
        },
119
        'im.approvalterms': {
120
            'Meta': {'object_name': 'ApprovalTerms'},
121
            'date': ('django.db.models.fields.DateTimeField', [], {'default': 'datetime.datetime(2012, 11, 5, 12, 47, 39, 465319)', 'db_index': 'True'}),
122
            'id': ('django.db.models.fields.AutoField', [], {'primary_key': 'True'}),
123
            'location': ('django.db.models.fields.CharField', [], {'max_length': '255'})
124
        },
125
        'im.astakosgroup': {
126
            'Meta': {'object_name': 'AstakosGroup', '_ormbases': ['auth.Group']},
127
            'approval_date': ('django.db.models.fields.DateTimeField', [], {'null': 'True', 'blank': 'True'}),
128
            'creation_date': ('django.db.models.fields.DateTimeField', [], {'default': 'datetime.datetime(2012, 11, 5, 12, 47, 39, 456067)'}),
129
            'desc': ('django.db.models.fields.TextField', [], {'null': 'True'}),
130
            'estimated_participants': ('django.db.models.fields.PositiveIntegerField', [], {'null': 'True', 'blank': 'True'}),
131
            'expiration_date': ('django.db.models.fields.DateTimeField', [], {'null': 'True'}),
132
            'group_ptr': ('django.db.models.fields.related.OneToOneField', [], {'to': "orm['auth.Group']", 'unique': 'True', 'primary_key': 'True'}),
133
            'homepage': ('django.db.models.fields.URLField', [], {'max_length': '255', 'null': 'True', 'blank': 'True'}),
134
            'issue_date': ('django.db.models.fields.DateTimeField', [], {'null': 'True'}),
135
            'kind': ('django.db.models.fields.related.ForeignKey', [], {'to': "orm['im.GroupKind']"}),
136
            'max_participants': ('django.db.models.fields.PositiveIntegerField', [], {'null': 'True', 'blank': 'True'}),
137
            'moderation_enabled': ('django.db.models.fields.BooleanField', [], {'default': 'True'}),
138
            'policy': ('django.db.models.fields.related.ManyToManyField', [], {'symmetrical': 'False', 'to': "orm['im.Resource']", 'null': 'True', 'through': "orm['im.AstakosGroupQuota']", 'blank': 'True'})
139
        },
140
        'im.astakosgroupquota': {
141
            'Meta': {'unique_together': "(('resource', 'group'),)", 'object_name': 'AstakosGroupQuota'},
142
            'group': ('django.db.models.fields.related.ForeignKey', [], {'to': "orm['im.AstakosGroup']", 'blank': 'True'}),
143
            'id': ('django.db.models.fields.AutoField', [], {'primary_key': 'True'}),
144
            'limit': ('django.db.models.fields.PositiveIntegerField', [], {'null': 'True'}),
145
            'resource': ('django.db.models.fields.related.ForeignKey', [], {'to': "orm['im.Resource']"}),
146
            'uplimit': ('django.db.models.fields.BigIntegerField', [], {'null': 'True'})
147
        },
148
        'im.astakosuser': {
149
            'Meta': {'unique_together': "(('provider', 'third_party_identifier'),)", 'object_name': 'AstakosUser', '_ormbases': ['auth.User']},
150
            'activation_sent': ('django.db.models.fields.DateTimeField', [], {'null': 'True', 'blank': 'True'}),
151
            'affiliation': ('django.db.models.fields.CharField', [], {'max_length': '255', 'blank': 'True'}),
152
            'astakos_groups': ('django.db.models.fields.related.ManyToManyField', [], {'to': "orm['im.AstakosGroup']", 'symmetrical': 'False', 'through': "orm['im.Membership']", 'blank': 'True'}),
153
            'auth_token': ('django.db.models.fields.CharField', [], {'max_length': '32', 'null': 'True', 'blank': 'True'}),
154
            'auth_token_created': ('django.db.models.fields.DateTimeField', [], {'null': 'True'}),
155
            'auth_token_expires': ('django.db.models.fields.DateTimeField', [], {'null': 'True'}),
156
            'date_signed_terms': ('django.db.models.fields.DateTimeField', [], {'null': 'True', 'blank': 'True'}),
157
            'disturbed_quota': ('django.db.models.fields.BooleanField', [], {'default': 'False', 'db_index': 'True'}),
158
            'email_verified': ('django.db.models.fields.BooleanField', [], {'default': 'False'}),
159
            'has_credits': ('django.db.models.fields.BooleanField', [], {'default': 'False'}),
160
            'has_signed_terms': ('django.db.models.fields.BooleanField', [], {'default': 'False'}),
161
            'invitations': ('django.db.models.fields.IntegerField', [], {'default': '100'}),
162
            'is_verified': ('django.db.models.fields.BooleanField', [], {'default': 'False'}),
163
            'level': ('django.db.models.fields.IntegerField', [], {'default': '0'}),
164
            'owner': ('django.db.models.fields.related.ManyToManyField', [], {'symmetrical': 'False', 'related_name': "'owner'", 'null': 'True', 'to': "orm['im.AstakosGroup']"}),
165
            'policy': ('django.db.models.fields.related.ManyToManyField', [], {'to': "orm['im.Resource']", 'null': 'True', 'through': "orm['im.AstakosUserQuota']", 'symmetrical': 'False'}),
166
            'provider': ('django.db.models.fields.CharField', [], {'max_length': '255', 'blank': 'True'}),
167
            'third_party_identifier': ('django.db.models.fields.CharField', [], {'max_length': '255', 'null': 'True', 'blank': 'True'}),
168
            'updated': ('django.db.models.fields.DateTimeField', [], {}),
169
            'user_ptr': ('django.db.models.fields.related.OneToOneField', [], {'to': "orm['auth.User']", 'unique': 'True', 'primary_key': 'True'})
170
        },
171
        'im.astakosuserquota': {
172
            'Meta': {'unique_together': "(('resource', 'user'),)", 'object_name': 'AstakosUserQuota'},
173
            'id': ('django.db.models.fields.AutoField', [], {'primary_key': 'True'}),
174
            'limit': ('django.db.models.fields.PositiveIntegerField', [], {'null': 'True'}),
175
            'resource': ('django.db.models.fields.related.ForeignKey', [], {'to': "orm['im.Resource']"}),
176
            'uplimit': ('django.db.models.fields.BigIntegerField', [], {'null': 'True'}),
177
            'user': ('django.db.models.fields.related.ForeignKey', [], {'to': "orm['im.AstakosUser']"})
178
        },
179
        'im.emailchange': {
180
            'Meta': {'object_name': 'EmailChange'},
181
            'activation_key': ('django.db.models.fields.CharField', [], {'unique': 'True', 'max_length': '40', 'db_index': 'True'}),
182
            'id': ('django.db.models.fields.AutoField', [], {'primary_key': 'True'}),
183
            'new_email_address': ('django.db.models.fields.EmailField', [], {'max_length': '75'}),
184
            'requested_at': ('django.db.models.fields.DateTimeField', [], {'default': 'datetime.datetime(2012, 11, 5, 12, 47, 39, 466946)'}),
185
            'user': ('django.db.models.fields.related.ForeignKey', [], {'related_name': "'emailchange_user'", 'unique': 'True', 'to': "orm['im.AstakosUser']"})
186
        },
187
        'im.groupkind': {
188
            'Meta': {'object_name': 'GroupKind'},
189
            'id': ('django.db.models.fields.AutoField', [], {'primary_key': 'True'}),
190
            'name': ('django.db.models.fields.CharField', [], {'unique': 'True', 'max_length': '255', 'db_index': 'True'})
191
        },
192
        'im.invitation': {
193
            'Meta': {'object_name': 'Invitation'},
194
            'code': ('django.db.models.fields.BigIntegerField', [], {'db_index': 'True'}),
195
            'consumed': ('django.db.models.fields.DateTimeField', [], {'null': 'True', 'blank': 'True'}),
196
            'created': ('django.db.models.fields.DateTimeField', [], {'auto_now_add': 'True', 'blank': 'True'}),
197
            'id': ('django.db.models.fields.AutoField', [], {'primary_key': 'True'}),
198
            'inviter': ('django.db.models.fields.related.ForeignKey', [], {'related_name': "'invitations_sent'", 'null': 'True', 'to': "orm['im.AstakosUser']"}),
199
            'is_consumed': ('django.db.models.fields.BooleanField', [], {'default': 'False'}),
200
            'realname': ('django.db.models.fields.CharField', [], {'max_length': '255'}),
201
            'username': ('django.db.models.fields.CharField', [], {'unique': 'True', 'max_length': '255'})
202
        },
203
        'im.membership': {
204
            'Meta': {'unique_together': "(('person', 'group'),)", 'object_name': 'Membership'},
205
            'date_joined': ('django.db.models.fields.DateField', [], {'db_index': 'True', 'null': 'True', 'blank': 'True'}),
206
            'date_requested': ('django.db.models.fields.DateField', [], {'default': 'datetime.datetime(2012, 11, 5, 12, 47, 39, 462857)', 'blank': 'True'}),
207
            'group': ('django.db.models.fields.related.ForeignKey', [], {'to': "orm['im.AstakosGroup']"}),
208
            'id': ('django.db.models.fields.AutoField', [], {'primary_key': 'True'}),
209
            'person': ('django.db.models.fields.related.ForeignKey', [], {'to': "orm['im.AstakosUser']"})
210
        },
211
        'im.resource': {
212
            'Meta': {'object_name': 'Resource'},
213
            'desc': ('django.db.models.fields.TextField', [], {'null': 'True'}),
214
            'group': ('django.db.models.fields.CharField', [], {'max_length': '255', 'null': 'True'}),
215
            'id': ('django.db.models.fields.AutoField', [], {'primary_key': 'True'}),
216
            'meta': ('django.db.models.fields.related.ManyToManyField', [], {'to': "orm['im.ResourceMetadata']", 'symmetrical': 'False'}),
217
            'name': ('django.db.models.fields.CharField', [], {'unique': 'True', 'max_length': '255', 'db_index': 'True'}),
218
            'service': ('django.db.models.fields.related.ForeignKey', [], {'to': "orm['im.Service']"}),
219
            'unit': ('django.db.models.fields.CharField', [], {'max_length': '255', 'null': 'True'})
220
        },
221
        'im.resourcemetadata': {
222
            'Meta': {'object_name': 'ResourceMetadata'},
223
            'id': ('django.db.models.fields.AutoField', [], {'primary_key': 'True'}),
224
            'key': ('django.db.models.fields.CharField', [], {'unique': 'True', 'max_length': '255', 'db_index': 'True'}),
225
            'value': ('django.db.models.fields.CharField', [], {'max_length': '255'})
226
        },
227
        'im.service': {
228
            'Meta': {'object_name': 'Service'},
229
            'auth_token': ('django.db.models.fields.CharField', [], {'max_length': '32', 'null': 'True', 'blank': 'True'}),
230
            'auth_token_created': ('django.db.models.fields.DateTimeField', [], {'null': 'True'}),
231
            'auth_token_expires': ('django.db.models.fields.DateTimeField', [], {'null': 'True'}),
232
            'icon': ('django.db.models.fields.FilePathField', [], {'max_length': '100', 'blank': 'True'}),
233
            'id': ('django.db.models.fields.AutoField', [], {'primary_key': 'True'}),
234
            'name': ('django.db.models.fields.CharField', [], {'unique': 'True', 'max_length': '255', 'db_index': 'True'}),
235
            'url': ('django.db.models.fields.FilePathField', [], {'max_length': '100'})
236
        }
237
    }
238

  
239
    complete_apps = ['im']

Also available in: Unified diff