Ejemplos de GritsConsumer en Python

Ejemplo n.º 1

Archivo: test__tools_grits_consumer.py Proyecto: ecohealthalliance/grits-net-consume

class TestGritsConsumer(unittest.TestCase):
    def setUp(self):
        self.cmd = GritsConsumer()
        self.validFile = open(os.path.join(_SCRIPT_DIR, 'data/GlobalDirectsSample_20150728.csv'))
        self.invalidFile = open(os.path.join(_SCRIPT_DIR, 'data/GlobalDirectsSample_20150728.txt'))

    def test_is_valid_extension(self):
        self.assertEqual(True, self.cmd.is_valid_file_type(self.validFile))

    def test_is_invalid_extension(self):
        self.assertEqual(False, self.cmd.is_valid_file_type(self.invalidFile))

Ejemplo n.º 2

Archivo: test__tools_grits_consumer.py Proyecto: ecohealthalliance/grits-net-consume

class TestGritsConsumer(unittest.TestCase):
    def setUp(self):
        self.cmd = GritsConsumer()
        self.validFile = open(
            os.path.join(_SCRIPT_DIR, 'data/GlobalDirectsSample_20150728.csv'))
        self.invalidFile = open(
            os.path.join(_SCRIPT_DIR, 'data/GlobalDirectsSample_20150728.txt'))

    def test_is_valid_extension(self):
        self.assertEqual(True, self.cmd.is_valid_file_type(self.validFile))

    def test_is_invalid_extension(self):
        self.assertEqual(False, self.cmd.is_valid_file_type(self.invalidFile))

Ejemplo n.º 3

Archivo: grits_consume.py Proyecto: ecohealthalliance/grits-net-consume

#!/usr/bin/env python
import os
import sys
import glob
import logging

from conf import settings
from tools.grits_consumer import GritsConsumer
""" wrapper for running GritsConsumer """
if __name__ == '__main__':

    cmd = GritsConsumer()

    def get_lastest_csv():
        """ get the most recent filename, sorted by date with extension .csv """
        latest_csv = None
        data_dir = os.path.join(os.getcwd() + settings._DATA_DIR)
        try:
            latest_csv = max(glob.iglob(data_dir + '*.[Cc][Ss][Vv]'),
                             key=os.path.getctime)
        except Exception as e:
            logging.error(e)
            logging.error('Have you run the FTP download?')
            sys.exit(1)
        return lastest_csv

    if len(sys.argv) <= 1:
        lastest_csv = get_lastest_csv()
        cmd.run('--type', 'FlightGlobal', lastest_csv)
    else:
        cmd.run()

Ejemplo n.º 4

Archivo: test__tools_grits_consumer.py Proyecto: ecohealthalliance/grits-net-consume

 def setUp(self):
     self.cmd = GritsConsumer()
     self.validFile = open(os.path.join(_SCRIPT_DIR, 'data/GlobalDirectsSample_20150728.csv'))
     self.invalidFile = open(os.path.join(_SCRIPT_DIR, 'data/GlobalDirectsSample_20150728.txt'))

Ejemplo n.º 5

Archivo: test__tools_grits_consumer.py Proyecto: ecohealthalliance/grits-net-consume

 def setUp(self):
     self.cmd = GritsConsumer()
     self.validFile = open(
         os.path.join(_SCRIPT_DIR, 'data/GlobalDirectsSample_20150728.csv'))
     self.invalidFile = open(
         os.path.join(_SCRIPT_DIR, 'data/GlobalDirectsSample_20150728.txt'))

Ejemplo n.º 6

Archivo: grits_consume.py Proyecto: ecohealthalliance/grits-net-consume

#!/usr/bin/env python
import os
import sys
import glob
import logging

from conf import settings
from tools.grits_consumer import GritsConsumer


""" wrapper for running GritsConsumer """
if __name__ == "__main__":

    cmd = GritsConsumer()

    def get_lastest_csv():
        """ get the most recent filename, sorted by date with extension .csv """
        latest_csv = None
        data_dir = os.path.join(os.getcwd() + settings._DATA_DIR)
        try:
            latest_csv = max(glob.iglob(data_dir + "*.[Cc][Ss][Vv]"), key=os.path.getctime)
        except Exception as e:
            logging.error(e)
            logging.error("Have you run the FTP download?")
            sys.exit(1)
        return lastest_csv

    if len(sys.argv) <= 1:
        lastest_csv = get_lastest_csv()
        cmd.run("--type", "FlightGlobal", lastest_csv)
    else: