Changeset 9057 for src/django_gheat/gheat/management/commands/add_random_points.py

Timestamp:

Apr 15, 2011, 9:27:48 PM (14 years ago)

Author:

rick

Message:

Make it a actual embedded django command.

File:

• src/django_gheat/gheat/management/commands/add_random_points.py (moved) (moved from src/django_gheat/gheat/random_db_init.py ) (2 diffs)

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

@@ -4 +4 @@
 #
 # Rick van der Zwet <info@rickvanderzwet.nl>
-import settings
-from django.core.management import setup_environ
-setup_environ(settings)
-
-
 from django.db import connection, transaction
+from django.core.management.base import BaseCommand
 from gheat.models import *
+from optparse import OptionParser, make_option
 import datetime
 import os
@@ -52 +49 @@
 
 
+class Command(BaseCommand):
+    option_list = BaseCommand.option_list + (
+        make_option("-c", "--count", dest="count", type="int", default=100, help="Number of points to add"),
+    )
+
+    def handle(self, *args, **options):
+        add_random_measurements(options['count'])
+
 if __name__ == "__main__":
   # Awefull hack for argument parsing