9711b721b2cf3b55e9f83b03ebaf3f9535df4052
[audio.git] /
1 # encoding: utf-8
2 import datetime
3 from hashlib import sha1
4 from south.db import db
5 from south.v2 import SchemaMigration
6 from django.db import models
7
8
9 def sha1_file(f):
10     sha = sha1()
11     for piece in iter(lambda: f.read(1024*1024), ''):
12         sha.update(piece)
13     return sha.hexdigest()
14
15
16 class Migration(SchemaMigration):
17
18     def forwards(self, orm):
19         
20         # Adding field 'Audiobook.source_sha1'
21         db.add_column('archive_audiobook', 'source_sha1', self.gf('django.db.models.fields.CharField')(default='', max_length=40), keep_default=False)
22
23         # Adding field 'Audiobook.translator'
24         db.add_column('archive_audiobook', 'translator', self.gf('django.db.models.fields.CharField')(max_length=255, null=True, blank=True), keep_default=False)
25
26         if not db.dry_run:
27             for obj in orm.Audiobook.objects.all():
28                 try:
29                     f = open(obj.source_file.path)
30                 except ValueError, e:
31                     print "Audiobook has no source file"
32                     pass
33                 else:
34                     obj.source_sha1 = sha1_file(f)
35                     f.close()
36                     obj.save()
37
38
39     def backwards(self, orm):
40         
41         # Deleting field 'Audiobook.source_sha1'
42         db.delete_column('archive_audiobook', 'source_sha1')
43
44         # Deleting field 'Audiobook.translator'
45         db.delete_column('archive_audiobook', 'translator')
46
47
48     models = {
49         'archive.audiobook': {
50             'Meta': {'ordering': "('title',)", 'object_name': 'Audiobook'},
51             'artist': ('django.db.models.fields.CharField', [], {'max_length': '255'}),
52             'conductor': ('django.db.models.fields.CharField', [], {'max_length': '255'}),
53             'date': ('django.db.models.fields.CharField', [], {'max_length': '255'}),
54             'encoded_by': ('django.db.models.fields.CharField', [], {'max_length': '255'}),
55             'id': ('django.db.models.fields.AutoField', [], {'primary_key': 'True'}),
56             'modified': ('django.db.models.fields.DateTimeField', [], {'null': 'True'}),
57             'mp3_file': ('django.db.models.fields.files.FileField', [], {'max_length': '100', 'null': 'True'}),
58             'mp3_published': ('django.db.models.fields.DateTimeField', [], {'null': 'True'}),
59             'mp3_published_tags': ('jsonfield.fields.JSONField', [], {'null': 'True'}),
60             'mp3_status': ('django.db.models.fields.SmallIntegerField', [], {'null': 'True'}),
61             'mp3_tags': ('jsonfield.fields.JSONField', [], {'null': 'True'}),
62             'mp3_task': ('django.db.models.fields.CharField', [], {'max_length': '64', 'null': 'True'}),
63             'ogg_file': ('django.db.models.fields.files.FileField', [], {'max_length': '100', 'null': 'True'}),
64             'ogg_published': ('django.db.models.fields.DateTimeField', [], {'null': 'True'}),
65             'ogg_published_tags': ('jsonfield.fields.JSONField', [], {'null': 'True'}),
66             'ogg_status': ('django.db.models.fields.SmallIntegerField', [], {'null': 'True'}),
67             'ogg_tags': ('jsonfield.fields.JSONField', [], {'null': 'True'}),
68             'ogg_task': ('django.db.models.fields.CharField', [], {'max_length': '64', 'null': 'True'}),
69             'project': ('django.db.models.fields.related.ForeignKey', [], {'to': "orm['archive.Project']"}),
70             'source_file': ('django.db.models.fields.files.FileField', [], {'max_length': '100'}),
71             'source_sha1': ('django.db.models.fields.CharField', [], {'max_length': '40'}),
72             'title': ('django.db.models.fields.CharField', [], {'max_length': '255'}),
73             'translator': ('django.db.models.fields.CharField', [], {'max_length': '255', 'null': 'True', 'blank': 'True'}),
74             'url': ('django.db.models.fields.URLField', [], {'max_length': '255'})
75         },
76         'archive.project': {
77             'Meta': {'ordering': "('name',)", 'object_name': 'Project'},
78             'id': ('django.db.models.fields.AutoField', [], {'primary_key': 'True'}),
79             'name': ('django.db.models.fields.CharField', [], {'unique': 'True', 'max_length': '128', 'db_index': 'True'}),
80             'sponsors': ('django.db.models.fields.TextField', [], {'null': 'True', 'blank': 'True'})
81         }
82     }
83
84     complete_apps = ['archive']