diff --git a/pywiUpload.py b/pywiUpload.py index fc9592d..5d57649 100755 --- a/pywiUpload.py +++ b/pywiUpload.py @@ -59,7 +59,7 @@ def main(): "main program" # command line options - usage = "usage: %prog [options]" + usage = "usage: %prog [options] inputdir outputdir" version = "version %s, %s" % (__version__, __date__) parser = OptionParser(usage=usage, version="%prog "+version) @@ -67,11 +67,23 @@ def main(): parser.add_option("-c", "--config", dest="config", help="specify an alternative config file") - parser.add_option("-i", "--imgpath", dest="imgpath", - help="specify images path") - (options, args) = parser.parse_args() + if len(args) != 2: + parser.print_help() + sys.exit() + + input_dir = args[0] + output_dir = args[1] + + if not os.path.isdir(input_dir): + print "Directory %s does not exist." % input_dir + sys.exit(1) + + if not os.path.isdir(output_dir): + print "Create %s" % output_dir + os.makedirs(output_dir) + # read params from config file config_file = options.config if options.config \ else os.path.join(sys.path[0], 'pywiUpload.conf') @@ -81,9 +93,7 @@ def main(): # create gallery gallery = Gallery(params) - - path = options.imgpath if options.imgpath else os.getcwd() - (galleryname, out_filelist) = gallery.create_gallery(path) + out_filelist = gallery.create_gallery(input_dir, output_dir) # upload if raw_input("Upload images to your FTP server ? (y/[n]) ") == 'y': diff --git a/pywiupload/ftp.py b/pywiupload/ftp.py index 80af4bc..80cc01c 100644 --- a/pywiupload/ftp.py +++ b/pywiupload/ftp.py @@ -36,6 +36,7 @@ class FtpUpload: self.ftp = ftplib.FTP(host, user, password) self.ftp.cwd(basedir) + # FIXME: galname def upload(self, imgfilelist, galname, thumb_dir, bigimg_dir=None): "Upload images to a FTP server" diff --git a/pywiupload/image.py b/pywiupload/image.py index 5933125..101656a 100644 --- a/pywiupload/image.py +++ b/pywiupload/image.py @@ -34,7 +34,6 @@ class Gallery: def __init__(self, params): self.filepath = "" - self.galname = "" self.params = params def getpath(self, pathname): @@ -42,18 +41,14 @@ class Gallery: return os.path.join(self.filepath, self.params[pathname]) \ if self.params.has_key(pathname) else "" - def create_gallery(self, path): + def create_gallery(self, input_dir, output_dir): "create image gallery" - imglist = get_filelist(path, self.params['fileExtList']) - print "Found %i images in %s" % (len(imglist), path) + imglist = get_filelist(input_dir, self.params['fileExtList']) + print "Found %i images in %s" % (len(imglist), input_dir) - while self.galname == "": - self.galname = raw_input('Enter gallery name: ') + self.filepath = output_dir - self.filepath = os.path.join(os.path.dirname(imglist[0]), - self.galname) - - print "Create output dir ..." + print "Create output directories ..." try: os.makedirs(self.getpath('thumb_dir')) except OSError: @@ -65,7 +60,7 @@ class Gallery: except OSError: pass - return (self.galname, self.process_images(imglist)) + return self.process_images(imglist) def process_images(self, imglist): "prepare images" diff --git a/runtests.py b/runtests.py index 547ce37..fdb474b 100755 --- a/runtests.py +++ b/runtests.py @@ -41,9 +41,8 @@ if __name__ == '__main__': # create gallery gallery = Gallery(params) - (galleryname, filelist) = gallery.create_gallery("./test") + filelist = gallery.create_gallery("./test", "./test/output") - print "\n:: Output for gallery %s" % galleryname print "images : %s" % filelist[0] print "thumbnails : %s" % filelist[1] print "big images : %s" % filelist[2]