bellingcat/auto-archiver
1
0
Fork 0
mirror of https://github.com/bellingcat/auto-archiver.git synced 2026-06-08 03:18:28 +03:00
Code Issues Packages Projects Releases Wiki Activity

Set the new session filename *before* copying

Browse Source 
Fixes a potential bug whereby if the copy fails for some reason, the 'cleanup' command would remove the original session file
This commit is contained in:
Patrick Robertson
2025-03-26 21:32:23 +04:00
parent 20bc80b9ef
commit 580de88366
1 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
src/auto_archiver/modules/telethon_extractor/telethon_extractor.py
View File

@@ -35,10 +35,10 @@ class TelethonExtractor(Extractor):
self.session_file = self.session_file.removesuffix(".session")
# make a copy of the session that is used exclusively with this archiver instance
new_session_file = os.path.join("secrets/", f"telethon-{time.strftime('%Y-%m-%d')}{random_str(8)}")
logger.debug(f"Making a copy of the session file {self.session_file}.session to {new_session_file}.session")
shutil.copy(f"{self.session_file}.session", f"{new_session_file}.session")
self.session_file = new_session_file
old_session_file = f"{self.session_file}.session"
self.session_file = os.path.join("secrets/", f"telethon-{time.strftime('%Y-%m-%d')}{random_str(8)}")
logger.debug(f"Making a copy of the session file {old_session_file} to {self.session_file}.session")
shutil.copy(old_session_file, f"{self.session_file}.session")
# initiate the client
self.client = TelegramClient(self.session_file, self.api_id, self.api_hash)