nltk 3.2__zip → 3.3__zip

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (406) hide show
  1. {nltk-3.2 → nltk-3.3}/INSTALL.txt +6 -6
  2. {nltk-3.2 → nltk-3.3}/LICENSE.txt +13 -13
  3. {nltk-3.2 → nltk-3.3}/MANIFEST.in +6 -6
  4. {nltk-3.2/nltk.egg-info → nltk-3.3}/PKG-INFO +13 -5
  5. nltk-3.2/README.txt → nltk-3.3/README.md +49 -48
  6. nltk-3.3/nltk/VERSION +1 -0
  7. {nltk-3.2 → nltk-3.3}/nltk/__init__.py +185 -185
  8. {nltk-3.2 → nltk-3.3}/nltk/app/__init__.py +52 -53
  9. {nltk-3.2 → nltk-3.3}/nltk/app/chartparser_app.py +2276 -2276
  10. {nltk-3.2 → nltk-3.3}/nltk/app/chunkparser_app.py +1262 -1264
  11. {nltk-3.2 → nltk-3.3}/nltk/app/collocations_app.py +347 -350
  12. {nltk-3.2 → nltk-3.3}/nltk/app/concordance_app.py +568 -570
  13. {nltk-3.2 → nltk-3.3}/nltk/app/nemo_app.py +156 -155
  14. {nltk-3.2 → nltk-3.3}/nltk/app/rdparser_app.py +892 -894
  15. {nltk-3.2 → nltk-3.3}/nltk/app/srparser_app.py +809 -809
  16. {nltk-3.2 → nltk-3.3}/nltk/app/wordfreq_app.py +32 -32
  17. {nltk-3.2 → nltk-3.3}/nltk/app/wordnet_app.py +968 -968
  18. {nltk-3.2 → nltk-3.3}/nltk/book.py +91 -90
  19. {nltk-3.2 → nltk-3.3}/nltk/ccg/__init__.py +22 -22
  20. {nltk-3.2 → nltk-3.3}/nltk/ccg/api.py +359 -335
  21. {nltk-3.2 → nltk-3.3}/nltk/ccg/chart.py +394 -361
  22. {nltk-3.2 → nltk-3.3}/nltk/ccg/combinator.py +343 -312
  23. {nltk-3.2 → nltk-3.3}/nltk/ccg/lexicon.py +328 -290
  24. nltk-3.3/nltk/ccg/logic.py +46 -0
  25. {nltk-3.2 → nltk-3.3}/nltk/chat/__init__.py +49 -49
  26. {nltk-3.2 → nltk-3.3}/nltk/chat/eliza.py +244 -244
  27. {nltk-3.2 → nltk-3.3}/nltk/chat/iesha.py +140 -140
  28. {nltk-3.2 → nltk-3.3}/nltk/chat/rude.py +92 -92
  29. {nltk-3.2 → nltk-3.3}/nltk/chat/suntsu.py +117 -117
  30. {nltk-3.2 → nltk-3.3}/nltk/chat/util.py +122 -120
  31. {nltk-3.2 → nltk-3.3}/nltk/chat/zen.py +282 -282
  32. {nltk-3.2 → nltk-3.3}/nltk/chunk/__init__.py +190 -190
  33. {nltk-3.2 → nltk-3.3}/nltk/chunk/api.py +51 -51
  34. {nltk-3.2 → nltk-3.3}/nltk/chunk/named_entity.py +331 -331
  35. {nltk-3.2 → nltk-3.3}/nltk/chunk/regexp.py +1390 -1384
  36. {nltk-3.2 → nltk-3.3}/nltk/chunk/util.py +600 -600
  37. {nltk-3.2 → nltk-3.3}/nltk/classify/__init__.py +98 -98
  38. {nltk-3.2 → nltk-3.3}/nltk/classify/api.py +193 -193
  39. {nltk-3.2 → nltk-3.3}/nltk/classify/decisiontree.py +295 -295
  40. {nltk-3.2 → nltk-3.3}/nltk/classify/maxent.py +1510 -1492
  41. {nltk-3.2 → nltk-3.3}/nltk/classify/megam.py +179 -178
  42. {nltk-3.2 → nltk-3.3}/nltk/classify/naivebayes.py +245 -242
  43. {nltk-3.2 → nltk-3.3}/nltk/classify/positivenaivebayes.py +170 -170
  44. {nltk-3.2 → nltk-3.3}/nltk/classify/rte_classify.py +160 -191
  45. {nltk-3.2 → nltk-3.3}/nltk/classify/scikitlearn.py +153 -151
  46. {nltk-3.2 → nltk-3.3}/nltk/classify/senna.py +183 -181
  47. {nltk-3.2 → nltk-3.3}/nltk/classify/svm.py +15 -15
  48. {nltk-3.2 → nltk-3.3}/nltk/classify/tadm.py +113 -112
  49. {nltk-3.2 → nltk-3.3}/nltk/classify/textcat.py +193 -193
  50. {nltk-3.2 → nltk-3.3}/nltk/classify/util.py +324 -312
  51. {nltk-3.2 → nltk-3.3}/nltk/classify/weka.py +346 -345
  52. {nltk-3.2 → nltk-3.3}/nltk/cluster/__init__.py +86 -86
  53. {nltk-3.2 → nltk-3.3}/nltk/cluster/api.py +74 -70
  54. {nltk-3.2 → nltk-3.3}/nltk/cluster/em.py +247 -247
  55. {nltk-3.2 → nltk-3.3}/nltk/cluster/gaac.py +168 -168
  56. {nltk-3.2 → nltk-3.3}/nltk/cluster/kmeans.py +221 -221
  57. {nltk-3.2 → nltk-3.3}/nltk/cluster/util.py +304 -289
  58. nltk-3.3/nltk/collections.py +640 -0
  59. {nltk-3.2 → nltk-3.3}/nltk/collocations.py +388 -388
  60. nltk-3.3/nltk/compat.py +365 -0
  61. {nltk-3.2 → nltk-3.3}/nltk/corpus/__init__.py +341 -320
  62. {nltk-3.2 → nltk-3.3}/nltk/corpus/europarl_raw.py +44 -44
  63. {nltk-3.2 → nltk-3.3}/nltk/corpus/reader/__init__.py +147 -145
  64. {nltk-3.2 → nltk-3.3}/nltk/corpus/reader/aligned.py +115 -114
  65. {nltk-3.2 → nltk-3.3}/nltk/corpus/reader/api.py +448 -446
  66. {nltk-3.2 → nltk-3.3}/nltk/corpus/reader/bnc.py +252 -252
  67. {nltk-3.2 → nltk-3.3}/nltk/corpus/reader/bracket_parse.py +227 -227
  68. {nltk-3.2 → nltk-3.3}/nltk/corpus/reader/categorized_sents.py +179 -178
  69. {nltk-3.2 → nltk-3.3}/nltk/corpus/reader/chasen.py +140 -139
  70. {nltk-3.2 → nltk-3.3}/nltk/corpus/reader/childes.py +522 -494
  71. {nltk-3.2 → nltk-3.3}/nltk/corpus/reader/chunked.py +212 -212
  72. {nltk-3.2 → nltk-3.3}/nltk/corpus/reader/cmudict.py +97 -95
  73. {nltk-3.2 → nltk-3.3}/nltk/corpus/reader/comparative_sents.py +280 -278
  74. {nltk-3.2 → nltk-3.3}/nltk/corpus/reader/conll.py +528 -521
  75. {nltk-3.2 → nltk-3.3}/nltk/corpus/reader/crubadan.py +116 -116
  76. {nltk-3.2 → nltk-3.3}/nltk/corpus/reader/dependency.py +101 -101
  77. {nltk-3.2 → nltk-3.3}/nltk/corpus/reader/framenet.py +3059 -2093
  78. {nltk-3.2 → nltk-3.3}/nltk/corpus/reader/ieer.py +112 -111
  79. {nltk-3.2 → nltk-3.3}/nltk/corpus/reader/indian.py +85 -86
  80. {nltk-3.2 → nltk-3.3}/nltk/corpus/reader/ipipan.py +332 -331
  81. {nltk-3.2 → nltk-3.3}/nltk/corpus/reader/knbc.py +187 -187
  82. {nltk-3.2 → nltk-3.3}/nltk/corpus/reader/lin.py +156 -156
  83. {nltk-3.2 → nltk-3.3}/nltk/corpus/reader/mte.py +311 -310
  84. {nltk-3.2 → nltk-3.3}/nltk/corpus/reader/nkjp.py +429 -428
  85. {nltk-3.2 → nltk-3.3}/nltk/corpus/reader/nombank.py +421 -421
  86. {nltk-3.2 → nltk-3.3}/nltk/corpus/reader/nps_chat.py +73 -73
  87. {nltk-3.2 → nltk-3.3}/nltk/corpus/reader/opinion_lexicon.py +115 -115
  88. {nltk-3.2 → nltk-3.3}/nltk/corpus/reader/panlex_lite.py +165 -165
  89. nltk-3.3/nltk/corpus/reader/pl196x.py +292 -0
  90. {nltk-3.2 → nltk-3.3}/nltk/corpus/reader/plaintext.py +232 -233
  91. {nltk-3.2 → nltk-3.3}/nltk/corpus/reader/ppattach.py +96 -95
  92. {nltk-3.2 → nltk-3.3}/nltk/corpus/reader/propbank.py +479 -481
  93. {nltk-3.2 → nltk-3.3}/nltk/corpus/reader/pros_cons.py +128 -126
  94. {nltk-3.2 → nltk-3.3}/nltk/corpus/reader/reviews.py +330 -328
  95. {nltk-3.2 → nltk-3.3}/nltk/corpus/reader/rte.py +144 -146
  96. {nltk-3.2 → nltk-3.3}/nltk/corpus/reader/semcor.py +256 -256
  97. {nltk-3.2 → nltk-3.3}/nltk/corpus/reader/senseval.py +203 -201
  98. {nltk-3.2 → nltk-3.3}/nltk/corpus/reader/sentiwordnet.py +137 -137
  99. {nltk-3.2 → nltk-3.3}/nltk/corpus/reader/sinica_treebank.py +75 -75
  100. {nltk-3.2 → nltk-3.3}/nltk/corpus/reader/string_category.py +60 -59
  101. {nltk-3.2 → nltk-3.3}/nltk/corpus/reader/switchboard.py +119 -119
  102. {nltk-3.2 → nltk-3.3}/nltk/corpus/reader/tagged.py +295 -294
  103. {nltk-3.2 → nltk-3.3}/nltk/corpus/reader/timit.py +452 -450
  104. {nltk-3.2 → nltk-3.3}/nltk/corpus/reader/toolbox.py +68 -68
  105. {nltk-3.2 → nltk-3.3}/nltk/corpus/reader/twitter.py +155 -157
  106. {nltk-3.2 → nltk-3.3}/nltk/corpus/reader/udhr.py +80 -80
  107. {nltk-3.2 → nltk-3.3}/nltk/corpus/reader/util.py +804 -800
  108. nltk-3.3/nltk/corpus/reader/verbnet.py +595 -0
  109. nltk-3.3/nltk/corpus/reader/wordlist.py +134 -0
  110. {nltk-3.2 → nltk-3.3}/nltk/corpus/reader/wordnet.py +2108 -2036
  111. {nltk-3.2 → nltk-3.3}/nltk/corpus/reader/xmldocs.py +392 -387
  112. {nltk-3.2 → nltk-3.3}/nltk/corpus/reader/ycoe.py +243 -242
  113. {nltk-3.2 → nltk-3.3}/nltk/corpus/util.py +144 -127
  114. {nltk-3.2 → nltk-3.3}/nltk/data.py +1487 -1451
  115. {nltk-3.2 → nltk-3.3}/nltk/decorators.py +219 -215
  116. {nltk-3.2 → nltk-3.3}/nltk/downloader.py +2278 -2274
  117. {nltk-3.2 → nltk-3.3}/nltk/draw/__init__.py +27 -27
  118. {nltk-3.2 → nltk-3.3}/nltk/draw/cfg.py +774 -777
  119. {nltk-3.2 → nltk-3.3}/nltk/draw/dispersion.py +58 -58
  120. {nltk-3.2 → nltk-3.3}/nltk/draw/table.py +1098 -1098
  121. {nltk-3.2 → nltk-3.3}/nltk/draw/tree.py +963 -966
  122. {nltk-3.2 → nltk-3.3}/nltk/draw/util.py +2356 -2356
  123. {nltk-3.2 → nltk-3.3}/nltk/featstruct.py +2617 -2500
  124. {nltk-3.2 → nltk-3.3}/nltk/grammar.py +1559 -1516
  125. {nltk-3.2 → nltk-3.3}/nltk/help.py +56 -56
  126. {nltk-3.2 → nltk-3.3}/nltk/inference/__init__.py +20 -20
  127. {nltk-3.2 → nltk-3.3}/nltk/inference/api.py +597 -589
  128. {nltk-3.2 → nltk-3.3}/nltk/inference/discourse.py +612 -608
  129. {nltk-3.2 → nltk-3.3}/nltk/inference/mace.py +311 -311
  130. {nltk-3.2 → nltk-3.3}/nltk/inference/nonmonotonic.py +509 -509
  131. {nltk-3.2 → nltk-3.3}/nltk/inference/prover9.py +431 -431
  132. {nltk-3.2 → nltk-3.3}/nltk/inference/resolution.py +687 -687
  133. {nltk-3.2 → nltk-3.3}/nltk/inference/tableau.py +607 -607
  134. {nltk-3.2 → nltk-3.3}/nltk/internals.py +982 -982
  135. {nltk-3.2 → nltk-3.3}/nltk/jsontags.py +63 -63
  136. {nltk-3.2 → nltk-3.3}/nltk/lazyimport.py +142 -142
  137. {nltk-3.2 → nltk-3.3}/nltk/metrics/__init__.py +30 -29
  138. {nltk-3.2 → nltk-3.3}/nltk/metrics/agreement.py +421 -401
  139. nltk-3.3/nltk/metrics/aline.py +607 -0
  140. {nltk-3.2 → nltk-3.3}/nltk/metrics/association.py +414 -410
  141. {nltk-3.2 → nltk-3.3}/nltk/metrics/confusionmatrix.py +206 -206
  142. {nltk-3.2 → nltk-3.3}/nltk/metrics/distance.py +207 -202
  143. {nltk-3.2 → nltk-3.3}/nltk/metrics/paice.py +381 -381
  144. {nltk-3.2 → nltk-3.3}/nltk/metrics/scores.py +228 -228
  145. {nltk-3.2 → nltk-3.3}/nltk/metrics/segmentation.py +231 -232
  146. {nltk-3.2 → nltk-3.3}/nltk/metrics/spearman.py +69 -69
  147. {nltk-3.2 → nltk-3.3}/nltk/misc/__init__.py +11 -11
  148. {nltk-3.2 → nltk-3.3}/nltk/misc/babelfish.py +10 -10
  149. {nltk-3.2 → nltk-3.3}/nltk/misc/chomsky.py +135 -133
  150. {nltk-3.2 → nltk-3.3}/nltk/misc/minimalset.py +83 -83
  151. {nltk-3.2 → nltk-3.3}/nltk/misc/sort.py +157 -157
  152. {nltk-3.2 → nltk-3.3}/nltk/misc/wordfinder.py +129 -129
  153. {nltk-3.2 → nltk-3.3}/nltk/parse/__init__.py +81 -80
  154. {nltk-3.2 → nltk-3.3}/nltk/parse/api.py +66 -66
  155. {nltk-3.2 → nltk-3.3}/nltk/parse/bllip.py +282 -282
  156. {nltk-3.2 → nltk-3.3}/nltk/parse/chart.py +1682 -1681
  157. nltk-3.3/nltk/parse/corenlp.py +780 -0
  158. {nltk-3.2 → nltk-3.3}/nltk/parse/dependencygraph.py +760 -758
  159. {nltk-3.2 → nltk-3.3}/nltk/parse/earleychart.py +452 -451
  160. {nltk-3.2 → nltk-3.3}/nltk/parse/evaluate.py +129 -129
  161. {nltk-3.2 → nltk-3.3}/nltk/parse/featurechart.py +580 -578
  162. {nltk-3.2 → nltk-3.3}/nltk/parse/generate.py +87 -76
  163. {nltk-3.2 → nltk-3.3}/nltk/parse/malt.py +356 -356
  164. {nltk-3.2 → nltk-3.3}/nltk/parse/nonprojectivedependencyparser.py +783 -783
  165. {nltk-3.2 → nltk-3.3}/nltk/parse/pchart.py +522 -522
  166. {nltk-3.2 → nltk-3.3}/nltk/parse/projectivedependencyparser.py +579 -578
  167. {nltk-3.2 → nltk-3.3}/nltk/parse/recursivedescent.py +655 -656
  168. {nltk-3.2 → nltk-3.3}/nltk/parse/shiftreduce.py +458 -459
  169. {nltk-3.2 → nltk-3.3}/nltk/parse/stanford.py +431 -406
  170. {nltk-3.2 → nltk-3.3}/nltk/parse/transitionparser.py +774 -772
  171. {nltk-3.2 → nltk-3.3}/nltk/parse/util.py +231 -231
  172. {nltk-3.2 → nltk-3.3}/nltk/parse/viterbi.py +401 -401
  173. {nltk-3.2 → nltk-3.3}/nltk/probability.py +2410 -2355
  174. {nltk-3.2 → nltk-3.3}/nltk/sem/__init__.py +61 -61
  175. {nltk-3.2 → nltk-3.3}/nltk/sem/boxer.py +1261 -1261
  176. {nltk-3.2 → nltk-3.3}/nltk/sem/chat80.py +782 -782
  177. {nltk-3.2 → nltk-3.3}/nltk/sem/cooper_storage.py +118 -118
  178. {nltk-3.2 → nltk-3.3}/nltk/sem/drt.py +1258 -1257
  179. {nltk-3.2 → nltk-3.3}/nltk/sem/drt_glue_demo.py +485 -489
  180. {nltk-3.2 → nltk-3.3}/nltk/sem/evaluate.py +790 -784
  181. {nltk-3.2 → nltk-3.3}/nltk/sem/glue.py +696 -694
  182. {nltk-3.2 → nltk-3.3}/nltk/sem/hole.py +386 -384
  183. {nltk-3.2 → nltk-3.3}/nltk/sem/lfg.py +210 -208
  184. {nltk-3.2 → nltk-3.3}/nltk/sem/linearlogic.py +451 -449
  185. {nltk-3.2 → nltk-3.3}/nltk/sem/logic.py +1910 -1910
  186. {nltk-3.2 → nltk-3.3}/nltk/sem/relextract.py +476 -483
  187. {nltk-3.2 → nltk-3.3}/nltk/sem/skolemize.py +101 -101
  188. {nltk-3.2 → nltk-3.3}/nltk/sem/util.py +249 -249
  189. {nltk-3.2 → nltk-3.3}/nltk/sentiment/__init__.py +14 -14
  190. {nltk-3.2 → nltk-3.3}/nltk/sentiment/sentiment_analyzer.py +229 -229
  191. {nltk-3.2 → nltk-3.3}/nltk/sentiment/util.py +762 -752
  192. {nltk-3.2 → nltk-3.3}/nltk/sentiment/vader.py +443 -445
  193. {nltk-3.2 → nltk-3.3}/nltk/stem/__init__.py +31 -33
  194. {nltk-3.2 → nltk-3.3}/nltk/stem/api.py +28 -25
  195. nltk-3.3/nltk/stem/arlstem.py +369 -0
  196. {nltk-3.2 → nltk-3.3}/nltk/stem/isri.py +345 -345
  197. {nltk-3.2 → nltk-3.3}/nltk/stem/lancaster.py +337 -310
  198. nltk-3.3/nltk/stem/porter.py +710 -0
  199. {nltk-3.2 → nltk-3.3}/nltk/stem/regexp.py +61 -61
  200. {nltk-3.2 → nltk-3.3}/nltk/stem/rslp.py +142 -142
  201. {nltk-3.2 → nltk-3.3}/nltk/stem/snowball.py +4236 -3712
  202. {nltk-3.2 → nltk-3.3}/nltk/stem/util.py +22 -12
  203. {nltk-3.2 → nltk-3.3}/nltk/stem/wordnet.py +51 -51
  204. {nltk-3.2 → nltk-3.3}/nltk/tag/__init__.py +152 -127
  205. {nltk-3.2 → nltk-3.3}/nltk/tag/api.py +86 -81
  206. {nltk-3.2 → nltk-3.3}/nltk/tag/brill.py +424 -426
  207. {nltk-3.2 → nltk-3.3}/nltk/tag/brill_trainer.py +608 -608
  208. {nltk-3.2 → nltk-3.3}/nltk/tag/crf.py +204 -200
  209. {nltk-3.2 → nltk-3.3}/nltk/tag/hmm.py +1277 -1277
  210. {nltk-3.2 → nltk-3.3}/nltk/tag/hunpos.py +132 -132
  211. {nltk-3.2 → nltk-3.3}/nltk/tag/mapping.py +101 -101
  212. {nltk-3.2 → nltk-3.3}/nltk/tag/perceptron.py +329 -314
  213. {nltk-3.2 → nltk-3.3}/nltk/tag/senna.py +147 -145
  214. {nltk-3.2 → nltk-3.3}/nltk/tag/sequential.py +740 -736
  215. {nltk-3.2 → nltk-3.3}/nltk/tag/stanford.py +219 -196
  216. {nltk-3.2 → nltk-3.3}/nltk/tag/tnt.py +595 -604
  217. {nltk-3.2 → nltk-3.3}/nltk/tag/util.py +72 -72
  218. {nltk-3.2 → nltk-3.3}/nltk/tbl/__init__.py +28 -28
  219. {nltk-3.2 → nltk-3.3}/nltk/tbl/api.py +1 -1
  220. {nltk-3.2 → nltk-3.3}/nltk/tbl/demo.py +366 -366
  221. {nltk-3.2 → nltk-3.3}/nltk/tbl/erroranalysis.py +39 -39
  222. {nltk-3.2 → nltk-3.3}/nltk/tbl/feature.py +259 -261
  223. {nltk-3.2 → nltk-3.3}/nltk/tbl/rule.py +321 -321
  224. {nltk-3.2 → nltk-3.3}/nltk/tbl/template.py +303 -308
  225. {nltk-3.2 → nltk-3.3}/nltk/test/__init__.py +18 -18
  226. {nltk-3.2 → nltk-3.3}/nltk/test/all.py +24 -23
  227. {nltk-3.2 → nltk-3.3}/nltk/test/bleu.doctest +14 -14
  228. {nltk-3.2 → nltk-3.3}/nltk/test/bnc.doctest +60 -60
  229. {nltk-3.2 → nltk-3.3}/nltk/test/ccg.doctest +376 -395
  230. nltk-3.3/nltk/test/ccg_semantics.doctest +553 -0
  231. {nltk-3.2 → nltk-3.3}/nltk/test/chat80.doctest +234 -234
  232. {nltk-3.2 → nltk-3.3}/nltk/test/childes.doctest +184 -184
  233. {nltk-3.2 → nltk-3.3}/nltk/test/childes_fixt.py +14 -13
  234. {nltk-3.2 → nltk-3.3}/nltk/test/chunk.doctest +373 -373
  235. {nltk-3.2 → nltk-3.3}/nltk/test/classify.doctest +185 -185
  236. {nltk-3.2 → nltk-3.3}/nltk/test/classify_fixt.py +11 -10
  237. nltk-3.3/nltk/test/collections.doctest +20 -0
  238. {nltk-3.2 → nltk-3.3}/nltk/test/collocations.doctest +276 -276
  239. {nltk-3.2 → nltk-3.3}/nltk/test/compat.doctest +134 -138
  240. {nltk-3.2 → nltk-3.3}/nltk/test/compat_fixt.py +9 -8
  241. nltk-3.3/nltk/test/concordance.doctest +55 -0
  242. {nltk-3.2 → nltk-3.3}/nltk/test/corpus.doctest +2200 -2212
  243. {nltk-3.2 → nltk-3.3}/nltk/test/corpus_fixt.py +3 -3
  244. {nltk-3.2 → nltk-3.3}/nltk/test/crubadan.doctest +65 -65
  245. {nltk-3.2 → nltk-3.3}/nltk/test/data.doctest +379 -379
  246. {nltk-3.2 → nltk-3.3}/nltk/test/dependency.doctest +241 -241
  247. {nltk-3.2 → nltk-3.3}/nltk/test/discourse.doctest +546 -546
  248. {nltk-3.2 → nltk-3.3}/nltk/test/discourse_fixt.py +14 -13
  249. {nltk-3.2 → nltk-3.3}/nltk/test/doctest_nose_plugin.py +159 -156
  250. {nltk-3.2 → nltk-3.3}/nltk/test/drt.doctest +517 -517
  251. {nltk-3.2 → nltk-3.3}/nltk/test/featgram.doctest +607 -607
  252. {nltk-3.2 → nltk-3.3}/nltk/test/featstruct.doctest +1229 -1230
  253. {nltk-3.2 → nltk-3.3}/nltk/test/framenet.doctest +288 -239
  254. {nltk-3.2 → nltk-3.3}/nltk/test/generate.doctest +67 -67
  255. {nltk-3.2 → nltk-3.3}/nltk/test/gensim.doctest +140 -141
  256. {nltk-3.2 → nltk-3.3}/nltk/test/gensim_fixt.py +10 -9
  257. {nltk-3.2 → nltk-3.3}/nltk/test/gluesemantics.doctest +384 -386
  258. {nltk-3.2 → nltk-3.3}/nltk/test/gluesemantics_malt.doctest +68 -68
  259. {nltk-3.2 → nltk-3.3}/nltk/test/gluesemantics_malt_fixt.py +12 -11
  260. {nltk-3.2 → nltk-3.3}/nltk/test/grammar.doctest +48 -48
  261. {nltk-3.2 → nltk-3.3}/nltk/test/grammartestsuites.doctest +109 -109
  262. {nltk-3.2 → nltk-3.3}/nltk/test/index.doctest +100 -100
  263. {nltk-3.2 → nltk-3.3}/nltk/test/inference.doctest +534 -534
  264. {nltk-3.2 → nltk-3.3}/nltk/test/inference_fixt.py +12 -11
  265. {nltk-3.2 → nltk-3.3}/nltk/test/internals.doctest +140 -140
  266. {nltk-3.2 → nltk-3.3}/nltk/test/japanese.doctest +48 -48
  267. {nltk-3.2 → nltk-3.3}/nltk/test/logic.doctest +1098 -1098
  268. {nltk-3.2 → nltk-3.3}/nltk/test/metrics.doctest +270 -270
  269. {nltk-3.2 → nltk-3.3}/nltk/test/misc.doctest +118 -118
  270. {nltk-3.2 → nltk-3.3}/nltk/test/nonmonotonic.doctest +286 -286
  271. {nltk-3.2 → nltk-3.3}/nltk/test/nonmonotonic_fixt.py +12 -11
  272. {nltk-3.2 → nltk-3.3}/nltk/test/paice.doctest +35 -35
  273. {nltk-3.2 → nltk-3.3}/nltk/test/parse.doctest +884 -884
  274. {nltk-3.2 → nltk-3.3}/nltk/test/portuguese_en.doctest +565 -565
  275. {nltk-3.2 → nltk-3.3}/nltk/test/portuguese_en_fixt.py +16 -16
  276. {nltk-3.2 → nltk-3.3}/nltk/test/probability.doctest +304 -304
  277. {nltk-3.2 → nltk-3.3}/nltk/test/probability_fixt.py +13 -12
  278. {nltk-3.2 → nltk-3.3}/nltk/test/propbank.doctest +176 -176
  279. {nltk-3.2 → nltk-3.3}/nltk/test/relextract.doctest +263 -263
  280. {nltk-3.2 → nltk-3.3}/nltk/test/resolution.doctest +221 -221
  281. {nltk-3.2 → nltk-3.3}/nltk/test/runtests.py +86 -64
  282. {nltk-3.2 → nltk-3.3}/nltk/test/segmentation_fixt.py +10 -10
  283. {nltk-3.2 → nltk-3.3}/nltk/test/semantics.doctest +665 -665
  284. {nltk-3.2 → nltk-3.3}/nltk/test/semantics_fixt.py +7 -7
  285. {nltk-3.2 → nltk-3.3}/nltk/test/sentiment.doctest +233 -233
  286. {nltk-3.2 → nltk-3.3}/nltk/test/sentiwordnet.doctest +41 -41
  287. {nltk-3.2 → nltk-3.3}/nltk/test/simple.doctest +85 -85
  288. {nltk-3.2 → nltk-3.3}/nltk/test/stem.doctest +78 -78
  289. {nltk-3.2 → nltk-3.3}/nltk/test/tag.doctest +33 -33
  290. {nltk-3.2 → nltk-3.3}/nltk/test/tokenize.doctest +273 -166
  291. {nltk-3.2 → nltk-3.3}/nltk/test/toolbox.doctest +307 -307
  292. {nltk-3.2 → nltk-3.3}/nltk/test/translate.doctest +242 -242
  293. {nltk-3.2 → nltk-3.3}/nltk/test/translate_fixt.py +3 -3
  294. {nltk-3.2 → nltk-3.3}/nltk/test/tree.doctest +1101 -1101
  295. {nltk-3.2 → nltk-3.3}/nltk/test/treeprettyprinter.doctest +127 -127
  296. {nltk-3.2 → nltk-3.3}/nltk/test/treetransforms.doctest +156 -156
  297. nltk-3.3/nltk/test/unit/test_2x_compat.py +63 -0
  298. nltk-3.3/nltk/test/unit/test_aline.py +52 -0
  299. nltk-3.3/nltk/test/unit/test_chunk.py +56 -0
  300. {nltk-3.2 → nltk-3.3}/nltk/test/unit/test_classify.py +55 -53
  301. {nltk-3.2 → nltk-3.3}/nltk/test/unit/test_collocations.py +88 -79
  302. nltk-3.3/nltk/test/unit/test_concordance.py +98 -0
  303. nltk-3.3/nltk/test/unit/test_corenlp.py +1000 -0
  304. {nltk-3.2 → nltk-3.3}/nltk/test/unit/test_corpora.py +200 -183
  305. {nltk-3.2 → nltk-3.3}/nltk/test/unit/test_corpus_views.py +45 -45
  306. {nltk-3.2 → nltk-3.3}/nltk/test/unit/test_hmm.py +88 -88
  307. {nltk-3.2 → nltk-3.3}/nltk/test/unit/test_json2csv_corpus.py +188 -187
  308. {nltk-3.2 → nltk-3.3}/nltk/test/unit/test_naivebayes.py +26 -26
  309. nltk-3.3/nltk/test/unit/test_rte_classify.py +86 -0
  310. {nltk-3.2 → nltk-3.3}/nltk/test/unit/test_seekable_unicode_stream_reader.py +124 -124
  311. nltk-3.3/nltk/test/unit/test_senna.py +74 -0
  312. nltk-3.3/nltk/test/unit/test_stem.py +121 -0
  313. {nltk-3.2 → nltk-3.3}/nltk/test/unit/test_tag.py +20 -20
  314. {nltk-3.2 → nltk-3.3}/nltk/test/unit/test_tgrep.py +628 -626
  315. nltk-3.3/nltk/test/unit/test_tokenize.py +168 -0
  316. {nltk-3.2 → nltk-3.3}/nltk/test/unit/test_twitter_auth.py +185 -185
  317. nltk-3.3/nltk/test/unit/test_wordnet.py +134 -0
  318. nltk-3.3/nltk/test/unit/translate/test_bleu.py +244 -0
  319. nltk-3.3/nltk/test/unit/translate/test_gdfa.py +57 -0
  320. {nltk-3.2 → nltk-3.3}/nltk/test/unit/translate/test_ibm1.py +75 -75
  321. {nltk-3.2 → nltk-3.3}/nltk/test/unit/translate/test_ibm2.py +87 -87
  322. {nltk-3.2 → nltk-3.3}/nltk/test/unit/translate/test_ibm3.py +105 -105
  323. {nltk-3.2 → nltk-3.3}/nltk/test/unit/translate/test_ibm4.py +126 -126
  324. {nltk-3.2 → nltk-3.3}/nltk/test/unit/translate/test_ibm5.py +166 -166
  325. {nltk-3.2 → nltk-3.3}/nltk/test/unit/translate/test_ibm_model.py +270 -270
  326. nltk-3.3/nltk/test/unit/translate/test_nist.py +37 -0
  327. {nltk-3.2 → nltk-3.3}/nltk/test/unit/translate/test_stack_decoder.py +300 -300
  328. {nltk-3.2 → nltk-3.3}/nltk/test/unit/utils.py +41 -41
  329. {nltk-3.2 → nltk-3.3}/nltk/test/util.doctest +48 -48
  330. {nltk-3.2 → nltk-3.3}/nltk/test/wordnet.doctest +595 -581
  331. {nltk-3.2 → nltk-3.3}/nltk/test/wordnet_fixt.py +7 -6
  332. {nltk-3.2 → nltk-3.3}/nltk/test/wordnet_lch.doctest +53 -53
  333. {nltk-3.2 → nltk-3.3}/nltk/test/wsd.doctest +68 -68
  334. {nltk-3.2 → nltk-3.3}/nltk/text.py +663 -610
  335. {nltk-3.2 → nltk-3.3}/nltk/tgrep.py +938 -937
  336. {nltk-3.2 → nltk-3.3}/nltk/tokenize/__init__.py +130 -107
  337. {nltk-3.2 → nltk-3.3}/nltk/tokenize/api.py +77 -78
  338. {nltk-3.2 → nltk-3.3}/nltk/tokenize/casual.py +343 -342
  339. {nltk-3.2 → nltk-3.3}/nltk/tokenize/mwe.py +120 -120
  340. nltk-3.3/nltk/tokenize/nist.py +173 -0
  341. {nltk-3.2 → nltk-3.3}/nltk/tokenize/punkt.py +1620 -1601
  342. {nltk-3.2 → nltk-3.3}/nltk/tokenize/regexp.py +208 -202
  343. nltk-3.3/nltk/tokenize/repp.py +151 -0
  344. {nltk-3.2 → nltk-3.3}/nltk/tokenize/sexpr.py +140 -142
  345. {nltk-3.2 → nltk-3.3}/nltk/tokenize/simple.py +138 -136
  346. {nltk-3.2 → nltk-3.3}/nltk/tokenize/stanford.py +117 -113
  347. nltk-3.3/nltk/tokenize/stanford_segmenter.py +246 -0
  348. {nltk-3.2 → nltk-3.3}/nltk/tokenize/texttiling.py +458 -460
  349. nltk-3.3/nltk/tokenize/toktok.py +155 -0
  350. nltk-3.3/nltk/tokenize/treebank.py +363 -0
  351. nltk-3.3/nltk/tokenize/util.py +261 -0
  352. {nltk-3.2 → nltk-3.3}/nltk/toolbox.py +499 -497
  353. {nltk-3.2 → nltk-3.3}/nltk/translate/__init__.py +24 -24
  354. {nltk-3.2 → nltk-3.3}/nltk/translate/api.py +331 -321
  355. {nltk-3.2 → nltk-3.3}/nltk/translate/bleu_score.py +615 -556
  356. nltk-3.3/nltk/translate/chrf_score.py +137 -0
  357. {nltk-3.2 → nltk-3.3}/nltk/translate/gale_church.py +248 -235
  358. {nltk-3.2 → nltk-3.3}/nltk/translate/gdfa.py +135 -131
  359. nltk-3.3/nltk/translate/gleu_score.py +193 -0
  360. {nltk-3.2 → nltk-3.3}/nltk/translate/ibm1.py +250 -241
  361. {nltk-3.2 → nltk-3.3}/nltk/translate/ibm2.py +308 -308
  362. {nltk-3.2 → nltk-3.3}/nltk/translate/ibm3.py +337 -337
  363. {nltk-3.2 → nltk-3.3}/nltk/translate/ibm4.py +467 -467
  364. {nltk-3.2 → nltk-3.3}/nltk/translate/ibm5.py +639 -639
  365. {nltk-3.2 → nltk-3.3}/nltk/translate/ibm_model.py +536 -533
  366. {nltk-3.2 → nltk-3.3}/nltk/translate/metrics.py +40 -40
  367. nltk-3.3/nltk/translate/nist_score.py +178 -0
  368. {nltk-3.2 → nltk-3.3}/nltk/translate/phrase_based.py +175 -175
  369. {nltk-3.2 → nltk-3.3}/nltk/translate/ribes_score.py +325 -325
  370. {nltk-3.2 → nltk-3.3}/nltk/translate/stack_decoder.py +499 -495
  371. {nltk-3.2 → nltk-3.3}/nltk/tree.py +1605 -1603
  372. {nltk-3.2 → nltk-3.3}/nltk/treeprettyprinter.py +564 -564
  373. {nltk-3.2 → nltk-3.3}/nltk/treetransforms.py +309 -309
  374. {nltk-3.2 → nltk-3.3}/nltk/twitter/__init__.py +28 -28
  375. {nltk-3.2 → nltk-3.3}/nltk/twitter/api.py +141 -136
  376. {nltk-3.2 → nltk-3.3}/nltk/twitter/common.py +264 -258
  377. {nltk-3.2 → nltk-3.3}/nltk/twitter/twitter_demo.py +288 -288
  378. {nltk-3.2 → nltk-3.3}/nltk/twitter/twitterclient.py +530 -530
  379. {nltk-3.2 → nltk-3.3}/nltk/twitter/util.py +147 -146
  380. nltk-3.3/nltk/util.py +751 -0
  381. {nltk-3.2 → nltk-3.3}/nltk/wsd.py +53 -53
  382. {nltk-3.2 → nltk-3.3/nltk.egg-info}/PKG-INFO +13 -5
  383. {nltk-3.2 → nltk-3.3}/nltk.egg-info/SOURCES.txt +26 -3
  384. nltk-3.3/nltk.egg-info/requires.txt +31 -0
  385. {nltk-3.2 → nltk-3.3}/setup.cfg +0 -1
  386. {nltk-3.2 → nltk-3.3}/setup.py +102 -75
  387. nltk-3.2/nltk/VERSION +0 -1
  388. nltk-3.2/nltk/compat.py +0 -680
  389. nltk-3.2/nltk/corpus/reader/pl196x.py +0 -278
  390. nltk-3.2/nltk/corpus/reader/verbnet.py +0 -408
  391. nltk-3.2/nltk/corpus/reader/wordlist.py +0 -37
  392. nltk-3.2/nltk/sentiment/vader_lexicon.txt +0 -7517
  393. nltk-3.2/nltk/six.py +0 -868
  394. nltk-3.2/nltk/stem/porter.py +0 -692
  395. nltk-3.2/nltk/test/unit/test_2x_compat.py +0 -26
  396. nltk-3.2/nltk/test/unit/test_stem.py +0 -36
  397. nltk-3.2/nltk/test/unit/test_tokenize.py +0 -23
  398. nltk-3.2/nltk/test/unit/translate/test_bleu.py +0 -84
  399. nltk-3.2/nltk/tokenize/treebank.py +0 -124
  400. nltk-3.2/nltk/tokenize/util.py +0 -90
  401. nltk-3.2/nltk/util.py +0 -1364
  402. {nltk-3.2 → nltk-3.3}/nltk/test/unit/__init__.py +0 -0
  403. {nltk-3.2 → nltk-3.3}/nltk/test/unit/translate/__init__.py +0 -0
  404. {nltk-3.2 → nltk-3.3}/nltk.egg-info/dependency_links.txt +0 -0
  405. {nltk-3.2 → nltk-3.3}/nltk.egg-info/not-zip-safe +0 -0
  406. {nltk-3.2 → nltk-3.3}/nltk.egg-info/top_level.txt +0 -0
@@ -1,530 +1,530 @@
1
- # -*- coding: utf-8 -*-
2
- # Natural Language Toolkit: Twitter client
3
- #
4
- # Copyright (C) 2001-2015 NLTK Project
5
- # Author: Ewan Klein <ewan@inf.ed.ac.uk>
6
- # Lorenzo Rubio <lrnzcig@gmail.com>
7
- # URL: <http://nltk.org/>
8
- # For license information, see LICENSE.TXT
9
-
10
-
11
- """
12
- NLTK Twitter client
13
-
14
- This module offers methods for collecting and processing Tweets. Most of the
15
- functionality depends on access to the Twitter APIs, and this is handled via
16
- the third party Twython library.
17
-
18
- If one of the methods below returns an integer, it is probably a `Twitter
19
- error code <https://dev.twitter.com/overview/api/response-codes>`_. For
20
- example, the response of '420' means that you have reached the limit of the
21
- requests you can currently make to the Twitter API. Currently, `rate limits
22
- for the search API <https://dev.twitter.com/rest/public/rate-limiting>`_ are
23
- divided into 15 minute windows.
24
- """
25
-
26
- import datetime
27
- import itertools
28
- import json
29
- import os
30
- import requests
31
- import time
32
- import gzip
33
-
34
-
35
- from twython import Twython, TwythonStreamer
36
- from twython.exceptions import TwythonRateLimitError, TwythonError
37
-
38
- from nltk.twitter.util import credsfromfile, guess_path
39
- from nltk.twitter.api import TweetHandlerI, BasicTweetHandler
40
-
41
-
42
-
43
- class Streamer(TwythonStreamer):
44
- """
45
- Retrieve data from the Twitter Streaming API.
46
-
47
- The streaming API requires
48
- `OAuth 1.0 <http://en.wikipedia.org/wiki/OAuth>`_ authentication.
49
- """
50
- def __init__(self, app_key, app_secret, oauth_token, oauth_token_secret):
51
-
52
- self.handler = None
53
- self.do_continue = True
54
- TwythonStreamer.__init__(self, app_key, app_secret, oauth_token,
55
- oauth_token_secret)
56
-
57
- def register(self, handler):
58
- """
59
- Register a method for handling Tweets.
60
-
61
- :param TweetHandlerI handler: method for viewing
62
- """
63
- self.handler = handler
64
-
65
- def on_success(self, data):
66
- """
67
- :param data: response from Twitter API
68
- """
69
- if self.do_continue:
70
- if self.handler is not None:
71
- if 'text' in data:
72
- self.handler.counter += 1
73
- self.handler.handle(data)
74
- self.do_continue = self.handler.do_continue()
75
- else:
76
- raise ValueError("No data handler has been registered.")
77
- else:
78
- self.disconnect()
79
- self.handler.on_finish()
80
-
81
-
82
- def on_error(self, status_code, data):
83
- """
84
- :param status_code: The status code returned by the Twitter API
85
- :param data: The response from Twitter API
86
-
87
- """
88
- print(status_code)
89
-
90
- def sample(self):
91
- """
92
- Wrapper for 'statuses / sample' API call
93
- """
94
- while self.do_continue:
95
-
96
- # Stream in an endless loop until limit is reached. See twython
97
- # issue 288: https://github.com/ryanmcgrath/twython/issues/288
98
- # colditzjb commented on 9 Dec 2014
99
-
100
- try:
101
- self.statuses.sample()
102
- except requests.exceptions.ChunkedEncodingError as e:
103
- if e is not None:
104
- print("Error (stream will continue): {0}".format(e))
105
- continue
106
-
107
- def filter(self, track='', follow='', lang='en'):
108
- """
109
- Wrapper for 'statuses / filter' API call
110
- """
111
- while self.do_continue:
112
- #Stream in an endless loop until limit is reached
113
-
114
- try:
115
- if track == '' and follow == '':
116
- msg = "Please supply a value for 'track', 'follow'"
117
- raise ValueError(msg)
118
- self.statuses.filter(track=track, follow=follow, lang=lang)
119
- except requests.exceptions.ChunkedEncodingError as e:
120
- if e is not None:
121
- print("Error (stream will continue): {0}".format(e))
122
- continue
123
-
124
-
125
- class Query(Twython):
126
- """
127
- Retrieve data from the Twitter REST API.
128
- """
129
- def __init__(self, app_key, app_secret, oauth_token,
130
- oauth_token_secret):
131
- self.handler = None
132
- self.do_continue = True
133
- Twython.__init__(self, app_key, app_secret, oauth_token, oauth_token_secret)
134
-
135
- def register(self, handler):
136
- """
137
- Register a method for handling Tweets.
138
-
139
- :param TweetHandlerI handler: method for viewing or writing Tweets to a file.
140
- """
141
- self.handler = handler
142
-
143
- def expand_tweetids(self, ids_f, verbose=True):
144
- """
145
- Given a file object containing a list of Tweet IDs, fetch the
146
- corresponding full Tweets from the Twitter API.
147
-
148
- The API call `statuses/lookup` will fail to retrieve a Tweet if the
149
- user has deleted it.
150
-
151
- This call to the Twitter API is rate-limited. See
152
- <https://dev.twitter.com/rest/reference/get/statuses/lookup> for details.
153
-
154
- :param ids_f: input file object consisting of Tweet IDs, one to a line
155
- :return: iterable of Tweet objects in JSON format
156
- """
157
- ids = [line.strip() for line in ids_f if line]
158
-
159
- if verbose:
160
- print("Counted {0} Tweet IDs in {1}.".format(len(ids), ids_f))
161
-
162
- # The Twitter endpoint takes lists of up to 100 ids, so we chunk the
163
- # ids.
164
- id_chunks = [ids[i:i+100] for i in range(0, len(ids), 100)]
165
-
166
- chunked_tweets = (self.lookup_status(id=chunk) for chunk in
167
- id_chunks)
168
-
169
- return itertools.chain.from_iterable(chunked_tweets)
170
-
171
-
172
-
173
- def _search_tweets(self, keywords, limit=100, lang='en'):
174
- """
175
- Assumes that the handler has been informed. Fetches Tweets from
176
- search_tweets generator output and passses them to handler
177
-
178
- :param str keywords: A list of query terms to search for, written as\
179
- a comma-separated string.
180
- :param int limit: Number of Tweets to process
181
- :param str lang: language
182
- """
183
- while True:
184
- tweets = self.search_tweets(keywords=keywords, limit=limit, lang=lang,
185
- max_id=self.handler.max_id)
186
- for tweet in tweets:
187
- self.handler.handle(tweet)
188
- if not (self.handler.do_continue() and self.handler.repeat):
189
- break
190
- self.handler.on_finish()
191
-
192
- def search_tweets(self, keywords, limit=100, lang='en', max_id=None,
193
- retries_after_twython_exception=0):
194
- """
195
- Call the REST API ``'search/tweets'`` endpoint with some plausible
196
- defaults. See `the Twitter search documentation
197
- <https://dev.twitter.com/rest/public/search>`_ for more information
198
- about admissable search parameters.
199
-
200
- :param str keywords: A list of query terms to search for, written as\
201
- a comma-separated string
202
- :param int limit: Number of Tweets to process
203
- :param str lang: language
204
- :param int max_id: id of the last tweet fetched
205
- :param int retries_after_twython_exception: number of retries when\
206
- searching Tweets before raising an exception
207
- :rtype: python generator
208
- """
209
- if not self.handler:
210
- # if no handler is provided, `BasicTweetHandler` provides minimum
211
- # functionality for limiting the number of Tweets retrieved
212
- self.handler = BasicTweetHandler(limit=limit)
213
-
214
- count_from_query = 0
215
- if max_id:
216
- self.handler.max_id = max_id
217
- else:
218
- results = self.search(q=keywords, count=min(100, limit), lang=lang,
219
- result_type='recent')
220
- count = len(results['statuses'])
221
- if count == 0:
222
- print("No Tweets available through REST API for those keywords")
223
- return
224
- count_from_query = count
225
- self.handler.max_id = results['statuses'][count - 1]['id'] - 1
226
-
227
- for result in results['statuses']:
228
- yield result
229
- self.handler.counter += 1
230
- if self.handler.do_continue() == False:
231
- return
232
-
233
-
234
- # Pagination loop: keep fetching Tweets until the desired count is
235
- # reached while dealing with Twitter rate limits.
236
- retries = 0
237
- while count_from_query < limit:
238
- try:
239
- mcount = min(100, limit-count_from_query)
240
- results = self.search(q=keywords, count=mcount, lang=lang,
241
- max_id=self.handler.max_id, result_type='recent')
242
- except TwythonRateLimitError as e:
243
- print("Waiting for 15 minutes -{0}".format(e))
244
- time.sleep(15*60) # wait 15 minutes
245
- continue
246
- except TwythonError as e:
247
- print("Fatal error in Twython request -{0}".format(e))
248
- if retries_after_twython_exception == retries:
249
- raise e
250
- retries += 1
251
-
252
- count = len(results['statuses'])
253
- if count == 0:
254
- print("No more Tweets available through rest api")
255
- return
256
- count_from_query += count
257
- # the max_id is also present in the Tweet metadata
258
- # results['search_metadata']['next_results'], but as part of a
259
- # query and difficult to fetch. This is doing the equivalent
260
- # (last tweet id minus one)
261
- self.handler.max_id = results['statuses'][count - 1]['id'] - 1
262
-
263
- for result in results['statuses']:
264
- yield result
265
- self.handler.counter += 1
266
- if self.handler.do_continue() == False:
267
- return
268
-
269
- def user_info_from_id(self, userids):
270
- """
271
- Convert a list of userIDs into a variety of information about the users.
272
-
273
- See <https://dev.twitter.com/rest/reference/get/users/show>.
274
-
275
- :param list userids: A list of integer strings corresponding to Twitter userIDs
276
- :rtype: list(json)
277
- """
278
- return [self.show_user(user_id=userid) for userid in userids]
279
-
280
- def user_tweets(self, screen_name, limit, include_rts='false'):
281
- """
282
- Return a collection of the most recent Tweets posted by the user
283
-
284
- :param str user: The user's screen name; the initial '@' symbol\
285
- should be omitted
286
- :param int limit: The number of Tweets to recover; 200 is the maximum allowed
287
- :param str include_rts: Whether to include statuses which have been\
288
- retweeted by the user; possible values are 'true' and 'false'
289
- """
290
- data = self.get_user_timeline(screen_name=screen_name, count=limit,
291
- include_rts=include_rts)
292
- for item in data:
293
- self.handler.handle(item)
294
-
295
-
296
-
297
-
298
- class Twitter(object):
299
- """
300
- Wrapper class with restricted functionality and fewer options.
301
- """
302
- def __init__(self):
303
- self._oauth = credsfromfile()
304
- self.streamer = Streamer(**self._oauth)
305
- self.query = Query(**self._oauth)
306
-
307
-
308
- def tweets(self, keywords='', follow='', to_screen=True, stream=True,
309
- limit=100, date_limit=None, lang='en', repeat=False,
310
- gzip_compress=False):
311
- """
312
- Process some Tweets in a simple manner.
313
-
314
- :param str keywords: Keywords to use for searching or filtering
315
- :param list follow: UserIDs to use for filtering Tweets from the public stream
316
- :param bool to_screen: If `True`, display the tweet texts on the screen,\
317
- otherwise print to a file
318
-
319
- :param bool stream: If `True`, use the live public stream,\
320
- otherwise search past public Tweets
321
-
322
- :param int limit: The number of data items to process in the current\
323
- round of processing.
324
-
325
- :param tuple date_limit: The date at which to stop collecting\
326
- new data. This should be entered as a tuple which can serve as the\
327
- argument to `datetime.datetime`.\
328
- E.g. `date_limit=(2015, 4, 1, 12, 40)` for 12:30 pm on April 1 2015.
329
- Note that, in the case of streaming, this is the maximum date, i.e.\
330
- a date in the future; if not, it is the minimum date, i.e. a date\
331
- in the past
332
-
333
- :param str lang: language
334
-
335
- :param bool repeat: A flag to determine whether multiple files should\
336
- be written. If `True`, the length of each file will be set by the\
337
- value of `limit`. Use only if `to_screen` is `False`. See also
338
- :py:func:`handle`.
339
-
340
- :param gzip_compress: if `True`, output files are compressed with gzip.
341
- """
342
- if stream:
343
- upper_date_limit = date_limit
344
- lower_date_limit = None
345
- else:
346
- upper_date_limit = None
347
- lower_date_limit = date_limit
348
-
349
- if to_screen:
350
- handler = TweetViewer(limit=limit,
351
- upper_date_limit=upper_date_limit,
352
- lower_date_limit=lower_date_limit)
353
- else:
354
- handler = TweetWriter(limit=limit,
355
- upper_date_limit=upper_date_limit,
356
- lower_date_limit=lower_date_limit, repeat=repeat,
357
- gzip_compress=gzip_compress)
358
-
359
-
360
-
361
- if to_screen:
362
- handler = TweetViewer(limit=limit)
363
- else:
364
- if stream:
365
- upper_date_limit = date_limit
366
- lower_date_limit = None
367
- else:
368
- upper_date_limit = None
369
- lower_date_limit = date_limit
370
-
371
- handler = TweetWriter(limit=limit, upper_date_limit=upper_date_limit,
372
- lower_date_limit=lower_date_limit, repeat=repeat,
373
- gzip_compress=gzip_compress)
374
-
375
- if stream:
376
- self.streamer.register(handler)
377
- if keywords == '' and follow == '':
378
- self.streamer.sample()
379
- else:
380
- self.streamer.filter(track=keywords, follow=follow, lang=lang)
381
- else:
382
- self.query.register(handler)
383
- if keywords == '':
384
- raise ValueError("Please supply at least one keyword to search for.")
385
- else:
386
- self.query._search_tweets(keywords, limit=limit, lang=lang)
387
-
388
-
389
-
390
- class TweetViewer(TweetHandlerI):
391
- """
392
- Handle data by sending it to the terminal.
393
- """
394
-
395
- def handle(self, data):
396
- """
397
- Direct data to `sys.stdout`
398
-
399
- :return: return ``False`` if processing should cease, otherwise return ``True``.
400
- :rtype: bool
401
- :param data: Tweet object returned by Twitter API
402
- """
403
- text = data['text']
404
- print(text)
405
-
406
- self.check_date_limit(data)
407
- if self.do_stop:
408
- return
409
-
410
- def on_finish(self):
411
- print('Written {0} Tweets'.format(self.counter))
412
-
413
-
414
- class TweetWriter(TweetHandlerI):
415
- """
416
- Handle data by writing it to a file.
417
- """
418
- def __init__(self, limit=2000, upper_date_limit=None, lower_date_limit=None,
419
- fprefix='tweets', subdir='twitter-files', repeat=False,
420
- gzip_compress=False):
421
- """
422
- The difference between the upper and lower date limits depends on
423
- whether Tweets are coming in an ascending date order (i.e. when
424
- streaming) or descending date order (i.e. when searching past Tweets).
425
-
426
- :param int limit: number of data items to process in the current\
427
- round of processing.
428
-
429
- :param tuple upper_date_limit: The date at which to stop collecting new\
430
- data. This should be entered as a tuple which can serve as the\
431
- argument to `datetime.datetime`. E.g. `upper_date_limit=(2015, 4, 1, 12,\
432
- 40)` for 12:30 pm on April 1 2015.
433
-
434
- :param tuple lower_date_limit: The date at which to stop collecting new\
435
- data. See `upper_data_limit` for formatting.
436
-
437
- :param str fprefix: The prefix to use in creating file names for Tweet\
438
- collections.
439
-
440
- :param str subdir: The name of the directory where Tweet collection\
441
- files should be stored.
442
-
443
- :param bool repeat: flag to determine whether multiple files should be\
444
- written. If `True`, the length of each file will be set by the value\
445
- of `limit`. See also :py:func:`handle`.
446
-
447
- :param gzip_compress: if `True`, ouput files are compressed with gzip.
448
- """
449
- self.fprefix = fprefix
450
- self.subdir = guess_path(subdir)
451
- self.gzip_compress = gzip_compress
452
- self.fname = self.timestamped_file()
453
- self.repeat = repeat
454
- self.output = None
455
- TweetHandlerI.__init__(self, limit, upper_date_limit, lower_date_limit)
456
-
457
-
458
- def timestamped_file(self):
459
- """
460
- :return: timestamped file name
461
- :rtype: str
462
- """
463
- subdir = self.subdir
464
- fprefix = self.fprefix
465
- if subdir:
466
- if not os.path.exists(subdir):
467
- os.mkdir(subdir)
468
-
469
- fname = os.path.join(subdir, fprefix)
470
- fmt = '%Y%m%d-%H%M%S'
471
- timestamp = datetime.datetime.now().strftime(fmt)
472
- if self.gzip_compress:
473
- suffix = '.gz'
474
- else:
475
- suffix = ''
476
- outfile = '{0}.{1}.json{2}'.format(fname, timestamp, suffix)
477
- return outfile
478
-
479
-
480
- def handle(self, data):
481
- """
482
- Write Twitter data as line-delimited JSON into one or more files.
483
-
484
- :return: return `False` if processing should cease, otherwise return `True`.
485
- :param data: tweet object returned by Twitter API
486
- """
487
- if self.startingup:
488
- if self.gzip_compress:
489
- self.output = gzip.open(self.fname, 'w')
490
- else:
491
- self.output = open(self.fname, 'w')
492
- print('Writing to {0}'.format(self.fname))
493
-
494
- json_data = json.dumps(data)
495
- if self.gzip_compress:
496
- self.output.write((json_data + "\n").encode('utf-8'))
497
- else:
498
- self.output.write(json_data + "\n")
499
-
500
- self.check_date_limit(data)
501
- if self.do_stop:
502
- return
503
-
504
- self.startingup = False
505
-
506
- def on_finish(self):
507
- print('Written {0} Tweets'.format(self.counter))
508
- if self.output:
509
- self.output.close()
510
-
511
- def do_continue(self):
512
- if self.repeat == False:
513
- return TweetHandlerI.do_continue(self)
514
-
515
- if self.do_stop:
516
- # stop for a functional cause (e.g. date limit)
517
- return False
518
-
519
- if self.counter == self.limit:
520
- # repeat is True, thus close output file and
521
- # create a new one
522
- self._restart_file()
523
- return True
524
-
525
-
526
- def _restart_file(self):
527
- self.on_finish()
528
- self.fname = self.timestamped_file()
529
- self.startingup = True
530
- self.counter = 0
1
+ # -*- coding: utf-8 -*-
2
+ # Natural Language Toolkit: Twitter client
3
+ #
4
+ # Copyright (C) 2001-2018 NLTK Project
5
+ # Author: Ewan Klein <ewan@inf.ed.ac.uk>
6
+ # Lorenzo Rubio <lrnzcig@gmail.com>
7
+ # URL: <http://nltk.org/>
8
+ # For license information, see LICENSE.TXT
9
+
10
+
11
+ """
12
+ NLTK Twitter client
13
+
14
+ This module offers methods for collecting and processing Tweets. Most of the
15
+ functionality depends on access to the Twitter APIs, and this is handled via
16
+ the third party Twython library.
17
+
18
+ If one of the methods below returns an integer, it is probably a `Twitter
19
+ error code <https://dev.twitter.com/overview/api/response-codes>`_. For
20
+ example, the response of '420' means that you have reached the limit of the
21
+ requests you can currently make to the Twitter API. Currently, `rate limits
22
+ for the search API <https://dev.twitter.com/rest/public/rate-limiting>`_ are
23
+ divided into 15 minute windows.
24
+ """
25
+
26
+ import datetime
27
+ import itertools
28
+ import json
29
+ import os
30
+ import requests
31
+ import time
32
+ import gzip
33
+
34
+
35
+ from twython import Twython, TwythonStreamer
36
+ from twython.exceptions import TwythonRateLimitError, TwythonError
37
+
38
+ from nltk.twitter.util import credsfromfile, guess_path
39
+ from nltk.twitter.api import TweetHandlerI, BasicTweetHandler
40
+
41
+
42
+
43
+ class Streamer(TwythonStreamer):
44
+ """
45
+ Retrieve data from the Twitter Streaming API.
46
+
47
+ The streaming API requires
48
+ `OAuth 1.0 <http://en.wikipedia.org/wiki/OAuth>`_ authentication.
49
+ """
50
+ def __init__(self, app_key, app_secret, oauth_token, oauth_token_secret):
51
+
52
+ self.handler = None
53
+ self.do_continue = True
54
+ TwythonStreamer.__init__(self, app_key, app_secret, oauth_token,
55
+ oauth_token_secret)
56
+
57
+ def register(self, handler):
58
+ """
59
+ Register a method for handling Tweets.
60
+
61
+ :param TweetHandlerI handler: method for viewing
62
+ """
63
+ self.handler = handler
64
+
65
+ def on_success(self, data):
66
+ """
67
+ :param data: response from Twitter API
68
+ """
69
+ if self.do_continue:
70
+ if self.handler is not None:
71
+ if 'text' in data:
72
+ self.handler.counter += 1
73
+ self.handler.handle(data)
74
+ self.do_continue = self.handler.do_continue()
75
+ else:
76
+ raise ValueError("No data handler has been registered.")
77
+ else:
78
+ self.disconnect()
79
+ self.handler.on_finish()
80
+
81
+
82
+ def on_error(self, status_code, data):
83
+ """
84
+ :param status_code: The status code returned by the Twitter API
85
+ :param data: The response from Twitter API
86
+
87
+ """
88
+ print(status_code)
89
+
90
+ def sample(self):
91
+ """
92
+ Wrapper for 'statuses / sample' API call
93
+ """
94
+ while self.do_continue:
95
+
96
+ # Stream in an endless loop until limit is reached. See twython
97
+ # issue 288: https://github.com/ryanmcgrath/twython/issues/288
98
+ # colditzjb commented on 9 Dec 2014
99
+
100
+ try:
101
+ self.statuses.sample()
102
+ except requests.exceptions.ChunkedEncodingError as e:
103
+ if e is not None:
104
+ print("Error (stream will continue): {0}".format(e))
105
+ continue
106
+
107
+ def filter(self, track='', follow='', lang='en'):
108
+ """
109
+ Wrapper for 'statuses / filter' API call
110
+ """
111
+ while self.do_continue:
112
+ #Stream in an endless loop until limit is reached
113
+
114
+ try:
115
+ if track == '' and follow == '':
116
+ msg = "Please supply a value for 'track', 'follow'"
117
+ raise ValueError(msg)
118
+ self.statuses.filter(track=track, follow=follow, lang=lang)
119
+ except requests.exceptions.ChunkedEncodingError as e:
120
+ if e is not None:
121
+ print("Error (stream will continue): {0}".format(e))
122
+ continue
123
+
124
+
125
+ class Query(Twython):
126
+ """
127
+ Retrieve data from the Twitter REST API.
128
+ """
129
+ def __init__(self, app_key, app_secret, oauth_token,
130
+ oauth_token_secret):
131
+ self.handler = None
132
+ self.do_continue = True
133
+ Twython.__init__(self, app_key, app_secret, oauth_token, oauth_token_secret)
134
+
135
+ def register(self, handler):
136
+ """
137
+ Register a method for handling Tweets.
138
+
139
+ :param TweetHandlerI handler: method for viewing or writing Tweets to a file.
140
+ """
141
+ self.handler = handler
142
+
143
+ def expand_tweetids(self, ids_f, verbose=True):
144
+ """
145
+ Given a file object containing a list of Tweet IDs, fetch the
146
+ corresponding full Tweets from the Twitter API.
147
+
148
+ The API call `statuses/lookup` will fail to retrieve a Tweet if the
149
+ user has deleted it.
150
+
151
+ This call to the Twitter API is rate-limited. See
152
+ <https://dev.twitter.com/rest/reference/get/statuses/lookup> for details.
153
+
154
+ :param ids_f: input file object consisting of Tweet IDs, one to a line
155
+ :return: iterable of Tweet objects in JSON format
156
+ """
157
+ ids = [line.strip() for line in ids_f if line]
158
+
159
+ if verbose:
160
+ print("Counted {0} Tweet IDs in {1}.".format(len(ids), ids_f))
161
+
162
+ # The Twitter endpoint takes lists of up to 100 ids, so we chunk the
163
+ # ids.
164
+ id_chunks = [ids[i:i+100] for i in range(0, len(ids), 100)]
165
+
166
+ chunked_tweets = (self.lookup_status(id=chunk) for chunk in
167
+ id_chunks)
168
+
169
+ return itertools.chain.from_iterable(chunked_tweets)
170
+
171
+
172
+
173
+ def _search_tweets(self, keywords, limit=100, lang='en'):
174
+ """
175
+ Assumes that the handler has been informed. Fetches Tweets from
176
+ search_tweets generator output and passses them to handler
177
+
178
+ :param str keywords: A list of query terms to search for, written as\
179
+ a comma-separated string.
180
+ :param int limit: Number of Tweets to process
181
+ :param str lang: language
182
+ """
183
+ while True:
184
+ tweets = self.search_tweets(keywords=keywords, limit=limit, lang=lang,
185
+ max_id=self.handler.max_id)
186
+ for tweet in tweets:
187
+ self.handler.handle(tweet)
188
+ if not (self.handler.do_continue() and self.handler.repeat):
189
+ break
190
+ self.handler.on_finish()
191
+
192
+ def search_tweets(self, keywords, limit=100, lang='en', max_id=None,
193
+ retries_after_twython_exception=0):
194
+ """
195
+ Call the REST API ``'search/tweets'`` endpoint with some plausible
196
+ defaults. See `the Twitter search documentation
197
+ <https://dev.twitter.com/rest/public/search>`_ for more information
198
+ about admissable search parameters.
199
+
200
+ :param str keywords: A list of query terms to search for, written as\
201
+ a comma-separated string
202
+ :param int limit: Number of Tweets to process
203
+ :param str lang: language
204
+ :param int max_id: id of the last tweet fetched
205
+ :param int retries_after_twython_exception: number of retries when\
206
+ searching Tweets before raising an exception
207
+ :rtype: python generator
208
+ """
209
+ if not self.handler:
210
+ # if no handler is provided, `BasicTweetHandler` provides minimum
211
+ # functionality for limiting the number of Tweets retrieved
212
+ self.handler = BasicTweetHandler(limit=limit)
213
+
214
+ count_from_query = 0
215
+ if max_id:
216
+ self.handler.max_id = max_id
217
+ else:
218
+ results = self.search(q=keywords, count=min(100, limit), lang=lang,
219
+ result_type='recent')
220
+ count = len(results['statuses'])
221
+ if count == 0:
222
+ print("No Tweets available through REST API for those keywords")
223
+ return
224
+ count_from_query = count
225
+ self.handler.max_id = results['statuses'][count - 1]['id'] - 1
226
+
227
+ for result in results['statuses']:
228
+ yield result
229
+ self.handler.counter += 1
230
+ if self.handler.do_continue() == False:
231
+ return
232
+
233
+
234
+ # Pagination loop: keep fetching Tweets until the desired count is
235
+ # reached while dealing with Twitter rate limits.
236
+ retries = 0
237
+ while count_from_query < limit:
238
+ try:
239
+ mcount = min(100, limit-count_from_query)
240
+ results = self.search(q=keywords, count=mcount, lang=lang,
241
+ max_id=self.handler.max_id, result_type='recent')
242
+ except TwythonRateLimitError as e:
243
+ print("Waiting for 15 minutes -{0}".format(e))
244
+ time.sleep(15*60) # wait 15 minutes
245
+ continue
246
+ except TwythonError as e:
247
+ print("Fatal error in Twython request -{0}".format(e))
248
+ if retries_after_twython_exception == retries:
249
+ raise e
250
+ retries += 1
251
+
252
+ count = len(results['statuses'])
253
+ if count == 0:
254
+ print("No more Tweets available through rest api")
255
+ return
256
+ count_from_query += count
257
+ # the max_id is also present in the Tweet metadata
258
+ # results['search_metadata']['next_results'], but as part of a
259
+ # query and difficult to fetch. This is doing the equivalent
260
+ # (last tweet id minus one)
261
+ self.handler.max_id = results['statuses'][count - 1]['id'] - 1
262
+
263
+ for result in results['statuses']:
264
+ yield result
265
+ self.handler.counter += 1
266
+ if self.handler.do_continue() == False:
267
+ return
268
+
269
+ def user_info_from_id(self, userids):
270
+ """
271
+ Convert a list of userIDs into a variety of information about the users.
272
+
273
+ See <https://dev.twitter.com/rest/reference/get/users/show>.
274
+
275
+ :param list userids: A list of integer strings corresponding to Twitter userIDs
276
+ :rtype: list(json)
277
+ """
278
+ return [self.show_user(user_id=userid) for userid in userids]
279
+
280
+ def user_tweets(self, screen_name, limit, include_rts='false'):
281
+ """
282
+ Return a collection of the most recent Tweets posted by the user
283
+
284
+ :param str user: The user's screen name; the initial '@' symbol\
285
+ should be omitted
286
+ :param int limit: The number of Tweets to recover; 200 is the maximum allowed
287
+ :param str include_rts: Whether to include statuses which have been\
288
+ retweeted by the user; possible values are 'true' and 'false'
289
+ """
290
+ data = self.get_user_timeline(screen_name=screen_name, count=limit,
291
+ include_rts=include_rts)
292
+ for item in data:
293
+ self.handler.handle(item)
294
+
295
+
296
+
297
+
298
+ class Twitter(object):
299
+ """
300
+ Wrapper class with restricted functionality and fewer options.
301
+ """
302
+ def __init__(self):
303
+ self._oauth = credsfromfile()
304
+ self.streamer = Streamer(**self._oauth)
305
+ self.query = Query(**self._oauth)
306
+
307
+
308
+ def tweets(self, keywords='', follow='', to_screen=True, stream=True,
309
+ limit=100, date_limit=None, lang='en', repeat=False,
310
+ gzip_compress=False):
311
+ """
312
+ Process some Tweets in a simple manner.
313
+
314
+ :param str keywords: Keywords to use for searching or filtering
315
+ :param list follow: UserIDs to use for filtering Tweets from the public stream
316
+ :param bool to_screen: If `True`, display the tweet texts on the screen,\
317
+ otherwise print to a file
318
+
319
+ :param bool stream: If `True`, use the live public stream,\
320
+ otherwise search past public Tweets
321
+
322
+ :param int limit: The number of data items to process in the current\
323
+ round of processing.
324
+
325
+ :param tuple date_limit: The date at which to stop collecting\
326
+ new data. This should be entered as a tuple which can serve as the\
327
+ argument to `datetime.datetime`.\
328
+ E.g. `date_limit=(2015, 4, 1, 12, 40)` for 12:30 pm on April 1 2015.
329
+ Note that, in the case of streaming, this is the maximum date, i.e.\
330
+ a date in the future; if not, it is the minimum date, i.e. a date\
331
+ in the past
332
+
333
+ :param str lang: language
334
+
335
+ :param bool repeat: A flag to determine whether multiple files should\
336
+ be written. If `True`, the length of each file will be set by the\
337
+ value of `limit`. Use only if `to_screen` is `False`. See also
338
+ :py:func:`handle`.
339
+
340
+ :param gzip_compress: if `True`, output files are compressed with gzip.
341
+ """
342
+ if stream:
343
+ upper_date_limit = date_limit
344
+ lower_date_limit = None
345
+ else:
346
+ upper_date_limit = None
347
+ lower_date_limit = date_limit
348
+
349
+ if to_screen:
350
+ handler = TweetViewer(limit=limit,
351
+ upper_date_limit=upper_date_limit,
352
+ lower_date_limit=lower_date_limit)
353
+ else:
354
+ handler = TweetWriter(limit=limit,
355
+ upper_date_limit=upper_date_limit,
356
+ lower_date_limit=lower_date_limit, repeat=repeat,
357
+ gzip_compress=gzip_compress)
358
+
359
+
360
+
361
+ if to_screen:
362
+ handler = TweetViewer(limit=limit)
363
+ else:
364
+ if stream:
365
+ upper_date_limit = date_limit
366
+ lower_date_limit = None
367
+ else:
368
+ upper_date_limit = None
369
+ lower_date_limit = date_limit
370
+
371
+ handler = TweetWriter(limit=limit, upper_date_limit=upper_date_limit,
372
+ lower_date_limit=lower_date_limit, repeat=repeat,
373
+ gzip_compress=gzip_compress)
374
+
375
+ if stream:
376
+ self.streamer.register(handler)
377
+ if keywords == '' and follow == '':
378
+ self.streamer.sample()
379
+ else:
380
+ self.streamer.filter(track=keywords, follow=follow, lang=lang)
381
+ else:
382
+ self.query.register(handler)
383
+ if keywords == '':
384
+ raise ValueError("Please supply at least one keyword to search for.")
385
+ else:
386
+ self.query._search_tweets(keywords, limit=limit, lang=lang)
387
+
388
+
389
+
390
+ class TweetViewer(TweetHandlerI):
391
+ """
392
+ Handle data by sending it to the terminal.
393
+ """
394
+
395
+ def handle(self, data):
396
+ """
397
+ Direct data to `sys.stdout`
398
+
399
+ :return: return ``False`` if processing should cease, otherwise return ``True``.
400
+ :rtype: bool
401
+ :param data: Tweet object returned by Twitter API
402
+ """
403
+ text = data['text']
404
+ print(text)
405
+
406
+ self.check_date_limit(data)
407
+ if self.do_stop:
408
+ return
409
+
410
+ def on_finish(self):
411
+ print('Written {0} Tweets'.format(self.counter))
412
+
413
+
414
+ class TweetWriter(TweetHandlerI):
415
+ """
416
+ Handle data by writing it to a file.
417
+ """
418
+ def __init__(self, limit=2000, upper_date_limit=None, lower_date_limit=None,
419
+ fprefix='tweets', subdir='twitter-files', repeat=False,
420
+ gzip_compress=False):
421
+ """
422
+ The difference between the upper and lower date limits depends on
423
+ whether Tweets are coming in an ascending date order (i.e. when
424
+ streaming) or descending date order (i.e. when searching past Tweets).
425
+
426
+ :param int limit: number of data items to process in the current\
427
+ round of processing.
428
+
429
+ :param tuple upper_date_limit: The date at which to stop collecting new\
430
+ data. This should be entered as a tuple which can serve as the\
431
+ argument to `datetime.datetime`. E.g. `upper_date_limit=(2015, 4, 1, 12,\
432
+ 40)` for 12:30 pm on April 1 2015.
433
+
434
+ :param tuple lower_date_limit: The date at which to stop collecting new\
435
+ data. See `upper_data_limit` for formatting.
436
+
437
+ :param str fprefix: The prefix to use in creating file names for Tweet\
438
+ collections.
439
+
440
+ :param str subdir: The name of the directory where Tweet collection\
441
+ files should be stored.
442
+
443
+ :param bool repeat: flag to determine whether multiple files should be\
444
+ written. If `True`, the length of each file will be set by the value\
445
+ of `limit`. See also :py:func:`handle`.
446
+
447
+ :param gzip_compress: if `True`, ouput files are compressed with gzip.
448
+ """
449
+ self.fprefix = fprefix
450
+ self.subdir = guess_path(subdir)
451
+ self.gzip_compress = gzip_compress
452
+ self.fname = self.timestamped_file()
453
+ self.repeat = repeat
454
+ self.output = None
455
+ TweetHandlerI.__init__(self, limit, upper_date_limit, lower_date_limit)
456
+
457
+
458
+ def timestamped_file(self):
459
+ """
460
+ :return: timestamped file name
461
+ :rtype: str
462
+ """
463
+ subdir = self.subdir
464
+ fprefix = self.fprefix
465
+ if subdir:
466
+ if not os.path.exists(subdir):
467
+ os.mkdir(subdir)
468
+
469
+ fname = os.path.join(subdir, fprefix)
470
+ fmt = '%Y%m%d-%H%M%S'
471
+ timestamp = datetime.datetime.now().strftime(fmt)
472
+ if self.gzip_compress:
473
+ suffix = '.gz'
474
+ else:
475
+ suffix = ''
476
+ outfile = '{0}.{1}.json{2}'.format(fname, timestamp, suffix)
477
+ return outfile
478
+
479
+
480
+ def handle(self, data):
481
+ """
482
+ Write Twitter data as line-delimited JSON into one or more files.
483
+
484
+ :return: return `False` if processing should cease, otherwise return `True`.
485
+ :param data: tweet object returned by Twitter API
486
+ """
487
+ if self.startingup:
488
+ if self.gzip_compress:
489
+ self.output = gzip.open(self.fname, 'w')
490
+ else:
491
+ self.output = open(self.fname, 'w')
492
+ print('Writing to {0}'.format(self.fname))
493
+
494
+ json_data = json.dumps(data)
495
+ if self.gzip_compress:
496
+ self.output.write((json_data + "\n").encode('utf-8'))
497
+ else:
498
+ self.output.write(json_data + "\n")
499
+
500
+ self.check_date_limit(data)
501
+ if self.do_stop:
502
+ return
503
+
504
+ self.startingup = False
505
+
506
+ def on_finish(self):
507
+ print('Written {0} Tweets'.format(self.counter))
508
+ if self.output:
509
+ self.output.close()
510
+
511
+ def do_continue(self):
512
+ if self.repeat == False:
513
+ return TweetHandlerI.do_continue(self)
514
+
515
+ if self.do_stop:
516
+ # stop for a functional cause (e.g. date limit)
517
+ return False
518
+
519
+ if self.counter == self.limit:
520
+ # repeat is True, thus close output file and
521
+ # create a new one
522
+ self._restart_file()
523
+ return True
524
+
525
+
526
+ def _restart_file(self):
527
+ self.on_finish()
528
+ self.fname = self.timestamped_file()
529
+ self.startingup = True
530
+ self.counter = 0