script/lib/iri_tweet/export_twitter_alchemy.py
changeset 84 b1029aa40ec3
parent 83 4a759c70e40f
child 122 4c3a15877f80
equal deleted inserted replaced
83:4a759c70e40f 84:b1029aa40ec3
    61         return None
    61         return None
    62 
    62 
    63 def get_options():
    63 def get_options():
    64     parser = OptionParser()
    64     parser = OptionParser()
    65     parser.add_option("-f", "--file", dest="filename",
    65     parser.add_option("-f", "--file", dest="filename",
    66                       help="write export to file", metavar="FILE", default="project_enmi.ldt")
    66                       help="write export to file", metavar="FILE", default="project.ldt")
    67     parser.add_option("-d", "--database", dest="database",
    67     parser.add_option("-d", "--database", dest="database",
    68                       help="Input database", metavar="DATABASE")
    68                       help="Input database", metavar="DATABASE")
    69     parser.add_option("-s", "--start-date", dest="start_date",
    69     parser.add_option("-s", "--start-date", dest="start_date",
    70                       help="start date", metavar="START_DATE")
    70                       help="start date", metavar="START_DATE")
    71     parser.add_option("-e", "--end-date", dest="end_date",
    71     parser.add_option("-e", "--end-date", dest="end_date",
    97     
    97     
    98     
    98     
    99     set_logging_options(parser)
    99     set_logging_options(parser)
   100 
   100 
   101     
   101     
   102     return parser.parse_args()
   102     return parser.parse_args() + (parser,)
   103 
   103 
   104 
   104 
   105 if __name__ == "__main__" :
   105 if __name__ == "__main__" :
   106 
   106 
   107     (options, args) = get_options()
   107     (options, args, parser) = get_options()
   108         
   108     
   109     set_logging(options)
   109     set_logging(options)
   110         
   110         
   111     logging.debug("OPTIONS : " + repr(options))
   111     logging.debug("OPTIONS : " + repr(options))
   112 
   112     
       
   113     if len(sys.argv) == 1 or options.database is None:
       
   114         parser.print_help()
       
   115         sys.exit(1)
       
   116     
   113     engine, metadata = setup_database('sqlite:///'+options.database, echo=((options.verbose-options.quiet)>0), create_all = False)        
   117     engine, metadata = setup_database('sqlite:///'+options.database, echo=((options.verbose-options.quiet)>0), create_all = False)        
   114     
   118     
   115     Session = sessionmaker()
   119     Session = sessionmaker()
   116     conn = engine.connect()
   120     conn = engine.connect()
   117     try :
   121     try :