import os

class Memory:

    def __init__(self):

        self.__free = 0
        self.__used = 0
        self.__total = 0
        self.__percent = 0
        self.__targets = []



    def get_mem(self, memtype):

        fd = open("/proc/meminfo", "r")
        mem = fd.read()
        fd.close()
        lines = mem.splitlines()

        if (memtype == "ram"):
            parts = lines[1].split()
            total = float(parts[1]) / (1024**2)
            free  = (float(parts[3]) + float(parts[6])) / (1024**2)
            used  = total - free

        elif (memtype == "swap"):
            parts = lines[2].split()
            total = float(parts[1]) / (1024**2)
            used  = float(parts[3]) / (1024**2)
            free  = total - used

        else:
            total = 0
            used = 0
            free = 0

        return (total, used, free)