Rooba
/
pupy
mirror of https://github.com/n1nj4sec/pupy.git
1
1
Fork 0
Code Issues Packages Projects Releases Wiki Activity

adding expandvars function

This commit is contained in:
Alessandro ZANNI 2016-10-15 01:32:38 +02:00
parent d013ee7b57
commit 65e7d7d0a0
1 changed files with 5 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
pupy/modules/upload.py
View File

@ -15,7 +15,8 @@ class UploaderScript(PupyModule):
self.arg_parser.add_argument('local_file', metavar='<local_path>', completer=path_completer)
self.arg_parser.add_argument('remote_file', metavar='<remote_path>')
def run(self, args):
if args.remote_file.endswith('.'):
args.remote_file = args.remote_file.replace('.', args.local_file.split(os.sep)[-1])
upload(self.client.conn, args.local_file, args.remote_file)
self.success("file local:%s uploaded to remote:%s"%(args.local_file, args.remote_file))
dst = self.client.conn.modules['os.path'].expandvars(args.remote_file)
if dst.endswith('.'):
dst = dst.replace('.', args.local_file.split(os.sep)[-1])
upload(self.client.conn, args.local_file, dst)
self.success("file local:%s uploaded to remote:%s"%(args.local_file, dst))