/
python-hadoop-mapreduce-analyzing-aws-s3-bucket-logs-with-mrjob.html
460 lines (371 loc) · 35.9 KB
/
python-hadoop-mapreduce-analyzing-aws-s3-bucket-logs-with-mrjob.html
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
284
285
286
287
288
289
290
291
292
293
294
295
296
297
298
299
300
301
302
303
304
305
306
307
308
309
310
311
312
313
314
315
316
317
318
319
320
321
322
323
324
325
326
327
328
329
330
331
332
333
334
335
336
337
338
339
340
341
342
343
344
345
346
347
348
349
350
351
352
353
354
355
356
357
358
359
360
361
362
363
364
365
366
367
368
369
370
371
372
373
374
375
376
377
378
379
380
381
382
383
384
385
386
387
388
389
390
391
392
393
394
395
396
397
398
399
400
401
402
403
404
405
406
407
408
409
410
411
412
413
414
415
416
417
418
419
420
421
422
423
424
425
426
427
428
429
430
431
432
433
434
435
436
437
438
439
440
441
442
443
444
445
446
447
448
449
450
451
452
453
454
455
456
457
458
459
460
<!DOCTYPE html>
<html lang="en" prefix="og: http://ogp.me/ns# fb: https://www.facebook.com/2008/fbml">
<head>
<title>Donne Martin</title>
<!-- Using the latest rendering mode for IE -->
<meta http-equiv="X-UA-Compatible" content="IE=edge">
<meta charset="utf-8">
<meta name="viewport" content="width=device-width, initial-scale=1">
<meta name="description" content="">
<meta name="author" content="">
<meta name="author" content="Donne Martin" />
<!-- Open Graph tags -->
<meta property="og:site_name" content="Donne Martin" />
<meta property="og:type" content="website"/>
<meta property="og:title" content="Donne Martin"/>
<meta property="og:url" content="."/>
<meta property="og:description" content="Donne Martin"/>
<!-- Bootstrap -->
<link rel="stylesheet" href="./theme/css/bootstrap.min.css" type="text/css"/>
<link href="./theme/css/pygments/monokai.css" rel="stylesheet">
<!-- Custom CSS -->
<link href="./theme/css/agency.css" rel="stylesheet">
<link href="./theme/css/custom.css" rel="stylesheet">
<!-- Custom Fonts -->
<link href="./theme/font-awesome/css/font-awesome.min.css" rel="stylesheet" type="text/css">
<link href="https://fonts.googleapis.com/css?family=Montserrat:400,700" rel="stylesheet" type="text/css">
<link href='https://fonts.googleapis.com/css?family=Kaushan+Script' rel='stylesheet' type='text/css'>
<link href='https://fonts.googleapis.com/css?family=Droid+Serif:400,700,400italic,700italic' rel='stylesheet' type='text/css'>
<link href='https://fonts.googleapis.com/css?family=Roboto+Slab:400,100,300,700' rel='stylesheet' type='text/css'>
<!-- HTML5 Shim and Respond.js IE8 support of HTML5 elements and media queries -->
<!-- WARNING: Respond.js doesn't work if you view the page via file:// -->
<!--[if lt IE 9]>
<script src="https://oss.maxcdn.com/libs/html5shiv/3.7.0/html5shiv.js"></script>
<script src="https://oss.maxcdn.com/libs/respond../theme/js/1.4.2/respond.min.js"></script>
<![endif]-->
</head><body id="page-top" class="index">
<!-- Banner -->
<!-- End Banner -->
<div class="container">
<div class="row">
<div class="col-lg-12">
<nav class="navbar navbar-default navbar-fixed-top" style="background-color: #000">
<div class="container">
<!-- Brand and toggle get grouped for better mobile display -->
<div class="navbar-header page-scroll">
<button type="button" class="navbar-toggle" data-toggle="collapse" data-target="#bs-example-navbar-collapse-1">
<span class="sr-only">Toggle navigation</span>
<span class="icon-bar"></span>
<span class="icon-bar"></span>
<span class="icon-bar"></span>
</button>
<a class="navbar-brand page-scroll" href=".">Donne Martin</a>
</div>
<!-- Collect the nav links, forms, and other content for toggling -->
<div class="collapse navbar-collapse" id="bs-example-navbar-collapse-1">
<ul class="nav navbar-nav navbar-right">
<li class="hidden">
<a href="#page-top"></a>
</li>
<li>
<a class="page-scroll" href="./#likes">Likes</a>
</li>
<li>
<a class="page-scroll" href="./#portfolio">GitHub</a>
</li>
<li>
<a class="page-scroll" href="./#about">About</a>
</li>
<li>
<a class="page-scroll" href="./#contact">Contact</a>
</li>
<li>
<a class="page-scroll" href="./archives">Blog</a>
</li>
<li>
<a class="page-scroll" href="http://donnemartin.com/viz/">Viz</a>
</li>
</ul>
</div>
<!-- /.navbar-collapse -->
</div>
<!-- /.container-fluid -->
</nav> <section id="content" class="section-top-padding">
<article class="article-top-padding">
<h1>
<a href="./python-hadoop-mapreduce-analyzing-aws-s3-bucket-logs-with-mrjob.html"
rel="bookmark"
title="Permalink to Python Hadoop MapReduce: Analyzing AWS S3 Bucket Logs with mrjob">
Python Hadoop MapReduce: Analyzing AWS S3 Bucket Logs with mrjob
</a>
</h1>
<i><time datetime="2015-05-17T00:00:00-04:00"> Sun 17 May 2015</time></i>
<div class="entry-content">
<div class="panel">
<br/>
</div>
<div class="container">
<br/>
<img class="img-responsive" src="https://raw.githubusercontent.com/donnemartin/donnemartin.github.io/master/images/posts/mrjob_cover.png">
</div>
<hr class="featurette-divider">
<h2>Introduction</h2>
<p>mrjob lets you write MapReduce jobs in Python 2.5+ and run them on several platforms. You can:</p>
<p>Write multi-step MapReduce jobs in pure Python
Test on your local machine
Run on a Hadoop cluster
Run in the cloud using Amazon Elastic MapReduce (EMR)</p>
<h2>Setup</h2>
<p>From PyPI:</p>
<div class="highlight"><pre>pip install mrjob
</pre></div>
<p>From source:</p>
<div class="highlight"><pre>python setup.py install
</pre></div>
<p>See "Sample .mrjob.conf" section below for additional config details.</p>
<h2>Processing S3 Logs</h2>
<p>Sample mrjob code that processes log files on Amazon S3 based on the S3 logging format:</p>
<div class="highlight"><pre><span class="n">mr_s3_log_parser</span><span class="o">.</span><span class="n">py</span>
<span class="kn">import</span> <span class="nn">time</span>
<span class="kn">from</span> <span class="nn">mrjob.job</span> <span class="kn">import</span> <span class="n">MRJob</span>
<span class="kn">from</span> <span class="nn">mrjob.protocol</span> <span class="kn">import</span> <span class="n">RawValueProtocol</span><span class="p">,</span> <span class="n">ReprProtocol</span>
<span class="kn">import</span> <span class="nn">re</span>
<span class="k">class</span> <span class="nc">MrS3LogParser</span><span class="p">(</span><span class="n">MRJob</span><span class="p">):</span>
<span class="sd">"""Parses the logs from S3 based on the S3 logging format:</span>
<span class="sd"> http://docs.aws.amazon.com/AmazonS3/latest/dev/LogFormat.html</span>
<span class="sd"> Aggregates a user's daily requests by user agent and operation</span>
<span class="sd"> Outputs date_time, requester, user_agent, operation, count</span>
<span class="sd"> """</span>
<span class="n">LOGPATS</span> <span class="o">=</span> <span class="s">r'(\S+) (\S+) \[(.*?)\] (\S+) (\S+) '</span> \
<span class="s">r'(\S+) (\S+) (\S+) ("([^"]+)"|-) '</span> \
<span class="s">r'(\S+) (\S+) (\S+) (\S+) (\S+) (\S+) '</span> \
<span class="s">r'("([^"]+)"|-) ("([^"]+)"|-)'</span>
<span class="n">NUM_ENTRIES_PER_LINE</span> <span class="o">=</span> <span class="mi">17</span>
<span class="n">logpat</span> <span class="o">=</span> <span class="n">re</span><span class="o">.</span><span class="n">compile</span><span class="p">(</span><span class="n">LOGPATS</span><span class="p">)</span>
<span class="p">(</span><span class="n">S3_LOG_BUCKET_OWNER</span><span class="p">,</span>
<span class="n">S3_LOG_BUCKET</span><span class="p">,</span>
<span class="n">S3_LOG_DATE_TIME</span><span class="p">,</span>
<span class="n">S3_LOG_IP</span><span class="p">,</span>
<span class="n">S3_LOG_REQUESTER_ID</span><span class="p">,</span>
<span class="n">S3_LOG_REQUEST_ID</span><span class="p">,</span>
<span class="n">S3_LOG_OPERATION</span><span class="p">,</span>
<span class="n">S3_LOG_KEY</span><span class="p">,</span>
<span class="n">S3_LOG_HTTP_METHOD</span><span class="p">,</span>
<span class="n">S3_LOG_HTTP_STATUS</span><span class="p">,</span>
<span class="n">S3_LOG_S3_ERROR</span><span class="p">,</span>
<span class="n">S3_LOG_BYTES_SENT</span><span class="p">,</span>
<span class="n">S3_LOG_OBJECT_SIZE</span><span class="p">,</span>
<span class="n">S3_LOG_TOTAL_TIME</span><span class="p">,</span>
<span class="n">S3_LOG_TURN_AROUND_TIME</span><span class="p">,</span>
<span class="n">S3_LOG_REFERER</span><span class="p">,</span>
<span class="n">S3_LOG_USER_AGENT</span><span class="p">)</span> <span class="o">=</span> <span class="nb">range</span><span class="p">(</span><span class="n">NUM_ENTRIES_PER_LINE</span><span class="p">)</span>
<span class="n">DELIMITER</span> <span class="o">=</span> <span class="s">'</span><span class="se">\t</span><span class="s">'</span>
<span class="c"># We use RawValueProtocol for input to be format agnostic</span>
<span class="c"># and avoid any type of parsing errors</span>
<span class="n">INPUT_PROTOCOL</span> <span class="o">=</span> <span class="n">RawValueProtocol</span>
<span class="c"># We use RawValueProtocol for output so we can output raw lines</span>
<span class="c"># instead of (k, v) pairs</span>
<span class="n">OUTPUT_PROTOCOL</span> <span class="o">=</span> <span class="n">RawValueProtocol</span>
<span class="c"># Encode the intermediate records using repr() instead of JSON, so the</span>
<span class="c"># record doesn't get Unicode-encoded</span>
<span class="n">INTERNAL_PROTOCOL</span> <span class="o">=</span> <span class="n">ReprProtocol</span>
<span class="k">def</span> <span class="nf">clean_date_time_zone</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">raw_date_time_zone</span><span class="p">):</span>
<span class="sd">"""Converts entry 22/Jul/2013:21:04:17 +0000 to the format</span>
<span class="sd"> 'YYYY-MM-DD HH:MM:SS' which is more suitable for loading into</span>
<span class="sd"> a database such as Redshift or RDS</span>
<span class="sd"> Note: requires the chars "[ ]" to be stripped prior to input</span>
<span class="sd"> Returns the converted datetime annd timezone</span>
<span class="sd"> or None for both values if failed</span>
<span class="sd"> TODO: Needs to combine timezone with date as one field</span>
<span class="sd"> """</span>
<span class="n">date_time</span> <span class="o">=</span> <span class="bp">None</span>
<span class="n">time_zone_parsed</span> <span class="o">=</span> <span class="bp">None</span>
<span class="c"># TODO: Probably cleaner to parse this with a regex</span>
<span class="n">date_parsed</span> <span class="o">=</span> <span class="n">raw_date_time_zone</span><span class="p">[:</span><span class="n">raw_date_time_zone</span><span class="o">.</span><span class="n">find</span><span class="p">(</span><span class="s">":"</span><span class="p">)]</span>
<span class="n">time_parsed</span> <span class="o">=</span> <span class="n">raw_date_time_zone</span><span class="p">[</span><span class="n">raw_date_time_zone</span><span class="o">.</span><span class="n">find</span><span class="p">(</span><span class="s">":"</span><span class="p">)</span> <span class="o">+</span> <span class="mi">1</span><span class="p">:</span>
<span class="n">raw_date_time_zone</span><span class="o">.</span><span class="n">find</span><span class="p">(</span><span class="s">"+"</span><span class="p">)</span> <span class="o">-</span> <span class="mi">1</span><span class="p">]</span>
<span class="n">time_zone_parsed</span> <span class="o">=</span> <span class="n">raw_date_time_zone</span><span class="p">[</span><span class="n">raw_date_time_zone</span><span class="o">.</span><span class="n">find</span><span class="p">(</span><span class="s">"+"</span><span class="p">):]</span>
<span class="k">try</span><span class="p">:</span>
<span class="n">date_struct</span> <span class="o">=</span> <span class="n">time</span><span class="o">.</span><span class="n">strptime</span><span class="p">(</span><span class="n">date_parsed</span><span class="p">,</span> <span class="s">"</span><span class="si">%d</span><span class="s">/%b/%Y"</span><span class="p">)</span>
<span class="n">converted_date</span> <span class="o">=</span> <span class="n">time</span><span class="o">.</span><span class="n">strftime</span><span class="p">(</span><span class="s">"%Y-%m-</span><span class="si">%d</span><span class="s">"</span><span class="p">,</span> <span class="n">date_struct</span><span class="p">)</span>
<span class="n">date_time</span> <span class="o">=</span> <span class="n">converted_date</span> <span class="o">+</span> <span class="s">" "</span> <span class="o">+</span> <span class="n">time_parsed</span>
<span class="c"># Throws a ValueError exception if the operation fails that is</span>
<span class="c"># caught by the calling function and is handled appropriately</span>
<span class="k">except</span> <span class="ne">ValueError</span> <span class="k">as</span> <span class="n">error</span><span class="p">:</span>
<span class="k">raise</span> <span class="ne">ValueError</span><span class="p">(</span><span class="n">error</span><span class="p">)</span>
<span class="k">else</span><span class="p">:</span>
<span class="k">return</span> <span class="n">converted_date</span><span class="p">,</span> <span class="n">date_time</span><span class="p">,</span> <span class="n">time_zone_parsed</span>
<span class="k">def</span> <span class="nf">mapper</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">_</span><span class="p">,</span> <span class="n">line</span><span class="p">):</span>
<span class="n">line</span> <span class="o">=</span> <span class="n">line</span><span class="o">.</span><span class="n">strip</span><span class="p">()</span>
<span class="n">match</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">logpat</span><span class="o">.</span><span class="n">search</span><span class="p">(</span><span class="n">line</span><span class="p">)</span>
<span class="n">date_time</span> <span class="o">=</span> <span class="bp">None</span>
<span class="n">requester</span> <span class="o">=</span> <span class="bp">None</span>
<span class="n">user_agent</span> <span class="o">=</span> <span class="bp">None</span>
<span class="n">operation</span> <span class="o">=</span> <span class="bp">None</span>
<span class="k">try</span><span class="p">:</span>
<span class="k">for</span> <span class="n">n</span> <span class="ow">in</span> <span class="nb">range</span><span class="p">(</span><span class="bp">self</span><span class="o">.</span><span class="n">NUM_ENTRIES_PER_LINE</span><span class="p">):</span>
<span class="n">group</span> <span class="o">=</span> <span class="n">match</span><span class="o">.</span><span class="n">group</span><span class="p">(</span><span class="mi">1</span> <span class="o">+</span> <span class="n">n</span><span class="p">)</span>
<span class="k">if</span> <span class="n">n</span> <span class="o">==</span> <span class="bp">self</span><span class="o">.</span><span class="n">S3_LOG_DATE_TIME</span><span class="p">:</span>
<span class="n">date</span><span class="p">,</span> <span class="n">date_time</span><span class="p">,</span> <span class="n">time_zone_parsed</span> <span class="o">=</span> \
<span class="bp">self</span><span class="o">.</span><span class="n">clean_date_time_zone</span><span class="p">(</span><span class="n">group</span><span class="p">)</span>
<span class="c"># Leave the following line of code if</span>
<span class="c"># you want to aggregate by date</span>
<span class="n">date_time</span> <span class="o">=</span> <span class="n">date</span> <span class="o">+</span> <span class="s">" 00:00:00"</span>
<span class="k">elif</span> <span class="n">n</span> <span class="o">==</span> <span class="bp">self</span><span class="o">.</span><span class="n">S3_LOG_REQUESTER_ID</span><span class="p">:</span>
<span class="n">requester</span> <span class="o">=</span> <span class="n">group</span>
<span class="k">elif</span> <span class="n">n</span> <span class="o">==</span> <span class="bp">self</span><span class="o">.</span><span class="n">S3_LOG_USER_AGENT</span><span class="p">:</span>
<span class="n">user_agent</span> <span class="o">=</span> <span class="n">group</span>
<span class="k">elif</span> <span class="n">n</span> <span class="o">==</span> <span class="bp">self</span><span class="o">.</span><span class="n">S3_LOG_OPERATION</span><span class="p">:</span>
<span class="n">operation</span> <span class="o">=</span> <span class="n">group</span>
<span class="k">else</span><span class="p">:</span>
<span class="k">pass</span>
<span class="k">except</span> <span class="ne">Exception</span><span class="p">:</span>
<span class="k">yield</span> <span class="p">((</span><span class="s">"Error while parsing line: </span><span class="si">%s</span><span class="s">"</span><span class="p">,</span> <span class="n">line</span><span class="p">),</span> <span class="mi">1</span><span class="p">)</span>
<span class="k">else</span><span class="p">:</span>
<span class="k">yield</span> <span class="p">((</span><span class="n">date_time</span><span class="p">,</span> <span class="n">requester</span><span class="p">,</span> <span class="n">user_agent</span><span class="p">,</span> <span class="n">operation</span><span class="p">),</span> <span class="mi">1</span><span class="p">)</span>
<span class="k">def</span> <span class="nf">reducer</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">key</span><span class="p">,</span> <span class="n">values</span><span class="p">):</span>
<span class="n">output</span> <span class="o">=</span> <span class="nb">list</span><span class="p">(</span><span class="n">key</span><span class="p">)</span>
<span class="n">output</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">DELIMITER</span><span class="o">.</span><span class="n">join</span><span class="p">(</span><span class="n">output</span><span class="p">)</span> <span class="o">+</span> \
<span class="bp">self</span><span class="o">.</span><span class="n">DELIMITER</span> <span class="o">+</span> \
<span class="nb">str</span><span class="p">(</span><span class="nb">sum</span><span class="p">(</span><span class="n">values</span><span class="p">))</span>
<span class="k">yield</span> <span class="bp">None</span><span class="p">,</span> <span class="n">output</span>
<span class="k">def</span> <span class="nf">steps</span><span class="p">(</span><span class="bp">self</span><span class="p">):</span>
<span class="k">return</span> <span class="p">[</span>
<span class="bp">self</span><span class="o">.</span><span class="n">mr</span><span class="p">(</span><span class="n">mapper</span><span class="o">=</span><span class="bp">self</span><span class="o">.</span><span class="n">mapper</span><span class="p">,</span>
<span class="n">reducer</span><span class="o">=</span><span class="bp">self</span><span class="o">.</span><span class="n">reducer</span><span class="p">)</span>
<span class="p">]</span>
<span class="k">if</span> <span class="n">__name__</span> <span class="o">==</span> <span class="s">'__main__'</span><span class="p">:</span>
<span class="n">MrS3LogParser</span><span class="o">.</span><span class="n">run</span><span class="p">()</span>
</pre></div>
<h2>Running Amazon Elastic MapReduce (EMR) Jobs</h2>
<p>Run an Amazon EMR job on the given input (must be a flat file hierarchy), placing the results in the output (output directory must not exist):</p>
<div class="highlight"><pre>python mr-mr_s3_log_parser.py -r emr s3://bucket-source/ --output-dir=s3://bucket-dest/
</pre></div>
<p>Run a MapReduce job locally on the specified input file, sending the results to the specified output file:</p>
<div class="highlight"><pre>python mr_s3_log_parser.py input_data.txt > output_data.txt
</pre></div>
<h2>Unit Testing S3 Logs</h2>
<p>Accompanying unit test:</p>
<div class="highlight"><pre><span class="n">test_mr_s3_log_parser</span><span class="o">.</span><span class="n">py</span>
<span class="kn">from</span> <span class="nn">StringIO</span> <span class="kn">import</span> <span class="n">StringIO</span>
<span class="kn">import</span> <span class="nn">unittest2</span> <span class="kn">as</span> <span class="nn">unittest</span>
<span class="kn">from</span> <span class="nn">mr_s3_log_parser</span> <span class="kn">import</span> <span class="n">MrS3LogParser</span>
<span class="k">class</span> <span class="nc">MrTestsUtil</span><span class="p">:</span>
<span class="k">def</span> <span class="nf">run_mr_sandbox</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">mr_job</span><span class="p">,</span> <span class="n">stdin</span><span class="p">):</span>
<span class="c"># inline runs the job in the same process so small jobs tend to</span>
<span class="c"># run faster and stack traces are simpler</span>
<span class="c"># --no-conf prevents options from local mrjob.conf from polluting</span>
<span class="c"># the testing environment</span>
<span class="c"># "-" reads from standard in</span>
<span class="n">mr_job</span><span class="o">.</span><span class="n">sandbox</span><span class="p">(</span><span class="n">stdin</span><span class="o">=</span><span class="n">stdin</span><span class="p">)</span>
<span class="c"># make_runner ensures job cleanup is performed regardless of</span>
<span class="c"># success or failure</span>
<span class="k">with</span> <span class="n">mr_job</span><span class="o">.</span><span class="n">make_runner</span><span class="p">()</span> <span class="k">as</span> <span class="n">runner</span><span class="p">:</span>
<span class="n">runner</span><span class="o">.</span><span class="n">run</span><span class="p">()</span>
<span class="k">for</span> <span class="n">line</span> <span class="ow">in</span> <span class="n">runner</span><span class="o">.</span><span class="n">stream_output</span><span class="p">():</span>
<span class="n">key</span><span class="p">,</span> <span class="n">value</span> <span class="o">=</span> <span class="n">mr_job</span><span class="o">.</span><span class="n">parse_output_line</span><span class="p">(</span><span class="n">line</span><span class="p">)</span>
<span class="k">yield</span> <span class="n">value</span>
<span class="k">class</span> <span class="nc">TestMrS3LogParser</span><span class="p">(</span><span class="n">unittest</span><span class="o">.</span><span class="n">TestCase</span><span class="p">):</span>
<span class="n">mr_job</span> <span class="o">=</span> <span class="bp">None</span>
<span class="n">mr_tests_util</span> <span class="o">=</span> <span class="bp">None</span>
<span class="n">RAW_LOG_LINE_INVALID</span> <span class="o">=</span> \
<span class="s">'00000fe9688b6e57f75bd2b7f7c1610689e8f01000000'</span> \
<span class="s">'00000388225bcc00000 '</span> \
<span class="s">'s3-storage [22/Jul/2013:21:03:27 +0000] '</span> \
<span class="s">'00.111.222.33 '</span> \
<span class="n">RAW_LOG_LINE_VALID</span> <span class="o">=</span> \
<span class="s">'00000fe9688b6e57f75bd2b7f7c1610689e8f01000000'</span> \
<span class="s">'00000388225bcc00000 '</span> \
<span class="s">'s3-storage [22/Jul/2013:21:03:27 +0000] '</span> \
<span class="s">'00.111.222.33 '</span> \
<span class="s">'arn:aws:sts::000005646931:federated-user/user 00000AB825500000 '</span> \
<span class="s">'REST.HEAD.OBJECT user/file.pdf '</span> \
<span class="s">'"HEAD /user/file.pdf?versionId=00000XMHZJp6DjM9x500000'</span> \
<span class="s">'00000SDZk '</span> \
<span class="s">'HTTP/1.1" 200 - - 4000272 18 - "-" '</span> \
<span class="s">'"Boto/2.5.1 (darwin) USER-AGENT/1.0.14.0" '</span> \
<span class="s">'00000XMHZJp6DjM9x5JVEAMo8MG00000'</span>
<span class="n">DATE_TIME_ZONE_INVALID</span> <span class="o">=</span> <span class="s">"AB/Jul/2013:21:04:17 +0000"</span>
<span class="n">DATE_TIME_ZONE_VALID</span> <span class="o">=</span> <span class="s">"22/Jul/2013:21:04:17 +0000"</span>
<span class="n">DATE_VALID</span> <span class="o">=</span> <span class="s">"2013-07-22"</span>
<span class="n">DATE_TIME_VALID</span> <span class="o">=</span> <span class="s">"2013-07-22 21:04:17"</span>
<span class="n">TIME_ZONE_VALID</span> <span class="o">=</span> <span class="s">"+0000"</span>
<span class="k">def</span> <span class="nf">__init__</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="o">*</span><span class="n">args</span><span class="p">,</span> <span class="o">**</span><span class="n">kwargs</span><span class="p">):</span>
<span class="nb">super</span><span class="p">(</span><span class="n">TestMrS3LogParser</span><span class="p">,</span> <span class="bp">self</span><span class="p">)</span><span class="o">.</span><span class="n">__init__</span><span class="p">(</span><span class="o">*</span><span class="n">args</span><span class="p">,</span> <span class="o">**</span><span class="n">kwargs</span><span class="p">)</span>
<span class="bp">self</span><span class="o">.</span><span class="n">mr_job</span> <span class="o">=</span> <span class="n">MrS3LogParser</span><span class="p">([</span><span class="s">'-r'</span><span class="p">,</span> <span class="s">'inline'</span><span class="p">,</span> <span class="s">'--no-conf'</span><span class="p">,</span> <span class="s">'-'</span><span class="p">])</span>
<span class="bp">self</span><span class="o">.</span><span class="n">mr_tests_util</span> <span class="o">=</span> <span class="n">MrTestsUtil</span><span class="p">()</span>
<span class="k">def</span> <span class="nf">test_invalid_log_lines</span><span class="p">(</span><span class="bp">self</span><span class="p">):</span>
<span class="n">stdin</span> <span class="o">=</span> <span class="n">StringIO</span><span class="p">(</span><span class="bp">self</span><span class="o">.</span><span class="n">RAW_LOG_LINE_INVALID</span><span class="p">)</span>
<span class="k">for</span> <span class="n">result</span> <span class="ow">in</span> <span class="bp">self</span><span class="o">.</span><span class="n">mr_tests_util</span><span class="o">.</span><span class="n">run_mr_sandbox</span><span class="p">(</span><span class="bp">self</span><span class="o">.</span><span class="n">mr_job</span><span class="p">,</span> <span class="n">stdin</span><span class="p">):</span>
<span class="bp">self</span><span class="o">.</span><span class="n">assertEqual</span><span class="p">(</span><span class="n">result</span><span class="o">.</span><span class="n">find</span><span class="p">(</span><span class="s">"Error"</span><span class="p">),</span> <span class="mi">0</span><span class="p">)</span>
<span class="k">def</span> <span class="nf">test_valid_log_lines</span><span class="p">(</span><span class="bp">self</span><span class="p">):</span>
<span class="n">stdin</span> <span class="o">=</span> <span class="n">StringIO</span><span class="p">(</span><span class="bp">self</span><span class="o">.</span><span class="n">RAW_LOG_LINE_VALID</span><span class="p">)</span>
<span class="k">for</span> <span class="n">result</span> <span class="ow">in</span> <span class="bp">self</span><span class="o">.</span><span class="n">mr_tests_util</span><span class="o">.</span><span class="n">run_mr_sandbox</span><span class="p">(</span><span class="bp">self</span><span class="o">.</span><span class="n">mr_job</span><span class="p">,</span> <span class="n">stdin</span><span class="p">):</span>
<span class="bp">self</span><span class="o">.</span><span class="n">assertEqual</span><span class="p">(</span><span class="n">result</span><span class="o">.</span><span class="n">find</span><span class="p">(</span><span class="s">"Error"</span><span class="p">),</span> <span class="o">-</span><span class="mi">1</span><span class="p">)</span>
<span class="k">def</span> <span class="nf">test_clean_date_time_zone</span><span class="p">(</span><span class="bp">self</span><span class="p">):</span>
<span class="n">date</span><span class="p">,</span> <span class="n">date_time</span><span class="p">,</span> <span class="n">time_zone_parsed</span> <span class="o">=</span> \
<span class="bp">self</span><span class="o">.</span><span class="n">mr_job</span><span class="o">.</span><span class="n">clean_date_time_zone</span><span class="p">(</span><span class="bp">self</span><span class="o">.</span><span class="n">DATE_TIME_ZONE_VALID</span><span class="p">)</span>
<span class="bp">self</span><span class="o">.</span><span class="n">assertEqual</span><span class="p">(</span><span class="n">date</span><span class="p">,</span> <span class="bp">self</span><span class="o">.</span><span class="n">DATE_VALID</span><span class="p">)</span>
<span class="bp">self</span><span class="o">.</span><span class="n">assertEqual</span><span class="p">(</span><span class="n">date_time</span><span class="p">,</span> <span class="bp">self</span><span class="o">.</span><span class="n">DATE_TIME_VALID</span><span class="p">)</span>
<span class="bp">self</span><span class="o">.</span><span class="n">assertEqual</span><span class="p">(</span><span class="n">time_zone_parsed</span><span class="p">,</span> <span class="bp">self</span><span class="o">.</span><span class="n">TIME_ZONE_VALID</span><span class="p">)</span>
<span class="c"># Use a lambda to delay the calling of clean_date_time_zone so that</span>
<span class="c"># assertRaises has enough time to handle it properly</span>
<span class="bp">self</span><span class="o">.</span><span class="n">assertRaises</span><span class="p">(</span><span class="ne">ValueError</span><span class="p">,</span>
<span class="k">lambda</span><span class="p">:</span> <span class="bp">self</span><span class="o">.</span><span class="n">mr_job</span><span class="o">.</span><span class="n">clean_date_time_zone</span><span class="p">(</span>
<span class="bp">self</span><span class="o">.</span><span class="n">DATE_TIME_ZONE_INVALID</span><span class="p">))</span>
<span class="k">if</span> <span class="n">__name__</span> <span class="o">==</span> <span class="s">'__main__'</span><span class="p">:</span>
<span class="n">unittest</span><span class="o">.</span><span class="n">main</span><span class="p">()</span>
</pre></div>
<h2>Running S3 Logs Unit Test</h2>
<p>Run the mrjob test:</p>
<div class="highlight"><pre>python test_mr_s3_log_parser.py -v
</pre></div>
<h2>Sample .mrjob.conf</h2>
<div class="highlight"><pre><span class="n">runners</span><span class="o">:</span>
<span class="n">emr</span><span class="o">:</span>
<span class="n">aws_access_key_id</span><span class="o">:</span> <span class="n">__ACCESS_KEY__</span>
<span class="n">aws_secret_access_key</span><span class="o">:</span> <span class="n">__SECRET_ACCESS_KEY__</span>
<span class="n">aws_region</span><span class="o">:</span> <span class="n">us</span><span class="o">-</span><span class="n">east</span><span class="o">-</span><span class="mi">1</span>
<span class="n">ec2_key_pair</span><span class="o">:</span> <span class="n">EMR</span>
<span class="n">ec2_key_pair_file</span><span class="o">:</span> <span class="o">~/.</span><span class="n">ssh</span><span class="o">/</span><span class="n">EMR</span><span class="o">.</span><span class="na">pem</span>
<span class="n">ssh_tunnel_to_job_tracker</span><span class="o">:</span> <span class="kc">true</span>
<span class="n">ec2_master_instance_type</span><span class="o">:</span> <span class="n">m1</span><span class="o">.</span><span class="na">small</span>
<span class="n">ec2_instance_type</span><span class="o">:</span> <span class="n">m1</span><span class="o">.</span><span class="na">small</span>
<span class="n">num_ec2_instances</span><span class="o">:</span> <span class="mi">5</span>
<span class="n">s3_scratch_uri</span><span class="o">:</span> <span class="n">s3</span><span class="o">://</span><span class="n">bucket</span><span class="sr">/tmp/</span>
<span class="n">s3_log_uri</span><span class="o">:</span> <span class="n">s3</span><span class="o">://</span><span class="n">bucket</span><span class="sr">/tmp/logs/</span>
<span class="n">enable_emr_debugging</span><span class="o">:</span> <span class="n">True</span>
<span class="n">bootstrap</span><span class="o">:</span>
<span class="o">-</span> <span class="n">sudo</span> <span class="n">apt</span><span class="o">-</span><span class="kd">get</span> <span class="n">install</span> <span class="o">-</span><span class="n">y</span> <span class="n">python</span><span class="o">-</span><span class="n">pip</span>
<span class="o">-</span> <span class="n">sudo</span> <span class="n">pip</span> <span class="n">install</span> <span class="o">--</span><span class="n">upgrade</span> <span class="n">simplejson</span>
</pre></div>
</div>
<hr class="featurette-divider">
<!-- /.entry-content -->
</article>
</section>
</div>
</div>
</div>
<footer>
<div class="container">
<div class="row">
<div class="col-md-12 text-left">
<span class="copyright">Copyright © Donne Martin 2014-Present</span>
</div>
</div>
</div>
</footer>
<script src="./theme/js/jquery.min.js"></script>
<!-- Include all compiled plugins (below), or include individual files as needed -->
<script src="./theme/js/bootstrap.min.js"></script>
<!-- Enable responsive features in IE8 with Respond.js (https://github.com/scottjehl/Respond) -->
<script src="./theme/js/respond.min.js"></script>
<!-- Plugin JavaScript -->
<script src="http://cdnjs.cloudflare.com/ajax/libs/jquery-easing/1.3/jquery.easing.min.js"></script>
<script src="./theme/js/classie.js"></script>
<script src="./theme/js/cbpAnimatedHeader.js"></script>
<!-- Custom Theme JavaScript -->
<script src="./theme/js/agency.js"></script>
<!-- Google Analytics Universal -->
<script>
(function(i,s,o,g,r,a,m){i['GoogleAnalyticsObject']=r;i[r]=i[r]||function(){
(i[r].q=i[r].q||[]).push(arguments)},i[r].l=1*new Date();a=s.createElement(o),
m=s.getElementsByTagName(o)[0];a.async=1;a.src=g;m.parentNode.insertBefore(a,m)
})(window,document,'script','//www.google-analytics.com/analytics.js','ga');
ga('create', 'UA-54747412-1', 'auto');
ga('send', 'pageview');
</script>
<!-- End Google Analytics Universal Code -->
</body>
</html>