aljaz/recon-pipeline
1
0
Fork 0
mirror of https://github.com/aljazceru/recon-pipeline.git synced 2025-12-21 00:04:24 +01:00
Code Issues Packages Projects Releases Wiki Activity

fixed lint getting caught on CI but not locally

Browse Source
This commit is contained in:
epi
2020-05-11 22:21:37 -05:00
parent c8bb606ecc
commit 9a64f4c50b
4 changed files with 16 additions and 18 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
pipeline/models/db_manager.py
View File

@@ -46,7 +46,7 @@ class DBManager:
self.session.add(item)
self.session.commit()
except (sqlite3.IntegrityError, exc.IntegrityError):
print(ansi.style(f"[-] unique key constraint handled, moving on...", fg="bright_white"))
print(ansi.style("[-] unique key constraint handled, moving on...", fg="bright_white"))
self.session.rollback()
def get_or_create_target_by_ip_or_hostname(self, ip_or_host):
@@ -86,15 +86,13 @@ class DBManager:
def get_all_ipv4_addresses(self) -> list:
""" Simple helper to return all ipv4 addresses from Target records """
return [
x[0] for x in self.session.query(IPAddress.ipv4_address).filter(IPAddress.ipv4_address != None)
] # noqa: E711
query = self.session.query(IPAddress.ipv4_address).filter(IPAddress.ipv4_address != None) # noqa: E711
return [x[0] for x in query]
def get_all_ipv6_addresses(self) -> list:
""" Simple helper to return all ipv6 addresses from Target records """
return [
x[0] for x in self.session.query(IPAddress.ipv6_address).filter(IPAddress.ipv6_address != None)
] # noqa: E711
query = self.session.query(IPAddress.ipv6_address).filter(IPAddress.ipv6_address != None) # noqa: E711
return [x[0] for x in query]
def close(self):
""" Simple helper to close the database session """