PC

# -*- coding: utf-8 -*-
"""
Created on Fri Jan 18 18:55:29 2019

@author: temp110
"""

class PC:
    cpu = None
    rams = []
    drives = []

    def setCPU(self, cpu):
        self.cpu = cpu

    def addRAM(self, ram):
        self.rams.append(ram)

    def addDRIVE(self, drive):
        self.drives.append(drive)

    def __str__(self):
        res = 'PC:\n'
        if (self.cpu != None):
            res += self.cpu.__str__()
        for elem1 in self.rams:
            res += elem1.__str__()
        for elem2 in self.drives:
            res += elem2.__str__()
        return res

    def price(self):
        res = 0
        if (self.cpu != None):
            res += self.cpu.price
        for elem1 in self.rams:
            res += elem1.price
        for elem2 in self.drives:
            res += elem2.price
        return res



class CPU:

    name = ''
    freq = 0
    price = 0

    def __init__(self, name, freq, price):
        self.freq = freq
        self.name = name
        self.price = price

    def __str__(self):
        res = 'CPU:{} {}MHz {}$\n' .format(self.name, self.freq, self.price)
        return res

class RAM:

    name = ''
    ramsize = 0
    price = 0

    def __init__(self, name, ramsize, price):
        self.ramsize = ramsize
        self.name = name
        self.price = price

    def __str__(self):
        res = 'RAM:{} {}MHz {}$\n' .format(self.name, self.ramsize, self.price)
        return res



class DRIVE:

    name = ''
    freq = 0
    typedrive = ''
    price = 0
    memorysize = 0


    def __init__(self,  typedrive,name,  memorysize, freq, price):
        self.typedrive = typedrive
        self.name = name
        self.price = price
        self.memorysize = memorysize
        self.freq = freq

    def __str__(self):

        if (self.typedrive == 'SSD'):
            res = 'SSD:{} {} MB {} MB/S {}$\n' .format( self.name, self.memorysize, self.freq, self.price)
        else:
            res = '{}:{} {} MB  {} MHz {}$\n' .format(self.typedrive, self.name, self.memorysize, self.freq, self.price)
        return res





drive1 = DRIVE('HDD', 'WD Blue', 1000, 5400, 45)
drive2 = DRIVE('SSD', 'Kingston', 120, 450, 45)





cpu1 = CPU('i7 8700k', 3500 , 189)
cpu2 = CPU('i5 8600k', 3400, 129)

mem1 = RAM('Kingston', 8, 42)
mem2 = RAM('Kingston', 8, 40)


mypc = PC()
mypc.setCPU (cpu1)
mypc.addRAM (mem1)
mypc.addRAM (mem2)
mypc.addDRIVE(drive1)
mypc.addDRIVE(drive2)

print(mypc.price(),'$')
#print(cpu2)
print(mypc)