예제 #1
0
def main():
    print ("Start of process...")
    
    #configuration parameters
    start = time.time()
    txt_path = config.conf["txt_path"]
    save_path = config.conf["save_path"]
    database_path = config.conf["database_path"]
    db_type = config.conf["db_type"]
    
    #use plaintext abstract factory
    txt_abs_factory = plaintext_abs_factory.plaintext_factory()
    txt_factory = txt_abs_factory.get_plaintext(txt_path)
    txt_prod = txt_factory.get_txt()
    URLs = txt_prod.get_lines()
    #use images abstract factory
    init_pics = Pic_Handler(URLs,save_path)
    pics_info = init_pics.pic_info()
    init_pics.pic_save()
    #use database abstract factory
    db_abs_factory = database_abs_factory.database_factory()
    db_factory = db_abs_factory.db_abs_fac(db_type,database_path)
    db_prod = db_factory.process_sql()
    db_prod.create_db()
    db_prod.populate_db(pics_info)
    
    print ("End of process" )
    end = time.time()
    print ("Process time: ", str(round(end-start,2)),"sec")
    raw_input("Press enter to exit.")       
class Test_pic_handler(unittest.TestCase):

    def setUp(self):
        self.__url = ["http://bilder.bild.de/fotos/lachsack-36229038/Bild/1.bild.jpg"]
        self.__save_path = "test_files"
        self.__pic = Pic_Handler(self.__url,self.__save_path)
        
        
    def test_pic_save(self):
        '''
        Method to test if the Save_Pic class returns 1
        '''
        self.assertEqual( self.__pic.pic_save(), 1)
        
    def test_pic_info(self):
        '''
        Method to test if the Save_Pic class returns the right information.
        '''
        pic_info =  self.__pic.pic_info()
        
        self.assertNotIn("/", pic_info[0][0])
        self.assertNotIn(":", pic_info[0][0])
        self.assertIn("http://", pic_info[0][1])
        self.assertEqual(type (pic_info[0][2]), str)