cqrenet/relay
3
0
Fork 0
mirror of https://github.com/chatmail/relay.git synced 2026-05-10 16:04:37 +00:00
Code Issues Packages Projects Releases Wiki Activity

Store raw tokens instead of dictionaries in metadata

Browse Source
This commit is contained in:
link2xt
2024-03-24 02:04:26 +00:00
committed by holger krekel
parent 9fdf4fd2af
commit c44f4efced
1 changed files with 9 additions and 24 deletions
Download Patch File Download Diff File Expand all files Collapse all files

33
chatmaild/src/chatmaild/metadata.py
View File

@@ -1,6 +1,6 @@
import pwd import pwd
import pathlib from pathlib import Path
from queue import Queue from queue import Queue
from threading import Thread from threading import Thread
from socketserver import ( from socketserver import (
@@ -13,7 +13,6 @@ import sys
import logging import logging
import os import os
import requests import requests
import marshal
DICTPROXY_LOOKUP_CHAR = "L" DICTPROXY_LOOKUP_CHAR = "L"
@@ -29,33 +28,19 @@ class Notifier:
self.metadata_dir = metadata_dir self.metadata_dir = metadata_dir
self.to_notify_queue = Queue() self.to_notify_queue = Queue()
def get_metadata(self, guid): def set_token(self, guid, token):
guid_path = self.metadata_dir.joinpath(guid)
if guid_path.exists():
with guid_path.open("rb") as f:
return marshal.load(f)
return {}
def set_metadata(self, guid, guid_data):
guid_path = self.metadata_dir.joinpath(guid) guid_path = self.metadata_dir.joinpath(guid)
write_path = guid_path.with_suffix(".tmp") write_path = guid_path.with_suffix(".tmp")
with write_path.open("wb") as f: write_path.write_text(token)
marshal.dump(guid_data, f) write_path.rename(guid_path)
os.rename(write_path, guid_path)
def set_token(self, guid, token):
guid_data = self.get_metadata(guid)
guid_data["token"] = token
self.set_metadata(guid, guid_data)
def del_token(self, guid): def del_token(self, guid):
guid_data = self.get_metadata(guid) self.metadata_dir.joinpath(guid).unlink(missing_ok=True)
if "token" in guid_data:
del guid_data["token"]
self.set_metadata(guid, guid_data)
def get_token(self, guid): def get_token(self, guid):
return self.get_metadata(guid).get("token") guid_path = self.metadata_dir / guid
if guid_path.exists():
return guid_path.read_text()
def new_message_for_guid(self, guid): def new_message_for_guid(self, guid):
self.to_notify_queue.put(guid) self.to_notify_queue.put(guid)
@@ -151,7 +136,7 @@ def main():
# XXX config is not currently used # XXX config is not currently used
config = read_config(config) config = read_config(config)
metadata_dir = pathlib.Path(metadata_dir) metadata_dir = Path(metadata_dir)
if not metadata_dir.exists(): if not metadata_dir.exists():
metadata_dir.mkdir() metadata_dir.mkdir()
notifier = Notifier(metadata_dir) notifier = Notifier(metadata_dir)