Statistics
| Branch: | Tag: | Revision:

root / snf-astakos-app / astakos / im / migrations / 0017_populate_resource_data.py @ 775d5ad7

History | View | Annotate | Download (15.2 kB)

1 304acb60 Olga Brani
# encoding: utf-8
2 304acb60 Olga Brani
3 304acb60 Olga Brani
from south.v2 import DataMigration
4 304acb60 Olga Brani
5 16cf0f73 Sofia Papagiannaki
from astakos.im.settings import SERVICES
6 16cf0f73 Sofia Papagiannaki
7 16cf0f73 Sofia Papagiannaki
import logging
8 16cf0f73 Sofia Papagiannaki
9 16cf0f73 Sofia Papagiannaki
logger = logging.getLogger(__name__)
10 9a06d96f Olga Brani
11 304acb60 Olga Brani
class Migration(DataMigration):
12 304acb60 Olga Brani
13 304acb60 Olga Brani
    def forwards(self, orm):
14 16cf0f73 Sofia Papagiannaki
        "Write your forwards methods here."
15 16cf0f73 Sofia Papagiannaki
16 16cf0f73 Sofia Papagiannaki
        try:
17 16cf0f73 Sofia Papagiannaki
            default = orm.AstakosGroup.objects.get(name='default')
18 16cf0f73 Sofia Papagiannaki
        except orm.AstakosGroup.DoesNotExist:
19 16cf0f73 Sofia Papagiannaki
            return
20 16cf0f73 Sofia Papagiannaki
21 16cf0f73 Sofia Papagiannaki
        def create_policies(args):
22 16cf0f73 Sofia Papagiannaki
            sn, dict = args
23 16cf0f73 Sofia Papagiannaki
            url = dict.get('url')
24 16cf0f73 Sofia Papagiannaki
            resources = dict.get('resources') or ()
25 16cf0f73 Sofia Papagiannaki
            s, created = orm.Service.objects.get_or_create(
26 16cf0f73 Sofia Papagiannaki
                name=sn,
27 16cf0f73 Sofia Papagiannaki
                defaults={'url': url}
28 16cf0f73 Sofia Papagiannaki
            )
29 16cf0f73 Sofia Papagiannaki
            
30 16cf0f73 Sofia Papagiannaki
            for r in resources:
31 16cf0f73 Sofia Papagiannaki
                try:
32 16cf0f73 Sofia Papagiannaki
                    rn = r.pop('name', '')
33 16cf0f73 Sofia Papagiannaki
                    uplimit = r.pop('uplimit', None)
34 16cf0f73 Sofia Papagiannaki
                    r, created = orm.Resource.objects.get_or_create(
35 16cf0f73 Sofia Papagiannaki
                        service=s,
36 16cf0f73 Sofia Papagiannaki
                        name=rn,
37 16cf0f73 Sofia Papagiannaki
                        defaults=r)
38 16cf0f73 Sofia Papagiannaki
                except Exception, e:
39 16cf0f73 Sofia Papagiannaki
                    print "Cannot create resource ", rn
40 16cf0f73 Sofia Papagiannaki
                    continue
41 16cf0f73 Sofia Papagiannaki
                else:
42 16cf0f73 Sofia Papagiannaki
                    q, created = orm.AstakosGroupQuota.objects.get_or_create(
43 16cf0f73 Sofia Papagiannaki
                        group=default,
44 16cf0f73 Sofia Papagiannaki
                        resource=r,
45 16cf0f73 Sofia Papagiannaki
                        defaults={
46 16cf0f73 Sofia Papagiannaki
                            'uplimit':uplimit,
47 16cf0f73 Sofia Papagiannaki
                        }
48 16cf0f73 Sofia Papagiannaki
                    )
49 16cf0f73 Sofia Papagiannaki
        map(create_policies, SERVICES.iteritems())
50 9a06d96f Olga Brani
51 304acb60 Olga Brani
    def backwards(self, orm):
52 16cf0f73 Sofia Papagiannaki
        try:
53 16cf0f73 Sofia Papagiannaki
            default = orm.AstakosGroup.objects.get(name='default')
54 16cf0f73 Sofia Papagiannaki
        except orm.AstakosGroup.DoesNotExist:
55 16cf0f73 Sofia Papagiannaki
            return
56 16cf0f73 Sofia Papagiannaki
57 16cf0f73 Sofia Papagiannaki
        def destroy_policies(args):
58 16cf0f73 Sofia Papagiannaki
            sn, dict = args
59 16cf0f73 Sofia Papagiannaki
            url = dict.get('url')
60 16cf0f73 Sofia Papagiannaki
            resources = dict.get('resources') or ()
61 16cf0f73 Sofia Papagiannaki
            for r in resources:
62 16cf0f73 Sofia Papagiannaki
                rn = r.get('name', '')
63 16cf0f73 Sofia Papagiannaki
                try:
64 16cf0f73 Sofia Papagiannaki
                    q = orm.AstakosGroupQuota.objects.get(
65 16cf0f73 Sofia Papagiannaki
                        group=default,
66 16cf0f73 Sofia Papagiannaki
                        resource__name=rn)
67 16cf0f73 Sofia Papagiannaki
                    q.delete()
68 16cf0f73 Sofia Papagiannaki
                    q = orm.Resource.objects.get(service__name=sn, name=rn)
69 16cf0f73 Sofia Papagiannaki
                    q.delete()
70 16cf0f73 Sofia Papagiannaki
                except Exception, e:
71 16cf0f73 Sofia Papagiannaki
                    print "Cannot create resource ", rn
72 16cf0f73 Sofia Papagiannaki
                    continue
73 16cf0f73 Sofia Papagiannaki
74 16cf0f73 Sofia Papagiannaki
        map(destroy_policies, SERVICES.iteritems())
75 9a06d96f Olga Brani
76 304acb60 Olga Brani
    models = {
77 304acb60 Olga Brani
        'auth.group': {
78 304acb60 Olga Brani
            'Meta': {'object_name': 'Group'},
79 304acb60 Olga Brani
            'id': ('django.db.models.fields.AutoField', [], {'primary_key': 'True'}),
80 304acb60 Olga Brani
            'name': ('django.db.models.fields.CharField', [], {'unique': 'True', 'max_length': '80'}),
81 304acb60 Olga Brani
            'permissions': ('django.db.models.fields.related.ManyToManyField', [], {'to': "orm['auth.Permission']", 'symmetrical': 'False', 'blank': 'True'})
82 304acb60 Olga Brani
        },
83 304acb60 Olga Brani
        'auth.permission': {
84 304acb60 Olga Brani
            'Meta': {'ordering': "('content_type__app_label', 'content_type__model', 'codename')", 'unique_together': "(('content_type', 'codename'),)", 'object_name': 'Permission'},
85 304acb60 Olga Brani
            'codename': ('django.db.models.fields.CharField', [], {'max_length': '100'}),
86 304acb60 Olga Brani
            'content_type': ('django.db.models.fields.related.ForeignKey', [], {'to': "orm['contenttypes.ContentType']"}),
87 304acb60 Olga Brani
            'id': ('django.db.models.fields.AutoField', [], {'primary_key': 'True'}),
88 304acb60 Olga Brani
            'name': ('django.db.models.fields.CharField', [], {'max_length': '50'})
89 304acb60 Olga Brani
        },
90 304acb60 Olga Brani
        'auth.user': {
91 304acb60 Olga Brani
            'Meta': {'object_name': 'User'},
92 304acb60 Olga Brani
            'date_joined': ('django.db.models.fields.DateTimeField', [], {'default': 'datetime.datetime.now'}),
93 304acb60 Olga Brani
            'email': ('django.db.models.fields.EmailField', [], {'max_length': '75', 'blank': 'True'}),
94 304acb60 Olga Brani
            'first_name': ('django.db.models.fields.CharField', [], {'max_length': '30', 'blank': 'True'}),
95 304acb60 Olga Brani
            'groups': ('django.db.models.fields.related.ManyToManyField', [], {'to': "orm['auth.Group']", 'symmetrical': 'False', 'blank': 'True'}),
96 304acb60 Olga Brani
            'id': ('django.db.models.fields.AutoField', [], {'primary_key': 'True'}),
97 304acb60 Olga Brani
            'is_active': ('django.db.models.fields.BooleanField', [], {'default': 'True'}),
98 304acb60 Olga Brani
            'is_staff': ('django.db.models.fields.BooleanField', [], {'default': 'False'}),
99 304acb60 Olga Brani
            'is_superuser': ('django.db.models.fields.BooleanField', [], {'default': 'False'}),
100 304acb60 Olga Brani
            'last_login': ('django.db.models.fields.DateTimeField', [], {'default': 'datetime.datetime.now'}),
101 304acb60 Olga Brani
            'last_name': ('django.db.models.fields.CharField', [], {'max_length': '30', 'blank': 'True'}),
102 304acb60 Olga Brani
            'password': ('django.db.models.fields.CharField', [], {'max_length': '128'}),
103 304acb60 Olga Brani
            'user_permissions': ('django.db.models.fields.related.ManyToManyField', [], {'to': "orm['auth.Permission']", 'symmetrical': 'False', 'blank': 'True'}),
104 304acb60 Olga Brani
            'username': ('django.db.models.fields.CharField', [], {'unique': 'True', 'max_length': '30'})
105 304acb60 Olga Brani
        },
106 304acb60 Olga Brani
        'contenttypes.contenttype': {
107 304acb60 Olga Brani
            'Meta': {'ordering': "('name',)", 'unique_together': "(('app_label', 'model'),)", 'object_name': 'ContentType', 'db_table': "'django_content_type'"},
108 304acb60 Olga Brani
            'app_label': ('django.db.models.fields.CharField', [], {'max_length': '100'}),
109 304acb60 Olga Brani
            'id': ('django.db.models.fields.AutoField', [], {'primary_key': 'True'}),
110 304acb60 Olga Brani
            'model': ('django.db.models.fields.CharField', [], {'max_length': '100'}),
111 304acb60 Olga Brani
            'name': ('django.db.models.fields.CharField', [], {'max_length': '100'})
112 304acb60 Olga Brani
        },
113 304acb60 Olga Brani
        'im.additionalmail': {
114 304acb60 Olga Brani
            'Meta': {'object_name': 'AdditionalMail'},
115 304acb60 Olga Brani
            'email': ('django.db.models.fields.EmailField', [], {'max_length': '75'}),
116 304acb60 Olga Brani
            'id': ('django.db.models.fields.AutoField', [], {'primary_key': 'True'}),
117 304acb60 Olga Brani
            'owner': ('django.db.models.fields.related.ForeignKey', [], {'to': "orm['im.AstakosUser']"})
118 304acb60 Olga Brani
        },
119 304acb60 Olga Brani
        'im.approvalterms': {
120 304acb60 Olga Brani
            'Meta': {'object_name': 'ApprovalTerms'},
121 304acb60 Olga Brani
            'date': ('django.db.models.fields.DateTimeField', [], {'default': 'datetime.datetime(2012, 8, 9, 11, 14, 9, 289091)', 'db_index': 'True'}),
122 304acb60 Olga Brani
            'id': ('django.db.models.fields.AutoField', [], {'primary_key': 'True'}),
123 304acb60 Olga Brani
            'location': ('django.db.models.fields.CharField', [], {'max_length': '255'})
124 304acb60 Olga Brani
        },
125 304acb60 Olga Brani
        'im.astakosgroup': {
126 304acb60 Olga Brani
            'Meta': {'object_name': 'AstakosGroup', '_ormbases': ['auth.Group']},
127 304acb60 Olga Brani
            'approval_date': ('django.db.models.fields.DateTimeField', [], {'null': 'True', 'blank': 'True'}),
128 304acb60 Olga Brani
            'creation_date': ('django.db.models.fields.DateTimeField', [], {'default': 'datetime.datetime(2012, 8, 9, 11, 14, 9, 283154)'}),
129 304acb60 Olga Brani
            'desc': ('django.db.models.fields.TextField', [], {'null': 'True'}),
130 304acb60 Olga Brani
            'estimated_participants': ('django.db.models.fields.PositiveIntegerField', [], {'null': 'True'}),
131 304acb60 Olga Brani
            'expiration_date': ('django.db.models.fields.DateTimeField', [], {'null': 'True'}),
132 304acb60 Olga Brani
            'group_ptr': ('django.db.models.fields.related.OneToOneField', [], {'to': "orm['auth.Group']", 'unique': 'True', 'primary_key': 'True'}),
133 304acb60 Olga Brani
            'issue_date': ('django.db.models.fields.DateTimeField', [], {'null': 'True'}),
134 304acb60 Olga Brani
            'kind': ('django.db.models.fields.related.ForeignKey', [], {'to': "orm['im.GroupKind']"}),
135 304acb60 Olga Brani
            'moderation_enabled': ('django.db.models.fields.BooleanField', [], {'default': 'True'}),
136 304acb60 Olga Brani
            'policy': ('django.db.models.fields.related.ManyToManyField', [], {'symmetrical': 'False', 'to': "orm['im.Resource']", 'null': 'True', 'through': "orm['im.AstakosGroupQuota']", 'blank': 'True'})
137 304acb60 Olga Brani
        },
138 304acb60 Olga Brani
        'im.astakosgroupquota': {
139 304acb60 Olga Brani
            'Meta': {'unique_together': "(('resource', 'group'),)", 'object_name': 'AstakosGroupQuota'},
140 304acb60 Olga Brani
            'group': ('django.db.models.fields.related.ForeignKey', [], {'to': "orm['im.AstakosGroup']", 'blank': 'True'}),
141 304acb60 Olga Brani
            'id': ('django.db.models.fields.AutoField', [], {'primary_key': 'True'}),
142 304acb60 Olga Brani
            'limit': ('django.db.models.fields.PositiveIntegerField', [], {}),
143 304acb60 Olga Brani
            'resource': ('django.db.models.fields.related.ForeignKey', [], {'to': "orm['im.Resource']"})
144 304acb60 Olga Brani
        },
145 304acb60 Olga Brani
        'im.astakosuser': {
146 304acb60 Olga Brani
            'Meta': {'unique_together': "(('provider', 'third_party_identifier'),)", 'object_name': 'AstakosUser', '_ormbases': ['auth.User']},
147 304acb60 Olga Brani
            'activation_sent': ('django.db.models.fields.DateTimeField', [], {'null': 'True', 'blank': 'True'}),
148 304acb60 Olga Brani
            'affiliation': ('django.db.models.fields.CharField', [], {'max_length': '255', 'blank': 'True'}),
149 304acb60 Olga Brani
            'astakos_groups': ('django.db.models.fields.related.ManyToManyField', [], {'to': "orm['im.AstakosGroup']", 'symmetrical': 'False', 'through': "orm['im.Membership']", 'blank': 'True'}),
150 304acb60 Olga Brani
            'auth_token': ('django.db.models.fields.CharField', [], {'max_length': '32', 'null': 'True', 'blank': 'True'}),
151 304acb60 Olga Brani
            'auth_token_created': ('django.db.models.fields.DateTimeField', [], {'null': 'True'}),
152 304acb60 Olga Brani
            'auth_token_expires': ('django.db.models.fields.DateTimeField', [], {'null': 'True'}),
153 304acb60 Olga Brani
            'date_signed_terms': ('django.db.models.fields.DateTimeField', [], {'null': 'True', 'blank': 'True'}),
154 304acb60 Olga Brani
            'email_verified': ('django.db.models.fields.BooleanField', [], {'default': 'False'}),
155 304acb60 Olga Brani
            'has_credits': ('django.db.models.fields.BooleanField', [], {'default': 'False'}),
156 304acb60 Olga Brani
            'has_signed_terms': ('django.db.models.fields.BooleanField', [], {'default': 'False'}),
157 304acb60 Olga Brani
            'invitations': ('django.db.models.fields.IntegerField', [], {'default': '100'}),
158 304acb60 Olga Brani
            'is_verified': ('django.db.models.fields.BooleanField', [], {'default': 'False'}),
159 304acb60 Olga Brani
            'level': ('django.db.models.fields.IntegerField', [], {'default': '0'}),
160 304acb60 Olga Brani
            'owner': ('django.db.models.fields.related.ManyToManyField', [], {'symmetrical': 'False', 'related_name': "'owner'", 'null': 'True', 'to': "orm['im.AstakosGroup']"}),
161 304acb60 Olga Brani
            'policy': ('django.db.models.fields.related.ManyToManyField', [], {'to': "orm['im.Resource']", 'null': 'True', 'through': "orm['im.AstakosUserQuota']", 'symmetrical': 'False'}),
162 304acb60 Olga Brani
            'provider': ('django.db.models.fields.CharField', [], {'max_length': '255', 'blank': 'True'}),
163 304acb60 Olga Brani
            'third_party_identifier': ('django.db.models.fields.CharField', [], {'max_length': '255', 'null': 'True', 'blank': 'True'}),
164 304acb60 Olga Brani
            'updated': ('django.db.models.fields.DateTimeField', [], {}),
165 304acb60 Olga Brani
            'user_ptr': ('django.db.models.fields.related.OneToOneField', [], {'to': "orm['auth.User']", 'unique': 'True', 'primary_key': 'True'})
166 304acb60 Olga Brani
        },
167 304acb60 Olga Brani
        'im.astakosuserquota': {
168 304acb60 Olga Brani
            'Meta': {'unique_together': "(('resource', 'user'),)", 'object_name': 'AstakosUserQuota'},
169 304acb60 Olga Brani
            'id': ('django.db.models.fields.AutoField', [], {'primary_key': 'True'}),
170 304acb60 Olga Brani
            'limit': ('django.db.models.fields.PositiveIntegerField', [], {}),
171 304acb60 Olga Brani
            'resource': ('django.db.models.fields.related.ForeignKey', [], {'to': "orm['im.Resource']"}),
172 304acb60 Olga Brani
            'user': ('django.db.models.fields.related.ForeignKey', [], {'to': "orm['im.AstakosUser']"})
173 304acb60 Olga Brani
        },
174 304acb60 Olga Brani
        'im.emailchange': {
175 304acb60 Olga Brani
            'Meta': {'object_name': 'EmailChange'},
176 304acb60 Olga Brani
            'activation_key': ('django.db.models.fields.CharField', [], {'unique': 'True', 'max_length': '40', 'db_index': 'True'}),
177 304acb60 Olga Brani
            'id': ('django.db.models.fields.AutoField', [], {'primary_key': 'True'}),
178 304acb60 Olga Brani
            'new_email_address': ('django.db.models.fields.EmailField', [], {'max_length': '75'}),
179 304acb60 Olga Brani
            'requested_at': ('django.db.models.fields.DateTimeField', [], {'default': 'datetime.datetime(2012, 8, 9, 11, 14, 9, 290713)'}),
180 304acb60 Olga Brani
            'user': ('django.db.models.fields.related.ForeignKey', [], {'related_name': "'emailchange_user'", 'unique': 'True', 'to': "orm['im.AstakosUser']"})
181 304acb60 Olga Brani
        },
182 304acb60 Olga Brani
        'im.groupkind': {
183 304acb60 Olga Brani
            'Meta': {'object_name': 'GroupKind'},
184 304acb60 Olga Brani
            'id': ('django.db.models.fields.AutoField', [], {'primary_key': 'True'}),
185 304acb60 Olga Brani
            'name': ('django.db.models.fields.CharField', [], {'unique': 'True', 'max_length': '255', 'db_index': 'True'})
186 304acb60 Olga Brani
        },
187 304acb60 Olga Brani
        'im.invitation': {
188 304acb60 Olga Brani
            'Meta': {'object_name': 'Invitation'},
189 304acb60 Olga Brani
            'code': ('django.db.models.fields.BigIntegerField', [], {'db_index': 'True'}),
190 304acb60 Olga Brani
            'consumed': ('django.db.models.fields.DateTimeField', [], {'null': 'True', 'blank': 'True'}),
191 304acb60 Olga Brani
            'created': ('django.db.models.fields.DateTimeField', [], {'auto_now_add': 'True', 'blank': 'True'}),
192 304acb60 Olga Brani
            'id': ('django.db.models.fields.AutoField', [], {'primary_key': 'True'}),
193 304acb60 Olga Brani
            'inviter': ('django.db.models.fields.related.ForeignKey', [], {'related_name': "'invitations_sent'", 'null': 'True', 'to': "orm['im.AstakosUser']"}),
194 304acb60 Olga Brani
            'is_consumed': ('django.db.models.fields.BooleanField', [], {'default': 'False'}),
195 304acb60 Olga Brani
            'realname': ('django.db.models.fields.CharField', [], {'max_length': '255'}),
196 304acb60 Olga Brani
            'username': ('django.db.models.fields.CharField', [], {'unique': 'True', 'max_length': '255'})
197 304acb60 Olga Brani
        },
198 304acb60 Olga Brani
        'im.membership': {
199 304acb60 Olga Brani
            'Meta': {'unique_together': "(('person', 'group'),)", 'object_name': 'Membership'},
200 304acb60 Olga Brani
            'date_joined': ('django.db.models.fields.DateField', [], {'db_index': 'True', 'null': 'True', 'blank': 'True'}),
201 304acb60 Olga Brani
            'date_requested': ('django.db.models.fields.DateField', [], {'default': 'datetime.datetime(2012, 8, 9, 11, 14, 9, 286925)', 'blank': 'True'}),
202 304acb60 Olga Brani
            'group': ('django.db.models.fields.related.ForeignKey', [], {'to': "orm['im.AstakosGroup']"}),
203 304acb60 Olga Brani
            'id': ('django.db.models.fields.AutoField', [], {'primary_key': 'True'}),
204 304acb60 Olga Brani
            'person': ('django.db.models.fields.related.ForeignKey', [], {'to': "orm['im.AstakosUser']"})
205 304acb60 Olga Brani
        },
206 304acb60 Olga Brani
        'im.resource': {
207 304acb60 Olga Brani
            'Meta': {'object_name': 'Resource'},
208 304acb60 Olga Brani
            'id': ('django.db.models.fields.AutoField', [], {'primary_key': 'True'}),
209 304acb60 Olga Brani
            'meta': ('django.db.models.fields.related.ManyToManyField', [], {'to': "orm['im.ResourceMetadata']", 'symmetrical': 'False'}),
210 304acb60 Olga Brani
            'name': ('django.db.models.fields.CharField', [], {'unique': 'True', 'max_length': '255', 'db_index': 'True'}),
211 304acb60 Olga Brani
            'service': ('django.db.models.fields.related.ForeignKey', [], {'to': "orm['im.Service']"})
212 304acb60 Olga Brani
        },
213 304acb60 Olga Brani
        'im.resourcemetadata': {
214 304acb60 Olga Brani
            'Meta': {'object_name': 'ResourceMetadata'},
215 304acb60 Olga Brani
            'id': ('django.db.models.fields.AutoField', [], {'primary_key': 'True'}),
216 304acb60 Olga Brani
            'key': ('django.db.models.fields.CharField', [], {'unique': 'True', 'max_length': '255', 'db_index': 'True'}),
217 304acb60 Olga Brani
            'value': ('django.db.models.fields.CharField', [], {'max_length': '255'})
218 304acb60 Olga Brani
        },
219 304acb60 Olga Brani
        'im.service': {
220 304acb60 Olga Brani
            'Meta': {'object_name': 'Service'},
221 304acb60 Olga Brani
            'auth_token': ('django.db.models.fields.CharField', [], {'max_length': '32', 'null': 'True', 'blank': 'True'}),
222 304acb60 Olga Brani
            'auth_token_created': ('django.db.models.fields.DateTimeField', [], {'null': 'True'}),
223 304acb60 Olga Brani
            'auth_token_expires': ('django.db.models.fields.DateTimeField', [], {'null': 'True'}),
224 304acb60 Olga Brani
            'icon': ('django.db.models.fields.FilePathField', [], {'max_length': '100', 'blank': 'True'}),
225 304acb60 Olga Brani
            'id': ('django.db.models.fields.AutoField', [], {'primary_key': 'True'}),
226 304acb60 Olga Brani
            'name': ('django.db.models.fields.CharField', [], {'unique': 'True', 'max_length': '255', 'db_index': 'True'}),
227 304acb60 Olga Brani
            'url': ('django.db.models.fields.FilePathField', [], {'max_length': '100'})
228 304acb60 Olga Brani
        }
229 304acb60 Olga Brani
    }
230 304acb60 Olga Brani
231 304acb60 Olga Brani
    complete_apps = ['im']