import logging import re import traceback from datetime import datetime from portal.base import UploadHandler, ContractCreator, REGEX_EMAIL from portal.model import db, UploadRegular class UploadHandlerRegular(UploadHandler): def __init__(self, **kwargs): super().__init__(**kwargs) self.contract_creator = ContractCreator(base_path=self.get_uploads_subdir('contracts'), template_path='contract/regular.html') def generate_upload_contract_pdf(self, upload_metadata): base = self.get_uploads_subdir('contracts') form_data = upload_metadata['form_data'] files_table_str = [] for file_name in upload_metadata['file_names']: files_table_str.append('') files_table_str.append(file_name) files_table_str.append('') files_table_str = ''.join(files_table_str) data = { 'ime_priimek': form_data['ime'], 'naslov': form_data.get('naslov', ''), 'posta': form_data.get('posta', ''), 'kontakt_narocnik': self.config['CONTRACT_CLIENT_CONTACT'], 'kontakt_imetnikpravic': form_data['ime'], 'files_table_str': files_table_str } self.contract_creator.create_pdf(upload_metadata['contract_file'], data) @staticmethod def store_metadata(upload_metadata, corpus_name): timestamp = datetime.fromtimestamp(upload_metadata['timestamp']) form_data = upload_metadata['form_data'] file_hashes = upload_metadata['file_hashes_dict'] sorted_f_hashes = list(file_hashes.values()) sorted_f_hashes.sort() try: model_obj = UploadRegular( upload_hash=upload_metadata['upload_id'], timestamp=timestamp, name=form_data['ime'], org=form_data.get('podjetje'), address=form_data.get('naslov'), zipcode=form_data.get('posta'), email=form_data['email'], file_contract=upload_metadata['contract_file'], upload_file_hashes=sorted_f_hashes, corpus_name=corpus_name ) db.session.add(model_obj) db.session.commit() except Exception: traceback.print_exc() def handle_upload(self, request, corpus_name): err = self.check_upload_request(request) if err: return err, 400 err = self.check_form(request.form) if err: return err, 400 # Parse request. upload_metadata = self.extract_upload_metadata(corpus_name, request) logging.info('Upload with id "{}" supplied form data: {}'.format(upload_metadata['upload_id'], str(upload_metadata['form_data']))) # Generate contract PDF file based on the uploads metadata. self.generate_upload_contract_pdf(upload_metadata) # Store uploaded files to disk. self.store_datafiles(request.files, upload_metadata) # Store metadata to database. self.store_metadata(upload_metadata, corpus_name) # Send confirmation mail along with the contract to the submitted email address. self.send_confirm_mail(upload_metadata, attach_contract_file=True) return 'Uspešno ste oddali datotek(e). Št. datotek: {}'.format(len(request.files)) @staticmethod def check_form(form): ime = form.get('ime') podjetje = form.get('podjetje') naslov = form.get('naslov') posta = form.get('posta') email = form.get('email') telefon = form.get('telefon') if not ime: return 'Prazno polje za ime.' if len(ime) > 100: return 'Predolgo ime.' if podjetje and len(podjetje) > 100: return 'Predolgo ime institucije.' if not email: return 'Prazno polje za elektronsko pošto.' if len(email) > 100: return 'Predolgi email naslov' elif not re.search(REGEX_EMAIL, email): return 'Email napačnega formata.' if telefon and len(telefon) > 100: return 'Predolga telefonska št.' if naslov and len(naslov) > 100: return 'Predolg naslov.' if posta and len(posta) > 100: return 'Predolga pošta' return None