uN09mzr3

1# ruby fill_database.rb dbname  user    pass       hours
2# ruby fill_database.rb ARGV[0] ARGV[1] ARGV[2]    ARGV[3]
3# ruby fill_database.rb 'test'  'root'  'шакуувщь' 43800
4
5require 'rubygems'
6require 'mysql'
7
8
9# Зве:
10# Читаем параметры. Этот код написан так с двумя целями:
11# 1. объяснить тому, кто читает скрипт, что значит каждый параметр
12# 2. сделать минимальную проверку параметров (здесь это нужно)
13
14puts "Usage: ruby fill_database.rb dbname user password hours_to_fill"
15
16dbname = ARGV[0] or raise(ArgumentError, "Database name empty")
17user = ARGV[1] or raise(ArgumentError, "User name empty")
18password = ARGV[2] or raise(ArgumentError, "Password empty")
19hours = ARGV[3].to_i
20hours.zero? and raise(ArgumentError, "Number of hours to fill is zero")
21
22
23# Connect to the MySQL server
24
25db = Mysql.new('localhost', user, password)
26
27# Зве: проверка результата на empty? более наглядна
28#db.query("CREATE DATABASE #{ARGV[0]}") if db.list_dbs(ARGV[0]) != [ARGV[0]]
29db.query("CREATE DATABASE #{dbname}") if db.list_dbs(dbname).empty?
30
31db.real_connect('localhost', ARGV[1], ARGV[2], ARGV[0])
32
33# Create tables
34
35db.query("DROP TABLE IF EXISTS metrics")
36
37db.query(%Q{
38CREATE TABLE metrics 
39(
40    id INT NOT NULL AUTO_INCREMENT,
41    timestamp DATETIME NULL ,
42    metrics VARCHAR(15) NULL ,
43    value INT NULL ,
44    PRIMARY KEY (id)
45)
46})
47
48# Calculates first timestamp
49# t_start = Time.new — Зве: в Руби нет никакой необходимости инициализировать переменные заранее
50
51# Зве: зря усложняешь, приходится долго вчитываться, чтобы понять, что ты всего лишь пытался
52# получить целое число часов. В постановке задачи ничего про это не было :)
53# t_start = t_start - (t_start.min * 60) - t_start.sec - ((ARGV[3].to_i - 1) * 3600)
54t_start = Time.now - hours * 3600
55# Но на самом деле t_start нам вообще не нужен, достаточно текущего времени :)
56
57# Suppresses automatic result set generation
58db.query_with_result = false
59
60now = Time.now
61
62METRICS = ['Mentioners', 'Followers', 'Readers', # real metrics
63
64'Engagement', 'Participation rate', 'Ratio of comments per post', 'Share of voice',
65'Participation level', 'Interactions rate per member', 'Mention frequency per user']
66
67# Insert data
68# Зве: если начать перебор с 0, и сделать невключающий диапазон (... вместо ..), то можно обойтись без вычитания часа
69(0...hours).each do |h|
70    #t = t_start + ((h - 1) * 3600) 
71    t = now - h * 3600
72    
73    metrics_insertion = METRICS.map{|name|
74        "('%s', '%s', %i)" % [t.strftime("%Y-%m-%d %H:%M"), name, rand(1000)]
75    }.join(', ')
76
77    db.query(%Q{
78        INSERT INTO metrics (timestamp, metrics, value)
79        VALUES #{metrics_insertion}
80    })
81
82    # Зве: если умножать не на целое 100, а на дробное 100.0, можно обойтись без to_f
83    #puts "%.3f" % (h*100.0/hours)
84    
85    # Зве: но на самом деле, когда скрипт выводит такое дикое количество строчек со статусом выполнения,
86    # это как-то некруто; как и проценты выполнения с точностью до 3 знака.
87    # Достаточно чего-нибудь такого, я так мыслю:
88    if (h % 100).zero?
89        puts "%i of %i ready (%.1f%%)" % [h, hours, h*100.0/hours]
90    end
91end
92
93# Зве: Во-первых, без этой строчки вообще можно обойтись — после выполнения скрипта он сам отконнектится
94# Во-вторых, старайся избегать очевидных комментариев — с них никому никакой пользы :)
95#Disconnect from server
96#db.close if db