Exemplo n.º 1
0
def initDb(redisHost, redisPort, redisDb):
    '''
        Load geodis packaged city data into the redis DB
        '''

    print "Initializing Redis data"

    try:
        scriptDir = os.path.dirname(__file__)
        zipFile = "/opt/geodis/data/zipcode.csv"
        importer = ZIPImporter(zipFile, redisHost, redisPort, redisDb)

        if not importer.runImport():
            print >> sys.stderr, "Could not import geonames database..."
            sys.exit(1)

        cityFile = "/opt/geodis/data/cities1000.txt"
        adminFile = "/opt/geodis/data/admin1Codes.txt"
        importer = GeonamesImporter(",".join([cityFile, adminFile]), redisHost,
                                    redisPort, redisDb)

        if not importer.runImport():
            print >> sys.stderr, "Could not import geonames database..."
            sys.exit(1)

    except:
        traceback.print_exc(file=sys.stderr)
        sys.exit(1)
Exemplo n.º 2
0
 def test3_ImportZIP(self):
     importer = ZIPImporter(atdir('data/zipcodes.csv'), self.redisHost, self.redisPort, self.redisDB)
     self.assertGreater(importer.runImport(), 0, 'Could not import zipcodes csv')