From 7b55220589d0d58716366b9991e50a7dcabbac01 Mon Sep 17 00:00:00 2001 From: agatha Date: Sun, 22 Sep 2024 15:02:22 -0400 Subject: [PATCH] feat: proxies are updated if already in database --- main.py | 24 ++++++++++++++++-------- 1 file changed, 16 insertions(+), 8 deletions(-) diff --git a/main.py b/main.py index 0a227c0..678962a 100644 --- a/main.py +++ b/main.py @@ -59,14 +59,22 @@ def main(): # Save to DB session = session_factory.create_session() - proxy = Proxy( - host=proxy.split(':')[0], - port=int(proxy.split(':')[1]), - egress_ip=ip, - date_added=datetime.now(), - date_validated=datetime.now(), - ) - session.add(proxy) + existing_proxy = session.query(Proxy).filter_by(host=proxy.split(':')[0], + port=int(proxy.split(':')[1])).first() + + if existing_proxy: + existing_proxy.date_validated = datetime.now() + existing_proxy.egress_ip = ip + else: + new_proxy = Proxy( + host=proxy.split(':')[0], + port=int(proxy.split(':')[1]), + egress_ip=ip, + date_added=datetime.now(), + date_validated=datetime.now(), + ) + session.add(new_proxy) + session.commit() # Create output directory if it does not exist