cube code als funktion

main.py

@@ -7,15 +7,12 @@ def cube():
         try:
             entry = int(input('1 drücken zum Würfeln.\n'))
             #entry = int(entry)
-            return entry
-            break
-
+            if entry == 1:
+                print(random.randint(1, 6))
+            elif entry > 1:
+                print('Ups! Da wurde nicht die 1 gedrückt. Bitte nochmal versuchen.\n')
         except ValueError:
-            print('Ups! Keine Zahl eingegeben. Bitte nochmal versuchen.\n')
+            print('Ups! Da wurde nicht die 1 gedrückt. Bitte nochmal versuchen.\n')
 
 cube()
-while cube() == 1:
-    print(random.randint(1, 6))
-    #print('1 drücken um erneut zu würfeln')
-else:
-    print('alles andere')
+