# core/management/commands/seed_data.py

import csv
import os
from django.core.management.base import BaseCommand
from django.conf import settings
from tqdm import tqdm
from tcgabcd.models import Cna, Cna_hg19, Gene, GeneExpression, Mutation, Patient, ClinicalPatient, ClinicalSample, SvData, TreatmentTimeline, HypoxiaScore,Dataset
import pandas as pd
import numpy as np
from tcgabcd.models import Cna_hg19, Mutation, Patient, ClinicalPatient, ClinicalSample, SvData, TreatmentTimeline, HypoxiaScore, Gene, GeneExpression, Dataset
from django.db import connection

class Command(BaseCommand):
    help = 'Delete TCGA breast cancer data'

    def handle(self, *args, **kwargs):
        self.stdout.write(' Deletion Started...')

        # Delete all records in bulk
        Dataset.objects.all().delete()
        Cna_hg19.objects.all().delete()
        Mutation.objects.all().delete()
        Patient.objects.all().delete()
        ClinicalPatient.objects.all().delete()
        ClinicalSample.objects.all().delete()
        SvData.objects.all().delete()
        TreatmentTimeline.objects.all().delete()
        HypoxiaScore.objects.all().delete()
        
        self.stdout.write(self.style.SUCCESS('Data deletion completed successfully!'))

        self.stdout.write("Running VACUUM on the SQLite database...")
        with connection.cursor() as cursor:
            cursor.execute("VACUUM;")
        self.stdout.write(self.style.SUCCESS("VACUUM completed successfully!"))