o [h@sdddlmZddlmZddlmZmZddlmZm Z ddl m Z ddl m Z GdddeZ d S) ) BaseCommand) transaction)Event EventType)FeedbackFeedbackConfiguration)datetime)timezonec@s eZdZdZddZddZdS)Commandz2Migrate existing feedback to be linked with eventscCs6|jdddd|jdtddd|jd dd ddS) Nz--create-legacy-event store_truez6Create a "Legacy Feedback" event for unlinked feedback)actionhelpz --event-namezLegacy Feedback Collectionz?Name for the legacy event (default: Legacy Feedback Collection))typedefaultr z --dry-runz.Show what would be done without making changes) add_argumentstr)selfparserrS/var/www/html/smartRegister/feedback/management/commands/migrate_legacy_feedback.py add_arguments s  zCommand.add_argumentsc Os|d}|d}|d}tjjdd}|}|jd|d|dkr1|j|jd dS|rn|j|jd |j ddd  }|D]} |j| d } |jd | d| dqH|rl|jd|ddSt |rt jjdddid\} } tjj|| dttjddttjddddddd\} } |j| d}|j|jd|d|dnUd}g}|D](}|jrtjj|jd }|r||_||d!7}q||q||q|j|jd"|d#|r|j|jt|d$|jd%Wdn 1swYtjjdd}|jd&||dkrC|j|jd'dSdS)(Ndry_runcreate_legacy_event event_nameT) event__isnullzFound z% feedback entries without event linksrz1No unlinked feedback found. Migration not needed.z&DRY RUN MODE - No changes will be made)flat)rz - 'z': z feedback entrieszWould create legacy event: ''Legacy descriptionz%Legacy events for historical feedback)namedefaultsz1Legacy event created to archive existing feedbackim)daysilVariousF) event_typer start_dateend_datelocationis_paid is_active)eventzCreated legacy event 'z ' and linked )name__icontainszMatched z$ feedback entries to existing eventsz9 feedback entries could not be matched to existing eventsz;Consider running with --create-legacy-event to handle thesezRemaining unlinked feedback: z%All feedback is now linked to events!)robjectsfiltercountstdoutwritestyleSUCCESSWARNING values_listdistinctratomicr get_or_createrr now timedeltaupdaterfirstr)saveappendlen)rargsoptionsrrrunlinked_feedbackunlinked_count event_namesrr. legacy_typecreated legacy_event updated_count matched_countunmatched_feedbackfeedbackexisting_eventremaining_unlinkedrrrhandles        A zCommand.handleN)__name__ __module__ __qualname__r rrMrrrrr s r N)django.core.management.baser django.dbrregistration.modelsrrfeedback.modelsrrr django.utilsr r rrrrs