code.grnet.gr Git - flowspy/blob - peers/management/commands/fetch_networks.py

   1 # -*- coding: utf-8 -*- vim:encoding=utf-8:
   2 # vim: tabstop=4:shiftwidth=4:softtabstop=4:expandtab
   3
   4 # Copyright (C) 2010-2014 GRNET S.A.
   5 #
   6 # This program is free software: you can redistribute it and/or modify
   7 # it under the terms of the GNU General Public License as published by
   8 # the Free Software Foundation, either version 3 of the License, or
   9 # (at your option) any later version.
  10 #
  11 # This program is distributed in the hope that it will be useful,
  12 # but WITHOUT ANY WARRANTY; without even the implied warranty of
  13 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  14 # GNU General Public License for more details.
  15 #
  16 # You should have received a copy of the GNU General Public License
  17 # along with this program.  If not, see <http://www.gnu.org/licenses/>.
  18 #
  19
  20 from django.core.management.base import BaseCommand, CommandError
  21 from flowspy.peers.models import *
  22
  23 class Command(BaseCommand):
  24     args = ''
  25     help = 'Fetches networks for each peer in database'
  26
  27     def handle(self, *args, **options):
  28         for p in Peer.objects.all():
  29             self.stdout.write('Fetching networks for %s...' % p.peer_name.encode('utf8'))
  30             p.fill_networks()
  31             self.stdout.write('done\n')
  32         self.stdout.write('Finished!\n')