# -*- coding: UTF-8 from datetime import datetime from globaleaks.db.migrations.update import MigrationBase from globaleaks.handlers.admin.operation import db_reset_smtp_settings from globaleaks.models import Model from globaleaks.models.config import ConfigFactory from globaleaks.models.enums import * from globaleaks.models.properties import * from globaleaks.utils.utility import datetime_now, datetime_never, datetime_null class InternalTip_v_52(Model): __tablename__ = 'internaltip' id = Column(UnicodeText(36), primary_key=True, default=uuid4) tid = Column(Integer, default=1, nullable=False) creation_date = Column(DateTime, default=datetime_now, nullable=False) update_date = Column(DateTime, default=datetime_now, nullable=False) context_id = Column(UnicodeText(36), nullable=False) progressive = Column(Integer, default=0, nullable=False) https = Column(Boolean, default=False, nullable=False) mobile = Column(Boolean, default=False, nullable=False) total_score = Column(Integer, default=0, nullable=False) expiration_date = Column(DateTime, default=datetime_never, nullable=False) enable_two_way_comments = Column(Boolean, default=True, nullable=False) enable_two_way_messages = Column(Boolean, default=True, nullable=False) enable_attachments = Column(Boolean, default=True, nullable=False) enable_whistleblower_identity = Column(Boolean, default=False, nullable=False) label = Column(UnicodeText, default='', nullable=False) wb_last_access = Column(DateTime, default=datetime_now, nullable=False) status = Column(UnicodeText(36), nullable=True) substatus = Column(UnicodeText(36), nullable=True) crypto_tip_pub_key = Column(UnicodeText(56), default='', nullable=False) class ReceiverTip_v_52(Model): __tablename__ = 'receivertip' id = Column(UnicodeText(36), primary_key=True, default=uuid4) internaltip_id = Column(UnicodeText(36), nullable=False) receiver_id = Column(UnicodeText(36), nullable=False) last_access = Column(DateTime, default=datetime_null, nullable=False) label = Column(UnicodeText, default='', nullable=False) can_access_whistleblower_identity = Column(Boolean, default=False, nullable=True) new = Column(Boolean, default=True, nullable=False) enable_notifications = Column(Boolean, default=True, nullable=False) crypto_tip_prv_key = Column(UnicodeText(84), default='', nullable=False) class Subscriber_v_52(Model): __tablename__ = 'signup' id = Column(Integer, primary_key=True) tid = Column(Integer, nullable=False) subdomain = Column(UnicodeText, unique=True, nullable=False) language = Column(UnicodeText(12), nullable=False) name = Column(UnicodeText, nullable=False) surname = Column(UnicodeText, nullable=False) role = Column(UnicodeText, default='', nullable=False) phone = Column(UnicodeText, default='', nullable=False) email = Column(UnicodeText, nullable=False) use_case = Column(UnicodeText, default='', nullable=False) use_case_other = Column(UnicodeText, default='', nullable=False) organization_name = Column(UnicodeText, default='', nullable=False) organization_type = Column(UnicodeText, default='', nullable=False) organization_location4 = Column(UnicodeText, default='', nullable=False) activation_token = Column(UnicodeText, unique=True, nullable=True) client_ip_address = Column(UnicodeText, default='', nullable=False) client_user_agent = Column(UnicodeText, default='', nullable=False) registration_date = Column(DateTime, default=datetime_now, nullable=False) tos1 = Column(UnicodeText, default='', nullable=False) tos2 = Column(UnicodeText, default='', nullable=False) class Tenant_v_52(Model): __tablename__ = 'tenant' id = Column(Integer, primary_key=True, nullable=False) label = Column(UnicodeText, default='', nullable=False) active = Column(Boolean, default=True, nullable=False) creation_date = Column(DateTime, default=datetime_now, nullable=False) subdomain = Column(UnicodeText, default='', nullable=False) class User_v_52(Model): __tablename__ = 'user' id = Column(UnicodeText(36), primary_key=True, default=uuid4) tid = Column(Integer, default=1, nullable=False) creation_date = Column(DateTime, default=datetime_now, nullable=False) username = Column(UnicodeText, default='', nullable=False) salt = Column(UnicodeText(24), default='', nullable=False) hash_alg = Column(UnicodeText, default='ARGON2', nullable=False) password = Column(UnicodeText, default='', nullable=False) name = Column(UnicodeText, default='', nullable=False) description = Column(JSON, default=dict, nullable=False) public_name = Column(UnicodeText, default='', nullable=False) role = Column(Integer, default='receiver', nullable=False) state = Column(Integer, default=1, nullable=False) last_login = Column(DateTime, default=datetime_null, nullable=False) mail_address = Column(UnicodeText, default='', nullable=False) language = Column(UnicodeText(12), nullable=False) password_change_needed = Column(Boolean, default=True, nullable=False) password_change_date = Column(DateTime, default=datetime_null, nullable=False) crypto_prv_key = Column(UnicodeText(84), default='', nullable=False) crypto_pub_key = Column(UnicodeText(56), default='', nullable=False) crypto_rec_key = Column(UnicodeText(80), default='', nullable=False) crypto_bkp_key = Column(UnicodeText(84), default='', nullable=False) crypto_escrow_prv_key = Column(UnicodeText(84), default='', nullable=False) crypto_escrow_bkp1_key = Column(UnicodeText(84), default='', nullable=False) crypto_escrow_bkp2_key = Column(UnicodeText(84), default='', nullable=False) change_email_address = Column(UnicodeText, default='', nullable=False) change_email_token = Column(UnicodeText, unique=True, nullable=True) change_email_date = Column(DateTime, default=datetime_null, nullable=False) notification = Column(Boolean, default=True, nullable=False) recipient_configuration = Column(Integer, default=0, nullable=False) can_delete_submission = Column(Boolean, default=False, nullable=False) can_postpone_expiration = Column(Boolean, default=False, nullable=False) can_edit_general_settings = Column(Boolean, default=False, nullable=False) readonly = Column(Boolean, default=False, nullable=False) two_factor_enable = Column(Boolean, default=False, nullable=False) two_factor_secret = Column(UnicodeText(16), default='', nullable=False) pgp_key_fingerprint = Column(UnicodeText, default='', nullable=False) pgp_key_public = Column(UnicodeText, default='', nullable=False) pgp_key_expiration = Column(DateTime, default=datetime_null, nullable=False) class MigrationScript(MigrationBase): def migrate_Context(self): for old_obj in self.session_old.query(self.model_from['Context']): new_obj = self.model_to['Context']() for key in new_obj.__mapper__.column_attrs.keys(): if key not in old_obj.__mapper__.column_attrs.keys(): continue value = getattr(old_obj, key) if key == 'tip_timetolive' and value < 0: value = 0 setattr(new_obj, key, value) self.session_new.add(new_obj) def migrate_Tenant(self): for old_obj in self.session_old.query(self.model_from['Tenant']): self.entries_count['Config'] += 1 new_obj = self.model_to['Tenant']() for key in new_obj.__mapper__.column_attrs.keys(): setattr(new_obj, key, getattr(old_obj, key)) for key in ['subdomain']: x = self.model_to['Config']() x.tid = old_obj.id x.var_name = key x.value = getattr(old_obj, key) self.session_new.add(x) self.session_new.add(new_obj) def migrate_User(self): for old_obj in self.session_old.query(self.model_from['User']): new_obj = self.model_to['User']() for key in new_obj.__mapper__.column_attrs.keys(): if key == 'forcefully_selected': new_obj.forcefully_selected = old_obj.recipient_configuration == 1 if hasattr(old_obj, key): setattr(new_obj, key, getattr(old_obj, key)) self.session_new.add(new_obj) def epilogue(self): m = self.model_to['Config'] self.session_new.query(m) \ .filter(m.var_name == 'enable_private_labels') \ .update({'var_name': 'private_annotations'}) self.session_new.query(m) \ .filter(m.var_name == 'https_priv_key') \ .update({'var_name': 'https_key'}) for tid in self.session_new.query(m.tid) \ .filter(m.var_name == 'smtp_port', m.value == 9267): db_reset_smtp_settings(self.session_new, tid[0]) for c in self.session_new.query(m).filter(m.var_name == 'onionservice'): if len(c.value) != 22: continue self.session_new.query(m) \ .filter(m.tid == c.tid, m.var_name == 'onionservice') \ .update({'value': ''}) self.session_new.query(m) \ .filter(m.tid == c.tid, m.var_name == 'tor_onion_key') \ .update({'value': ''}) m = self.model_to['ConfigL10N'] self.session_new.query(m) \ .filter(m.var_name == 'contexts_clarification') \ .update({'value': '', 'update_date': datetime_null()})