calculator_v2.0.py

#!/usr/bin/env python
# -*- coding: utf-8 -*-

class textColor:
    RESET = '\033[0m'
    BOLD = '\033[01m'
    GREEN = '\033[32m'
    ORANGE = '\033[33m'

from decimal import Decimal

racun = None
while True:
    racun = raw_input("Želiš računati? ")
    if racun == "y":

        prvo_stevilo = None
        while True:
            prvo_stevilo = raw_input("Vnesite prvo število: ")
            if prvo_stevilo.isdigit(): break
            else:
                print "\033[91mTo ni število\033[0m"

        operacija = None
        while True:
            operacija = raw_input("Vnesi željeno funkcijo (+, -, * ali /): ")
            if operacija in ("+", "-", "*", "/"): break
            else:
                print "\033[91mNepoznana funkcija\033[0m"

        drugo_stevilo = None
        while True:
            drugo_stevilo = raw_input("Vnesite drugo število: ")
            if drugo_stevilo.isdigit(): break
            else:
                print "\033[91mTo ni število\033[0m"


        if operacija == "+":
            vsota = float(prvo_stevilo) + float(drugo_stevilo)
            print textColor.GREEN + textColor.BOLD + str(prvo_stevilo) + " + " + str(drugo_stevilo) + " = " + (str(round(vsota, 3))).rstrip('0').rstrip('.') + textColor.RESET if '.' in (str(round(vsota, 3))) else (str(round(vsota, 3))) + textColor.RESET


        elif operacija == "-":
            razlika = float(prvo_stevilo) - float(drugo_stevilo)
            print textColor.GREEN + textColor.BOLD + str(prvo_stevilo) + " - " + str(drugo_stevilo) + " = " + (str(round(razlika, 3))).rstrip('0').rstrip('.') + textColor.RESET if '.' in (str(round(razlika, 3))) else (str(round(razlika, 3))) + textColor.RESET

        elif operacija == "*":
            kolicnik = float(prvo_stevilo) * float(drugo_stevilo)
            print textColor.GREEN + textColor.BOLD + str(prvo_stevilo) + " * " + str(drugo_stevilo) + " = " + (str(round(kolicnik, 3))).rstrip('0').rstrip('.') + textColor.RESET if '.' in (str(round(kolicnik, 3))) else (str(round(kolicnik, 3))) + textColor.RESET


        elif operacija == "/":
            ulomek = float(prvo_stevilo) / float(drugo_stevilo)
            print textColor.GREEN + textColor.BOLD + str(prvo_stevilo) + " / " + str(drugo_stevilo) + " = " + (str(round(ulomek, 3))).rstrip('0').rstrip('.') + textColor.RESET if '.' in (str(round(ulomek, 3))) else (str(round(ulomek, 3))) + textColor.RESET

        else:
            print "Neznana funkcija!"

    else:
        print "Hvala!"
        break