bashclub/piler
1
0
Fork 0
mirror of https://bitbucket.org/jsuto/piler.git synced 2024-12-25 07:20:11 +01:00
Code Issues Packages Projects Releases Wiki Activity

Improved imapfetch.py

Browse Source 
Signed-off-by: Janos SUTO <sj@acts.hu>
This commit is contained in:
Janos SUTO 2020-10-27 22:11:21 +01:00
parent 1d615a1967
commit f7f91c0855
1 changed files with 13 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
util/imapfetch.py
View File

@ -66,7 +66,7 @@ def process_folder(conn, folder):
if opts['id']: if opts['id']:
cursor = opts['db'].cursor() cursor = opts['db'].cursor()
data = (ST_RUNNING, n, opts['id']) data = (ST_RUNNING, n, opts['id'])
cursor.execute("UPDATE import SET status=%s, total = total + %s WHERE id=%s", data) cursor.execute("UPDATE import SET status=%s, total=total+%s WHERE id=%s", data)
opts['db'].commit() opts['db'].commit()
rc, data = conn.search(None, 'ALL') rc, data = conn.search(None, 'ALL')
@ -91,11 +91,16 @@ def main():
default="junk,trash,spam,draft") default="junk,trash,spam,draft")
parser.add_argument("-f", "--folders", type=str, parser.add_argument("-f", "--folders", type=str,
help="Comma separated list of IMAP folders to download") help="Comma separated list of IMAP folders to download")
parser.add_argument("-i", "--import-from-table", help="Read imap conn data from import table", action='store_true') parser.add_argument("-d", "--dir", help="directory to chdir",
default="/var/piler/imap")
parser.add_argument("-i", "--import-from-table", action='store_true',
help="Read imap conn data from import table")
parser.add_argument("-v", "--verbose", help="verbose mode", action='store_true') parser.add_argument("-v", "--verbose", help="verbose mode", action='store_true')
args = parser.parse_args() args = parser.parse_args()
os.chdir(args.dir)
if not bool(args.import_from_table or args.server): if not bool(args.import_from_table or args.server):
print("Please specify either --import-from-table or --server <imap host>") print("Please specify either --import-from-table or --server <imap host>")
sys.exit(1) sys.exit(1)
@ -158,10 +163,15 @@ def main():
conn.close() conn.close()
if opts['db']: if opts['db']:
subprocess.call(["pilerimport", "-d", "/var/piler/imap", "-r"]) if opts['id']:
subprocess.call(["pilerimport",
"-d", args.dir,
"-r",
"-T", str(opts['id'])])
opts['db'].close() opts['db'].close()
print("Processed {} messages".format(opts['counter'])) print("Processed {} messages".format(opts['counter']))
if __name__ == "__main__": if __name__ == "__main__":
main() main()