from cmd import Ssh
from collections import OrderedDict


class Memory():
    def setSsh(self, ssh: Ssh):
        self.ssh = ssh

    def getMem(self):
        out = self.ssh.ssh("dmidecode -t memory")
        tmp = OrderedDict()
        for i in out.split("\n"):
            if len(tmp) == 7:
                print("{Locator}|{tp}|{Size}|{Speed}|{Manufacturer}|{pn}|{sn}".format(Locator=i["Locator"],
                                                                                      tp=i["tp"], Size=i["Size"], Speed=i["Speed"], Manufacturer=i["Manufacturer"], pn=i["pn"], sn=i["sn"]))
                tmp.clear()
            if "Locator" in i:
                loc = i.split(":")[-1].strip()
                tmp.update({"Locator": loc})
            if "Size" in i:
                size = i.split(":")[-1].strip()
                tmp.update({"Size": size})
            if "Speed" in i:
                speed = i.split(":")[-1].strip()
                tmp.update({"Speed": speed})
            if "Part Number" in i:
                pn = i.split(":")[-1].strip()
                tmp.update({"pn": pn})
            if "Manufacturer" in i:
                manu = i.split(":")[-1].strip()
                tmp.update({"Manufacturer": manu})
            if "Type" in i:
                tp = i.split(":")[-1].strip()
                tmp.update({"tp": tp})
            if "Serial Number" in i:
                sn = i.split(":")[-1].strip()
                tmp.update({"sn": sn})


if __name__ == "__main__":         # test
    r = Memory()
    s = Ssh("172.17.0.64", "root", "1")
    r.setSsh(r)
    r.getMem()