From a73847454f542c505710e26cbe4d56fa66989fc1 Mon Sep 17 00:00:00 2001 From: Roman Hergenreder Date: Fri, 5 Dec 2025 09:21:33 +0100 Subject: [PATCH] minor additions --- git-dumper.py | 9 ++++++++- sqli.py | 5 +++-- 2 files changed, 11 insertions(+), 3 deletions(-) diff --git a/git-dumper.py b/git-dumper.py index 3cbf774..d043d7d 100755 --- a/git-dumper.py +++ b/git-dumper.py @@ -530,7 +530,7 @@ if __name__ == '__main__': description='Dump a git repository from a website.') parser.add_argument('url', metavar='URL', help='url') - parser.add_argument('directory', metavar='DIR', + parser.add_argument('--directory', metavar='DIR', default=None, type=str, help='output directory') parser.add_argument('--proxy', help='use the specified proxy') @@ -577,6 +577,13 @@ if __name__ == '__main__': parser.error('invalid proxy') # output directory + if args.directory is None: + parsed_url = urllib.parse.urlparse(args.url) + if not parsed_url or not parsed_url.hostname: + parser.error('no output directory given and cannot derive from URL') + else: + args.directory = parsed_url.hostname + if not os.path.exists(args.directory): os.makedirs(args.directory) diff --git a/sqli.py b/sqli.py index ff3a33d..c9ee93f 100644 --- a/sqli.py +++ b/sqli.py @@ -5,12 +5,13 @@ import string class SQLi(ABC): @staticmethod - def build_query(column: str|list, table=None, condition=None, offset=None): + def build_query(column: str|list, table=None, condition=None, offset=None, limit=1): column = column if isinstance(column, str) else ",".join(column) condition = "" if not condition else f" WHERE {condition}" offset = "" if offset is None else f" OFFSET {offset}" table = "" if not table else f" FROM {table}" - return f"SELECT {column}{table}{condition} LIMIT 1{offset}" + limit = "" if limit is None else f" LIMIT {limit}" + return f"SELECT {column}{table}{condition}{limit}{offset}" def extract_multiple_ints(self, column: str, table=None, condition=None, verbose=False): row_count = self.extract_int(f"COUNT({column})", table=table, condition=condition, verbose=verbose)