import os
import sys
import pdb

sys.path.append(os.path.dirname(os.path.realpath(__file__))+'/..')
from malware_data_science.campaign_timeline import CampaignTimeline


campaign_timeline = CampaignTimeline(1, 'Malware campaign by domain')
campaign_timeline.plot_timeline()
campaign_timeline.save_fig('data/results/campaign_timeline_by_domain.png')

campaign_timeline = CampaignTimeline(2, 'Malware campaign by domain')
campaign_timeline.plot_timeline()
campaign_timeline.save_fig('data/results/campaign_timeline_by_domain.png')

campaign_timeline = CampaignTimeline(5, 'Malware campaign by files written on disk')
campaign_timeline.plot_timeline()
campaign_timeline.save_fig('data/results/campaign_timeline_by_files_written_on_disk.png')
