# ---------------------------------------------------------------------------- #

import numpy as np
import pandas as pd
import matplotlib.pyplot as plt
import pickle
import math
import datetime
from datetime import date

def main():
    # These limits will select only the valid data.
    lowest_tension = 250
    highest_tension = 450

    lowest_frequency = 80
    highest_frequency = 140

    lowest_length = 1130

    tension             = 'tension'
    leak_user           = 'leakUser'
    swager_comment      = 'swagerComment'
    tension_date        = 'tensionDateTime'
    leak_status         = 'leakStatus'
    leak_date           = 'leakDateTime'
    frequency           = 'frequency'
    filename            = 'filename'
    swager_user         = 'swagerUser'
    final_current       = 'finalCurrent'
    leak_rate           = 'leakRate'
    current_test_date   = 'currentTestDate'
    tension_length      = 'tensionLength'
    swager_date         = 'swagerDate'
    tension_user        = 'tensionUser'
    data_set_number     = 'dataSetNumber'

    file_name = 'database.p'

    master_dict = dict()

    with open(file_name, 'rb') as file_object:
        master_dict = pickle.load(file_object)

    barcodes = D.keys()
    first_sort = dict()

    for code in barcodes:
        try:
            first_sort[code] = list()

            tube_length     = master_dict[code][tension_length][0]
            tension_list    = master_dict[code][tension]
            date_list       = master_dict[code][tension_date]
            frequency_list  = master_dict[code][frequency]
        except KeyError:
            print("Error")
            pass
        else:
            if tube_length < lowest_length:
                pass

            for i in range(len(frequency_list)):
                f = frequency_list[i]
                T = tension_list[i]
                date_str = date_list[i][0:8]
                D = datetime.datetime.strptime(date_str, '%d.%m.%y')

                is_valid = False

                if f not in range(lowest_frequency, highest_frequency):
                    is_valid = False
                elif T not in range(lowest_tension, highest_tension):
                    is_valid = False
                else:
                    is_valid = True

                if is_valid:
                    data = (D, f, t)
                    first_sort[code].append(data)

    for tup in

if __name__ == "__main__":
    main()