Changeset 9637 for src/django_gheat/gheat

Timestamp:

Aug 30, 2011, 8:10:12 PM (13 years ago)

Author:

rick

Message:

Make sure we store the file we process.

Location:

src/django_gheat/gheat

Files:

• admin.py (modified) (1 diff)
• management/commands/import_datafile.py (modified) (2 diffs)
• models.py (modified) (1 diff)

src/django_gheat/gheat/admin.py

@@ -24 +24 @@
 admin.site.register(Organization, OrganizationAdmin)
 
+class MeetBestandAdmin(admin.ModelAdmin):
+   list_display = ['meetrondje', 'bestand', 'is_imported']
+admin.site.register(MeetBestand, MeetBestandAdmin)
+
 admin.site.register(Apparatuur)
 admin.site.register(Gebruiker)

src/django_gheat/gheat/management/commands/import_datafile.py

@@ -10 +10 @@
 #
 from _mysql_exceptions import OperationalError
+from django.core.files import File
 from django.core.management.base import BaseCommand,CommandError
 from django.db import connection, transaction
@@ -267 +268 @@
         }
       logger.info('Meetrondje: %s', mr)
+      meetbestand = MeetBestand(meetrondje=mr,is_imported=True)
+      meetbestand.bestand.save(os.path.basename(filename),File(open(filename)))
+      meetbestand.save()
+
       fh = open_file(filename)
       if 'ns1' in filename:

src/django_gheat/gheat/models.py

@@ -146 +146 @@
   class Meta:
     verbose_name_plural = 'MeetBestanden'
+  def __unicode__(self):
+    return "%s - %s" % (self.meetrondje, self.bestand)