Advertisement
Guest User

Untitled

a guest
Oct 18th, 2019
106
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 3.51 KB | None | 0 0
  1. import datetime as dt
  2.  
  3.  
  4. DATABASE = {
  5. 'Сергей': 'Омск',
  6. 'Соня': 'Москва',
  7. 'Алексей': 'Калининград',
  8. 'Миша': 'Москва',
  9. 'Дима': 'Челябинск',
  10. 'Алина': 'Красноярск',
  11. 'Егор': 'Пермь',
  12. 'Коля': 'Красноярск',
  13. 'Артём': 'Владивосток',
  14. 'Петя': 'Михайловка'
  15. }
  16.  
  17. UTC_OFFSET = {
  18. 'Москва': 3,
  19. 'Санкт-Петербург': 3,
  20. 'Новосибирск': 7,
  21. 'Екатеринбург': 5,
  22. 'Нижний Новгород': 3,
  23. 'Казань': 3,
  24. 'Челябинск': 5,
  25. 'Омск': 6,
  26. 'Самара': 4,
  27. 'Ростов-на-Дону': 3,
  28. 'Уфа': 5,
  29. 'Красноярск': 7,
  30. 'Воронеж': 3,
  31. 'Пермь': 5,
  32. 'Волгоград': 4,
  33. 'Краснодар': 3,
  34. 'Калининград': 2,
  35. 'Владивосток': 10
  36. }
  37.  
  38.  
  39. def format_count_friends(count_friends):
  40. if count_friends == 1:
  41. return '1 друг'
  42. elif 2 <= count_friends <= 4:
  43. return f'{count_friends} друга'
  44. else:
  45. return f'{count_friends} друзей'
  46.  
  47.  
  48. def what_time(city):
  49. offset = UTC_OFFSET[city]
  50. city_time = dt.datetime.utcnow() + dt.timedelta(hours=offset)
  51. f_time = city_time.strftime("%H:%M")
  52. return f_time
  53.  
  54.  
  55. def process_anfisa(query):
  56. if query == 'сколько у меня друзей?':
  57. count_string = format_count_friends(len(DATABASE))
  58. return f'У тебя {count_string}'
  59. elif query == 'кто все мои друзья?':
  60. friends_string = ', '.join(DATABASE.keys())
  61. return f'Твои друзья: {friends_string}'
  62. elif query == 'где все мои друзья?':
  63. unique_cities = set(DATABASE.values())
  64. cities_string = ', '.join(unique_cities)
  65. return f'Твои друзья в городах: {cities_string}'
  66. else:
  67. return '<неизвестный запрос>'
  68.  
  69.  
  70. def process_friend(name, query):
  71. if name in DATABASE:
  72. city = DATABASE[name]
  73. if query == 'ты где?':
  74. return f'{name} в городе {city}'
  75. elif query == 'который час?':
  76. if city not in UTC_OFFSET:
  77. return f'<не могу определить время в городе {city}>'
  78. time = what_time(city)
  79. return f'Там сейчас {time}'
  80. else:
  81. return '<неизвестный запрос>'
  82. else:
  83. return f'У тебя нет друга по имени {name}'
  84.  
  85.  
  86. def process_query(query):
  87. tokens = query.split(', ')
  88. name = tokens[0]
  89. if name == 'Анфиса':
  90. return process_anfisa(tokens[1])
  91. else:
  92. return process_friend(name, tokens[1])
  93.  
  94.  
  95. def runner():
  96. queries = [
  97. 'Анфиса, сколько у меня друзей?',
  98. 'Анфиса, кто все мои друзья?',
  99. 'Анфиса, где все мои друзья?',
  100. 'Анфиса, кто виноват?',
  101. 'Коля, ты где?',
  102. 'Соня, что делать?',
  103. 'Антон, ты где?',
  104. 'Алексей, который час?',
  105. 'Артём, который час?',
  106. 'Антон, который час?',
  107. 'Петя, который час?'
  108. ]
  109. for query in queries:
  110. print(query, '-', process_query(query))
  111.  
  112.  
  113. runner()
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement