from flask import Flask, request, Response, redirect import json import datetime import requests from adapter import MailService from search import SearchQuery app = Flask(__name__, static_url_path='', static_folder='../../web-ui/app') client = None converter = None account = None loaded = False mail_service = MailService() def respond_json(entity): response = json.dumps(entity) return Response(response=response, mimetype="application/json") @app.route('/disabled_features') def disabled_features(): return respond_json([]) @app.route('/mails', methods=['POST']) def save_draft_or_send(): mail = request.json if mail['ident']: ident = mail_service.send(mail) else: ident = mail_service.save_draft(mail) return respond_json({'ident': ident}) @app.route('/mails', methods=['PUT']) def update_draft(): mail = request.json ident = mail_service.update_draft(mail) return respond_json({'ident': ident}) @app.route('/mails') def mails(): query = SearchQuery.compile(request.args.get('q', '')) page = request.args.get('p', '') window_size = request.args.get('w', '') fetched_mails = mail_service.mails(query, page, window_size) mails = [mail.__dict__ for mail in fetched_mails] response = { "stats": { "total": len(mails), "read": 0, "starred": 0, "replied": 0 }, "mails": mails } return respond_json(response) @app.route('/mail/', methods=['DELETE']) def delete_mails(mail_id): mail_service.delete_mail(mail_id) return respond_json(None) @app.route('/tags') def tags(): tags = mail_service.tagsset.all_tags() return respond_json([tag.__dict__ for tag in tags]) @app.route('/mail/') def mail(mail_id): return respond_json(mail_service.mail(mail_id).__dict__) @app.route('/mail//tags', methods=['POST']) def mail_tags(mail_id): new_tags = request.json['newtags'] mail_service.update_tags_for(mail_id, new_tags) return respond_json(request.json['newtags']) @app.route('/mail//read', methods=['POST']) def mark_mail_as_read(mail_id): mail_service.mark_as_read(mail_id) return "" @app.route('/contacts') def contacts(): contacts_query = request.args.get('q') return respond_json( {'contacts': mail_service.search_contacts(contacts_query)}) @app.route('/draft_reply_for/') def draft_reply_for(mail_id): mail = mail_service.draft_reply_for(mail_id) if mail: return respond_json(mail.__dict__) else: return respond_json(None) @app.route('/control/mailset//load', methods=['POST']) def load_mailset(mailset): import os from tarfile import TarFile mbox_root = os.path.join(os.environ['HOME'], 'mailsets') if not os.path.isdir(os.path.join(mbox_root)): os.mkdir(mbox_root) if len(os.listdir(mbox_root)) == 0: response = requests.get( 'https://example.wazokazi.is:8154/go/static/mediumtagged.tar.gz', verify=False) mbox_archive_path = os.path.join(mbox_root, 'mediumtagged.tar.gz') mbox_archive = open(mbox_archive_path, 'w') mbox_archive.write(response.content) mbox_archive.close() tarfile = TarFile(name=mbox_archive_path) tarfile.extractall(path=mbox_root) mail_service.load_mailset() return respond_json(None) @app.route('/') def index(): global loaded if not loaded: load_mailset('mediumtagged') loaded = True return app.send_static_file('index.html') def setup(): app.run(host="0.0.0.0", debug=True, port=4567) if __name__ == '__main__': setup()