arekit 0.24.0__tar.gz → 0.25.0__tar.gz

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 (391) hide show
  1. arekit-0.25.0/PKG-INFO +82 -0
  2. arekit-0.25.0/README.md +60 -0
  3. {arekit-0.24.0 → arekit-0.25.0}/arekit/common/docs/entities_grouping.py +2 -1
  4. arekit-0.25.0/arekit/common/docs/parser.py +66 -0
  5. arekit-0.25.0/arekit/common/pipeline/base.py +21 -0
  6. arekit-0.25.0/arekit/common/pipeline/batching.py +28 -0
  7. {arekit-0.24.0 → arekit-0.25.0}/arekit/common/pipeline/context.py +5 -1
  8. arekit-0.25.0/arekit/common/pipeline/items/base.py +49 -0
  9. {arekit-0.24.0 → arekit-0.25.0}/arekit/common/pipeline/items/flatten.py +5 -1
  10. {arekit-0.24.0 → arekit-0.25.0}/arekit/common/pipeline/items/handle.py +2 -1
  11. {arekit-0.24.0 → arekit-0.25.0}/arekit/common/pipeline/items/iter.py +2 -1
  12. {arekit-0.24.0 → arekit-0.25.0}/arekit/common/pipeline/items/map.py +2 -1
  13. {arekit-0.24.0 → arekit-0.25.0}/arekit/common/pipeline/items/map_nested.py +4 -0
  14. arekit-0.25.0/arekit/common/pipeline/utils.py +32 -0
  15. arekit-0.25.0/arekit/common/service/sqlite.py +36 -0
  16. arekit-0.24.0/arekit/common/text/partitioning/str.py → arekit-0.25.0/arekit/common/text/partitioning.py +14 -9
  17. {arekit-0.24.0 → arekit-0.25.0}/arekit/common/utils.py +0 -44
  18. {arekit-0.24.0 → arekit-0.25.0}/arekit/contrib/utils/data/contents/opinions.py +13 -3
  19. arekit-0.25.0/arekit/contrib/utils/data/readers/sqlite.py +14 -0
  20. {arekit-0.24.0 → arekit-0.25.0}/arekit/contrib/utils/data/storages/row_cache.py +6 -1
  21. arekit-0.25.0/arekit/contrib/utils/data/storages/sqlite_based.py +17 -0
  22. {arekit-0.24.0 → arekit-0.25.0}/arekit/contrib/utils/data/writers/sqlite_native.py +4 -0
  23. {arekit-0.24.0 → arekit-0.25.0}/arekit/contrib/utils/io_utils/utils.py +1 -18
  24. {arekit-0.24.0 → arekit-0.25.0}/arekit/contrib/utils/pipelines/items/sampling/base.py +7 -12
  25. {arekit-0.24.0 → arekit-0.25.0}/arekit/contrib/utils/pipelines/items/sampling/networks.py +3 -2
  26. {arekit-0.24.0 → arekit-0.25.0}/arekit/contrib/utils/pipelines/items/text/entities_default.py +2 -2
  27. {arekit-0.24.0 → arekit-0.25.0}/arekit/contrib/utils/pipelines/items/text/frames.py +2 -3
  28. {arekit-0.24.0 → arekit-0.25.0}/arekit/contrib/utils/pipelines/items/text/frames_lemmatized.py +2 -2
  29. {arekit-0.24.0 → arekit-0.25.0}/arekit/contrib/utils/pipelines/items/text/frames_negation.py +2 -1
  30. {arekit-0.24.0 → arekit-0.25.0}/arekit/contrib/utils/pipelines/items/text/tokenizer.py +2 -4
  31. {arekit-0.24.0 → arekit-0.25.0}/arekit/contrib/utils/pipelines/items/text/translator.py +2 -1
  32. {arekit-0.24.0 → arekit-0.25.0}/arekit/contrib/utils/pipelines/text_opinion/extraction.py +6 -9
  33. {arekit-0.24.0 → arekit-0.25.0}/arekit/contrib/utils/serializer.py +1 -2
  34. arekit-0.25.0/arekit.egg-info/PKG-INFO +82 -0
  35. {arekit-0.24.0 → arekit-0.25.0}/arekit.egg-info/SOURCES.txt +8 -123
  36. arekit-0.25.0/logo.png +0 -0
  37. {arekit-0.24.0 → arekit-0.25.0}/setup.py +8 -3
  38. arekit-0.24.0/PKG-INFO +0 -19
  39. arekit-0.24.0/README.md +0 -56
  40. arekit-0.24.0/arekit/common/docs/objects_parser.py +0 -37
  41. arekit-0.24.0/arekit/common/docs/parser.py +0 -34
  42. arekit-0.24.0/arekit/common/pipeline/base.py +0 -25
  43. arekit-0.24.0/arekit/common/pipeline/items/base.py +0 -12
  44. arekit-0.24.0/arekit/common/text/parser.py +0 -12
  45. arekit-0.24.0/arekit/common/text/partitioning/base.py +0 -4
  46. arekit-0.24.0/arekit/common/text/partitioning/terms.py +0 -35
  47. arekit-0.24.0/arekit/contrib/source/brat/annot.py +0 -84
  48. arekit-0.24.0/arekit/contrib/source/brat/doc.py +0 -28
  49. arekit-0.24.0/arekit/contrib/source/brat/entities/compound.py +0 -13
  50. arekit-0.24.0/arekit/contrib/source/brat/entities/entity.py +0 -42
  51. arekit-0.24.0/arekit/contrib/source/brat/entities/parser.py +0 -53
  52. arekit-0.24.0/arekit/contrib/source/brat/opinions/converter.py +0 -19
  53. arekit-0.24.0/arekit/contrib/source/brat/relation.py +0 -32
  54. arekit-0.24.0/arekit/contrib/source/brat/sentence.py +0 -69
  55. arekit-0.24.0/arekit/contrib/source/brat/sentences_reader.py +0 -128
  56. arekit-0.24.0/arekit/contrib/source/download.py +0 -41
  57. arekit-0.24.0/arekit/contrib/source/nerel/entities.py +0 -55
  58. arekit-0.24.0/arekit/contrib/source/nerel/folding/fixed.py +0 -74
  59. arekit-0.24.0/arekit/contrib/source/nerel/io_utils.py +0 -62
  60. arekit-0.24.0/arekit/contrib/source/nerel/labels.py +0 -241
  61. arekit-0.24.0/arekit/contrib/source/nerel/reader.py +0 -46
  62. arekit-0.24.0/arekit/contrib/source/nerel/utils.py +0 -24
  63. arekit-0.24.0/arekit/contrib/source/nerel/versions.py +0 -12
  64. arekit-0.24.0/arekit/contrib/source/nerelbio/io_utils.py +0 -62
  65. arekit-0.24.0/arekit/contrib/source/nerelbio/labels.py +0 -265
  66. arekit-0.24.0/arekit/contrib/source/nerelbio/reader.py +0 -8
  67. arekit-0.24.0/arekit/contrib/source/nerelbio/versions.py +0 -8
  68. arekit-0.24.0/arekit/contrib/source/ruattitudes/collection.py +0 -36
  69. arekit-0.24.0/arekit/contrib/source/ruattitudes/doc.py +0 -51
  70. arekit-0.24.0/arekit/contrib/source/ruattitudes/doc_brat.py +0 -44
  71. arekit-0.24.0/arekit/contrib/source/ruattitudes/entity/parser.py +0 -7
  72. arekit-0.24.0/arekit/contrib/source/ruattitudes/io_utils.py +0 -56
  73. arekit-0.24.0/arekit/contrib/source/ruattitudes/labels_fmt.py +0 -12
  74. arekit-0.24.0/arekit/contrib/source/ruattitudes/opinions/base.py +0 -28
  75. arekit-0.24.0/arekit/contrib/source/ruattitudes/opinions/converter.py +0 -37
  76. arekit-0.24.0/arekit/contrib/source/ruattitudes/reader.py +0 -268
  77. arekit-0.24.0/arekit/contrib/source/ruattitudes/sentence.py +0 -73
  78. arekit-0.24.0/arekit/contrib/source/ruattitudes/synonyms.py +0 -17
  79. arekit-0.24.0/arekit/contrib/source/ruattitudes/text_object.py +0 -59
  80. arekit-0.24.0/arekit/contrib/source/rusentiframes/collection.py +0 -157
  81. arekit-0.24.0/arekit/contrib/source/rusentiframes/effect.py +0 -24
  82. arekit-0.24.0/arekit/contrib/source/rusentiframes/io_utils.py +0 -19
  83. arekit-0.24.0/arekit/contrib/source/rusentiframes/labels_fmt.py +0 -22
  84. arekit-0.24.0/arekit/contrib/source/rusentiframes/polarity.py +0 -35
  85. arekit-0.24.0/arekit/contrib/source/rusentiframes/role.py +0 -15
  86. arekit-0.24.0/arekit/contrib/source/rusentiframes/state.py +0 -24
  87. arekit-0.24.0/arekit/contrib/source/rusentiframes/types.py +0 -42
  88. arekit-0.24.0/arekit/contrib/source/rusentiframes/value.py +0 -2
  89. arekit-0.24.0/arekit/contrib/source/rusentrel/const.py +0 -3
  90. arekit-0.24.0/arekit/contrib/source/rusentrel/docs_reader.py +0 -51
  91. arekit-0.24.0/arekit/contrib/source/rusentrel/entities.py +0 -26
  92. arekit-0.24.0/arekit/contrib/source/rusentrel/io_utils.py +0 -125
  93. arekit-0.24.0/arekit/contrib/source/rusentrel/labels_fmt.py +0 -12
  94. arekit-0.24.0/arekit/contrib/source/rusentrel/opinions/collection.py +0 -30
  95. arekit-0.24.0/arekit/contrib/source/rusentrel/opinions/converter.py +0 -40
  96. arekit-0.24.0/arekit/contrib/source/rusentrel/opinions/provider.py +0 -54
  97. arekit-0.24.0/arekit/contrib/source/rusentrel/opinions/writer.py +0 -42
  98. arekit-0.24.0/arekit/contrib/source/rusentrel/synonyms.py +0 -17
  99. arekit-0.24.0/arekit/contrib/source/sentinerel/entities.py +0 -52
  100. arekit-0.24.0/arekit/contrib/source/sentinerel/folding/factory.py +0 -31
  101. arekit-0.24.0/arekit/contrib/source/sentinerel/folding/fixed.py +0 -70
  102. arekit-0.24.0/arekit/contrib/source/sentinerel/io_utils.py +0 -87
  103. arekit-0.24.0/arekit/contrib/source/sentinerel/labels.py +0 -53
  104. arekit-0.24.0/arekit/contrib/source/sentinerel/labels_scaler.py +0 -30
  105. arekit-0.24.0/arekit/contrib/source/sentinerel/reader.py +0 -42
  106. arekit-0.24.0/arekit/contrib/source/synonyms/utils.py +0 -19
  107. arekit-0.24.0/arekit/contrib/source/zip_utils.py +0 -47
  108. arekit-0.24.0/arekit/contrib/utils/connotations/rusentiframes_sentiment.py +0 -23
  109. arekit-0.24.0/arekit/contrib/utils/download.py +0 -77
  110. arekit-0.24.0/arekit/contrib/utils/io_utils/__init__.py +0 -0
  111. arekit-0.24.0/arekit/contrib/utils/io_utils/opinions.py +0 -37
  112. arekit-0.24.0/arekit/contrib/utils/io_utils/samples.py +0 -79
  113. arekit-0.24.0/arekit/contrib/utils/lexicons/__init__.py +0 -0
  114. arekit-0.24.0/arekit/contrib/utils/lexicons/lexicon.py +0 -41
  115. arekit-0.24.0/arekit/contrib/utils/lexicons/relation.py +0 -42
  116. arekit-0.24.0/arekit/contrib/utils/lexicons/rusentilex.py +0 -37
  117. arekit-0.24.0/arekit/contrib/utils/nn/__init__.py +0 -0
  118. arekit-0.24.0/arekit/contrib/utils/nn/rows.py +0 -83
  119. arekit-0.24.0/arekit/contrib/utils/np_utils/__init__.py +0 -0
  120. arekit-0.24.0/arekit/contrib/utils/pipelines/__init__.py +0 -0
  121. arekit-0.24.0/arekit/contrib/utils/pipelines/items/__init__.py +0 -0
  122. arekit-0.24.0/arekit/contrib/utils/pipelines/items/sampling/__init__.py +0 -0
  123. arekit-0.24.0/arekit/contrib/utils/pipelines/items/text/__init__.py +0 -0
  124. arekit-0.24.0/arekit/contrib/utils/pipelines/items/text/terms_splitter.py +0 -10
  125. arekit-0.24.0/arekit/contrib/utils/pipelines/sources/__init__.py +0 -0
  126. arekit-0.24.0/arekit/contrib/utils/pipelines/sources/nerel/__init__.py +0 -0
  127. arekit-0.24.0/arekit/contrib/utils/pipelines/sources/nerel/doc_provider.py +0 -27
  128. arekit-0.24.0/arekit/contrib/utils/pipelines/sources/nerel/extract_text_relations.py +0 -65
  129. arekit-0.24.0/arekit/contrib/utils/pipelines/sources/nerel/labels_fmt.py +0 -60
  130. arekit-0.24.0/arekit/contrib/utils/pipelines/sources/nerel_bio/__init__.py +0 -0
  131. arekit-0.24.0/arekit/contrib/utils/pipelines/sources/nerel_bio/doc_provider.py +0 -29
  132. arekit-0.24.0/arekit/contrib/utils/pipelines/sources/nerel_bio/extrat_text_relations.py +0 -64
  133. arekit-0.24.0/arekit/contrib/utils/pipelines/sources/nerel_bio/labels_fmt.py +0 -79
  134. arekit-0.24.0/arekit/contrib/utils/pipelines/sources/ruattitudes/__init__.py +0 -0
  135. arekit-0.24.0/arekit/contrib/utils/pipelines/sources/ruattitudes/doc_provider.py +0 -56
  136. arekit-0.24.0/arekit/contrib/utils/pipelines/sources/ruattitudes/entity_filter.py +0 -20
  137. arekit-0.24.0/arekit/contrib/utils/pipelines/sources/ruattitudes/extract_text_opinions.py +0 -65
  138. arekit-0.24.0/arekit/contrib/utils/pipelines/sources/rusentrel/__init__.py +0 -0
  139. arekit-0.24.0/arekit/contrib/utils/pipelines/sources/rusentrel/doc_provider.py +0 -21
  140. arekit-0.24.0/arekit/contrib/utils/pipelines/sources/rusentrel/extract_text_opinions.py +0 -107
  141. arekit-0.24.0/arekit/contrib/utils/pipelines/sources/sentinerel/__init__.py +0 -0
  142. arekit-0.24.0/arekit/contrib/utils/pipelines/sources/sentinerel/doc_provider.py +0 -29
  143. arekit-0.24.0/arekit/contrib/utils/pipelines/sources/sentinerel/entity_filter.py +0 -62
  144. arekit-0.24.0/arekit/contrib/utils/pipelines/sources/sentinerel/extract_text_opinions.py +0 -180
  145. arekit-0.24.0/arekit/contrib/utils/pipelines/sources/sentinerel/labels_fmt.py +0 -50
  146. arekit-0.24.0/arekit/contrib/utils/pipelines/text_opinion/__init__.py +0 -0
  147. arekit-0.24.0/arekit/contrib/utils/pipelines/text_opinion/annot/__init__.py +0 -0
  148. arekit-0.24.0/arekit/contrib/utils/pipelines/text_opinion/annot/predefined.py +0 -88
  149. arekit-0.24.0/arekit/contrib/utils/pipelines/text_opinion/filters/__init__.py +0 -0
  150. arekit-0.24.0/arekit/contrib/utils/processing/__init__.py +0 -0
  151. arekit-0.24.0/arekit/contrib/utils/processing/languages/__init__.py +0 -0
  152. arekit-0.24.0/arekit/contrib/utils/processing/languages/ru/__init__.py +0 -0
  153. arekit-0.24.0/arekit/contrib/utils/processing/lemmatization/__init__.py +0 -0
  154. arekit-0.24.0/arekit/contrib/utils/processing/pos/__init__.py +0 -0
  155. arekit-0.24.0/arekit/contrib/utils/processing/text/__init__.py +0 -0
  156. arekit-0.24.0/arekit/contrib/utils/resources.py +0 -25
  157. arekit-0.24.0/arekit/contrib/utils/sources/__init__.py +0 -0
  158. arekit-0.24.0/arekit/contrib/utils/sources/sentinerel/__init__.py +0 -0
  159. arekit-0.24.0/arekit/contrib/utils/sources/sentinerel/text_opinion/__init__.py +0 -0
  160. arekit-0.24.0/arekit/contrib/utils/sources/sentinerel/text_opinion/prof_per_org_filter.py +0 -63
  161. arekit-0.24.0/arekit/contrib/utils/synonyms/__init__.py +0 -0
  162. arekit-0.24.0/arekit/contrib/utils/vectorizers/__init__.py +0 -0
  163. arekit-0.24.0/arekit/download_data.py +0 -11
  164. arekit-0.24.0/arekit.egg-info/PKG-INFO +0 -19
  165. {arekit-0.24.0 → arekit-0.25.0}/LICENSE +0 -0
  166. {arekit-0.24.0 → arekit-0.25.0}/arekit/__init__.py +0 -0
  167. {arekit-0.24.0 → arekit-0.25.0}/arekit/common/__init__.py +0 -0
  168. {arekit-0.24.0 → arekit-0.25.0}/arekit/common/bound.py +0 -0
  169. {arekit-0.24.0 → arekit-0.25.0}/arekit/common/context/__init__.py +0 -0
  170. {arekit-0.24.0 → arekit-0.25.0}/arekit/common/context/terms_mapper.py +0 -0
  171. {arekit-0.24.0 → arekit-0.25.0}/arekit/common/context/token.py +0 -0
  172. {arekit-0.24.0 → arekit-0.25.0}/arekit/common/data/__init__.py +0 -0
  173. {arekit-0.24.0 → arekit-0.25.0}/arekit/common/data/const.py +0 -0
  174. {arekit-0.24.0 → arekit-0.25.0}/arekit/common/data/doc_provider.py +0 -0
  175. {arekit-0.24.0 → arekit-0.25.0}/arekit/common/data/input/__init__.py +0 -0
  176. {arekit-0.24.0 → arekit-0.25.0}/arekit/common/data/input/providers/__init__.py +0 -0
  177. {arekit-0.24.0 → arekit-0.25.0}/arekit/common/data/input/providers/columns/__init__.py +0 -0
  178. {arekit-0.24.0 → arekit-0.25.0}/arekit/common/data/input/providers/columns/base.py +0 -0
  179. {arekit-0.24.0 → arekit-0.25.0}/arekit/common/data/input/providers/columns/sample.py +0 -0
  180. {arekit-0.24.0 → arekit-0.25.0}/arekit/common/data/input/providers/const.py +0 -0
  181. {arekit-0.24.0 → arekit-0.25.0}/arekit/common/data/input/providers/contents.py +0 -0
  182. {arekit-0.24.0 → arekit-0.25.0}/arekit/common/data/input/providers/instances/__init__.py +0 -0
  183. {arekit-0.24.0 → arekit-0.25.0}/arekit/common/data/input/providers/instances/base.py +0 -0
  184. {arekit-0.24.0 → arekit-0.25.0}/arekit/common/data/input/providers/instances/multiple.py +0 -0
  185. {arekit-0.24.0 → arekit-0.25.0}/arekit/common/data/input/providers/instances/single.py +0 -0
  186. {arekit-0.24.0 → arekit-0.25.0}/arekit/common/data/input/providers/label/__init__.py +0 -0
  187. {arekit-0.24.0 → arekit-0.25.0}/arekit/common/data/input/providers/label/base.py +0 -0
  188. {arekit-0.24.0 → arekit-0.25.0}/arekit/common/data/input/providers/label/binary.py +0 -0
  189. {arekit-0.24.0 → arekit-0.25.0}/arekit/common/data/input/providers/label/multiple.py +0 -0
  190. {arekit-0.24.0 → arekit-0.25.0}/arekit/common/data/input/providers/rows/__init__.py +0 -0
  191. {arekit-0.24.0 → arekit-0.25.0}/arekit/common/data/input/providers/rows/base.py +0 -0
  192. {arekit-0.24.0 → arekit-0.25.0}/arekit/common/data/input/providers/rows/samples.py +0 -0
  193. {arekit-0.24.0 → arekit-0.25.0}/arekit/common/data/input/providers/sample/__init__.py +0 -0
  194. {arekit-0.24.0 → arekit-0.25.0}/arekit/common/data/input/providers/sample/cropped.py +0 -0
  195. {arekit-0.24.0 → arekit-0.25.0}/arekit/common/data/input/providers/text/__init__.py +0 -0
  196. {arekit-0.24.0 → arekit-0.25.0}/arekit/common/data/input/providers/text/single.py +0 -0
  197. {arekit-0.24.0 → arekit-0.25.0}/arekit/common/data/input/repositories/__init__.py +0 -0
  198. {arekit-0.24.0 → arekit-0.25.0}/arekit/common/data/input/repositories/base.py +0 -0
  199. {arekit-0.24.0 → arekit-0.25.0}/arekit/common/data/input/repositories/sample.py +0 -0
  200. {arekit-0.24.0 → arekit-0.25.0}/arekit/common/data/input/sample.py +0 -0
  201. {arekit-0.24.0 → arekit-0.25.0}/arekit/common/data/input/terms_mapper.py +0 -0
  202. {arekit-0.24.0 → arekit-0.25.0}/arekit/common/data/rows_fmt.py +0 -0
  203. {arekit-0.24.0 → arekit-0.25.0}/arekit/common/data/rows_parser.py +0 -0
  204. {arekit-0.24.0 → arekit-0.25.0}/arekit/common/data/storages/__init__.py +0 -0
  205. {arekit-0.24.0 → arekit-0.25.0}/arekit/common/data/storages/base.py +0 -0
  206. {arekit-0.24.0 → arekit-0.25.0}/arekit/common/data/views/__init__.py +0 -0
  207. {arekit-0.24.0 → arekit-0.25.0}/arekit/common/data/views/samples.py +0 -0
  208. {arekit-0.24.0 → arekit-0.25.0}/arekit/common/docs/__init__.py +0 -0
  209. {arekit-0.24.0 → arekit-0.25.0}/arekit/common/docs/base.py +0 -0
  210. {arekit-0.24.0 → arekit-0.25.0}/arekit/common/docs/entity.py +0 -0
  211. {arekit-0.24.0 → arekit-0.25.0}/arekit/common/docs/parsed/__init__.py +0 -0
  212. {arekit-0.24.0 → arekit-0.25.0}/arekit/common/docs/parsed/base.py +0 -0
  213. {arekit-0.24.0 → arekit-0.25.0}/arekit/common/docs/parsed/providers/__init__.py +0 -0
  214. {arekit-0.24.0 → arekit-0.25.0}/arekit/common/docs/parsed/providers/base.py +0 -0
  215. {arekit-0.24.0 → arekit-0.25.0}/arekit/common/docs/parsed/providers/base_pairs.py +0 -0
  216. {arekit-0.24.0 → arekit-0.25.0}/arekit/common/docs/parsed/providers/entity_service.py +0 -0
  217. {arekit-0.24.0 → arekit-0.25.0}/arekit/common/docs/parsed/providers/opinion_pairs.py +0 -0
  218. {arekit-0.24.0 → arekit-0.25.0}/arekit/common/docs/parsed/providers/text_opinion_pairs.py +0 -0
  219. {arekit-0.24.0 → arekit-0.25.0}/arekit/common/docs/parsed/service.py +0 -0
  220. {arekit-0.24.0 → arekit-0.25.0}/arekit/common/docs/parsed/term_position.py +0 -0
  221. {arekit-0.24.0 → arekit-0.25.0}/arekit/common/docs/sentence.py +0 -0
  222. {arekit-0.24.0 → arekit-0.25.0}/arekit/common/entities/__init__.py +0 -0
  223. {arekit-0.24.0 → arekit-0.25.0}/arekit/common/entities/base.py +0 -0
  224. {arekit-0.24.0 → arekit-0.25.0}/arekit/common/entities/collection.py +0 -0
  225. {arekit-0.24.0 → arekit-0.25.0}/arekit/common/entities/str_fmt.py +0 -0
  226. {arekit-0.24.0 → arekit-0.25.0}/arekit/common/entities/types.py +0 -0
  227. {arekit-0.24.0 → arekit-0.25.0}/arekit/common/experiment/__init__.py +0 -0
  228. {arekit-0.24.0 → arekit-0.25.0}/arekit/common/experiment/api/__init__.py +0 -0
  229. {arekit-0.24.0 → arekit-0.25.0}/arekit/common/experiment/api/base_samples_io.py +0 -0
  230. {arekit-0.24.0 → arekit-0.25.0}/arekit/common/experiment/data_type.py +0 -0
  231. {arekit-0.24.0 → arekit-0.25.0}/arekit/common/frames/__init__.py +0 -0
  232. {arekit-0.24.0 → arekit-0.25.0}/arekit/common/frames/connotations/__init__.py +0 -0
  233. {arekit-0.24.0 → arekit-0.25.0}/arekit/common/frames/connotations/descriptor.py +0 -0
  234. {arekit-0.24.0 → arekit-0.25.0}/arekit/common/frames/connotations/provider.py +0 -0
  235. {arekit-0.24.0 → arekit-0.25.0}/arekit/common/frames/text_variant.py +0 -0
  236. {arekit-0.24.0 → arekit-0.25.0}/arekit/common/frames/variants/__init__.py +0 -0
  237. {arekit-0.24.0 → arekit-0.25.0}/arekit/common/frames/variants/base.py +0 -0
  238. {arekit-0.24.0 → arekit-0.25.0}/arekit/common/frames/variants/collection.py +0 -0
  239. {arekit-0.24.0 → arekit-0.25.0}/arekit/common/labels/__init__.py +0 -0
  240. {arekit-0.24.0 → arekit-0.25.0}/arekit/common/labels/base.py +0 -0
  241. {arekit-0.24.0 → arekit-0.25.0}/arekit/common/labels/provider/__init__.py +0 -0
  242. {arekit-0.24.0 → arekit-0.25.0}/arekit/common/labels/provider/base.py +0 -0
  243. {arekit-0.24.0 → arekit-0.25.0}/arekit/common/labels/provider/constant.py +0 -0
  244. {arekit-0.24.0 → arekit-0.25.0}/arekit/common/labels/scaler/__init__.py +0 -0
  245. {arekit-0.24.0 → arekit-0.25.0}/arekit/common/labels/scaler/base.py +0 -0
  246. {arekit-0.24.0 → arekit-0.25.0}/arekit/common/labels/scaler/sentiment.py +0 -0
  247. {arekit-0.24.0 → arekit-0.25.0}/arekit/common/labels/scaler/single.py +0 -0
  248. {arekit-0.24.0 → arekit-0.25.0}/arekit/common/labels/str_fmt.py +0 -0
  249. {arekit-0.24.0 → arekit-0.25.0}/arekit/common/linkage/__init__.py +0 -0
  250. {arekit-0.24.0 → arekit-0.25.0}/arekit/common/linkage/base.py +0 -0
  251. {arekit-0.24.0 → arekit-0.25.0}/arekit/common/linkage/meta.py +0 -0
  252. {arekit-0.24.0 → arekit-0.25.0}/arekit/common/linkage/opinions.py +0 -0
  253. {arekit-0.24.0 → arekit-0.25.0}/arekit/common/linkage/text_opinions.py +0 -0
  254. {arekit-0.24.0 → arekit-0.25.0}/arekit/common/log_utils.py +0 -0
  255. {arekit-0.24.0 → arekit-0.25.0}/arekit/common/model/__init__.py +0 -0
  256. {arekit-0.24.0 → arekit-0.25.0}/arekit/common/model/labeling/__init__.py +0 -0
  257. {arekit-0.24.0 → arekit-0.25.0}/arekit/common/model/labeling/base.py +0 -0
  258. {arekit-0.24.0 → arekit-0.25.0}/arekit/common/model/labeling/modes.py +0 -0
  259. {arekit-0.24.0 → arekit-0.25.0}/arekit/common/model/labeling/single.py +0 -0
  260. {arekit-0.24.0 → arekit-0.25.0}/arekit/common/opinions/__init__.py +0 -0
  261. {arekit-0.24.0 → arekit-0.25.0}/arekit/common/opinions/annot/__init__.py +0 -0
  262. {arekit-0.24.0 → arekit-0.25.0}/arekit/common/opinions/annot/algo/__init__.py +0 -0
  263. {arekit-0.24.0 → arekit-0.25.0}/arekit/common/opinions/annot/algo/base.py +0 -0
  264. {arekit-0.24.0 → arekit-0.25.0}/arekit/common/opinions/annot/algo/pair_based.py +0 -0
  265. {arekit-0.24.0 → arekit-0.25.0}/arekit/common/opinions/annot/algo/predefined.py +0 -0
  266. {arekit-0.24.0 → arekit-0.25.0}/arekit/common/opinions/annot/algo_based.py +0 -0
  267. {arekit-0.24.0 → arekit-0.25.0}/arekit/common/opinions/annot/base.py +0 -0
  268. {arekit-0.24.0 → arekit-0.25.0}/arekit/common/opinions/base.py +0 -0
  269. {arekit-0.24.0 → arekit-0.25.0}/arekit/common/opinions/collection.py +0 -0
  270. {arekit-0.24.0 → arekit-0.25.0}/arekit/common/opinions/enums.py +0 -0
  271. {arekit-0.24.0 → arekit-0.25.0}/arekit/common/opinions/provider.py +0 -0
  272. {arekit-0.24.0 → arekit-0.25.0}/arekit/common/opinions/writer.py +0 -0
  273. {arekit-0.24.0 → arekit-0.25.0}/arekit/common/pipeline/__init__.py +0 -0
  274. {arekit-0.24.0 → arekit-0.25.0}/arekit/common/pipeline/conts.py +0 -0
  275. {arekit-0.24.0 → arekit-0.25.0}/arekit/common/pipeline/items/__init__.py +0 -0
  276. {arekit-0.24.0/arekit/common/synonyms → arekit-0.25.0/arekit/common/service}/__init__.py +0 -0
  277. {arekit-0.24.0/arekit/common/text → arekit-0.25.0/arekit/common/synonyms}/__init__.py +0 -0
  278. {arekit-0.24.0 → arekit-0.25.0}/arekit/common/synonyms/base.py +0 -0
  279. {arekit-0.24.0 → arekit-0.25.0}/arekit/common/synonyms/grouping.py +0 -0
  280. {arekit-0.24.0/arekit/common/text/partitioning → arekit-0.25.0/arekit/common/text}/__init__.py +0 -0
  281. {arekit-0.24.0 → arekit-0.25.0}/arekit/common/text/enums.py +0 -0
  282. {arekit-0.24.0 → arekit-0.25.0}/arekit/common/text/parsed.py +0 -0
  283. {arekit-0.24.0 → arekit-0.25.0}/arekit/common/text/stemmer.py +0 -0
  284. {arekit-0.24.0 → arekit-0.25.0}/arekit/common/text_opinions/__init__.py +0 -0
  285. {arekit-0.24.0 → arekit-0.25.0}/arekit/common/text_opinions/base.py +0 -0
  286. {arekit-0.24.0 → arekit-0.25.0}/arekit/contrib/__init__.py +0 -0
  287. {arekit-0.24.0 → arekit-0.25.0}/arekit/contrib/bert/__init__.py +0 -0
  288. {arekit-0.24.0 → arekit-0.25.0}/arekit/contrib/bert/input/__init__.py +0 -0
  289. {arekit-0.24.0 → arekit-0.25.0}/arekit/contrib/bert/input/providers/__init__.py +0 -0
  290. {arekit-0.24.0 → arekit-0.25.0}/arekit/contrib/bert/input/providers/cropped_sample.py +0 -0
  291. {arekit-0.24.0 → arekit-0.25.0}/arekit/contrib/bert/input/providers/text_pair.py +0 -0
  292. {arekit-0.24.0 → arekit-0.25.0}/arekit/contrib/bert/terms/__init__.py +0 -0
  293. {arekit-0.24.0 → arekit-0.25.0}/arekit/contrib/bert/terms/mapper.py +0 -0
  294. {arekit-0.24.0 → arekit-0.25.0}/arekit/contrib/networks/__init__.py +0 -0
  295. {arekit-0.24.0 → arekit-0.25.0}/arekit/contrib/networks/embedding.py +0 -0
  296. {arekit-0.24.0 → arekit-0.25.0}/arekit/contrib/networks/embedding_io.py +0 -0
  297. {arekit-0.24.0 → arekit-0.25.0}/arekit/contrib/networks/input/__init__.py +0 -0
  298. {arekit-0.24.0 → arekit-0.25.0}/arekit/contrib/networks/input/const.py +0 -0
  299. {arekit-0.24.0 → arekit-0.25.0}/arekit/contrib/networks/input/ctx_serialization.py +0 -0
  300. {arekit-0.24.0 → arekit-0.25.0}/arekit/contrib/networks/input/embedding/__init__.py +0 -0
  301. {arekit-0.24.0 → arekit-0.25.0}/arekit/contrib/networks/input/embedding/matrix.py +0 -0
  302. {arekit-0.24.0 → arekit-0.25.0}/arekit/contrib/networks/input/embedding/offsets.py +0 -0
  303. {arekit-0.24.0 → arekit-0.25.0}/arekit/contrib/networks/input/formatters/__init__.py +0 -0
  304. {arekit-0.24.0 → arekit-0.25.0}/arekit/contrib/networks/input/formatters/pos_mapper.py +0 -0
  305. {arekit-0.24.0 → arekit-0.25.0}/arekit/contrib/networks/input/providers/__init__.py +0 -0
  306. {arekit-0.24.0 → arekit-0.25.0}/arekit/contrib/networks/input/providers/sample.py +0 -0
  307. {arekit-0.24.0 → arekit-0.25.0}/arekit/contrib/networks/input/providers/term_connotation.py +0 -0
  308. {arekit-0.24.0 → arekit-0.25.0}/arekit/contrib/networks/input/providers/text.py +0 -0
  309. {arekit-0.24.0 → arekit-0.25.0}/arekit/contrib/networks/input/rows_parser.py +0 -0
  310. {arekit-0.24.0 → arekit-0.25.0}/arekit/contrib/networks/input/term_types.py +0 -0
  311. {arekit-0.24.0 → arekit-0.25.0}/arekit/contrib/networks/input/terms_mapping.py +0 -0
  312. {arekit-0.24.0 → arekit-0.25.0}/arekit/contrib/networks/vectorizer.py +0 -0
  313. {arekit-0.24.0 → arekit-0.25.0}/arekit/contrib/prompt/__init__.py +0 -0
  314. {arekit-0.24.0 → arekit-0.25.0}/arekit/contrib/prompt/sample.py +0 -0
  315. {arekit-0.24.0/arekit/contrib/source → arekit-0.25.0/arekit/contrib/utils}/__init__.py +0 -0
  316. {arekit-0.24.0/arekit/contrib/source/brat → arekit-0.25.0/arekit/contrib/utils/bert}/__init__.py +0 -0
  317. {arekit-0.24.0 → arekit-0.25.0}/arekit/contrib/utils/bert/samplers.py +0 -0
  318. {arekit-0.24.0/arekit/contrib/source/brat/entities → arekit-0.25.0/arekit/contrib/utils/data}/__init__.py +0 -0
  319. {arekit-0.24.0/arekit/contrib/source/brat/opinions → arekit-0.25.0/arekit/contrib/utils/data/contents}/__init__.py +0 -0
  320. {arekit-0.24.0/arekit/contrib/source/nerel → arekit-0.25.0/arekit/contrib/utils/data/doc_provider}/__init__.py +0 -0
  321. {arekit-0.24.0 → arekit-0.25.0}/arekit/contrib/utils/data/doc_provider/dict_based.py +0 -0
  322. {arekit-0.24.0 → arekit-0.25.0}/arekit/contrib/utils/data/doc_provider/dir_based.py +0 -0
  323. {arekit-0.24.0/arekit/contrib/source/nerel/folding → arekit-0.25.0/arekit/contrib/utils/data/readers}/__init__.py +0 -0
  324. {arekit-0.24.0 → arekit-0.25.0}/arekit/contrib/utils/data/readers/base.py +0 -0
  325. {arekit-0.24.0 → arekit-0.25.0}/arekit/contrib/utils/data/readers/csv_pd.py +0 -0
  326. {arekit-0.24.0 → arekit-0.25.0}/arekit/contrib/utils/data/readers/jsonl.py +0 -0
  327. {arekit-0.24.0/arekit/contrib/source/nerelbio → arekit-0.25.0/arekit/contrib/utils/data/service}/__init__.py +0 -0
  328. {arekit-0.24.0 → arekit-0.25.0}/arekit/contrib/utils/data/service/balance.py +0 -0
  329. {arekit-0.24.0/arekit/contrib/source/ruattitudes → arekit-0.25.0/arekit/contrib/utils/data/storages}/__init__.py +0 -0
  330. {arekit-0.24.0 → arekit-0.25.0}/arekit/contrib/utils/data/storages/jsonl_based.py +0 -0
  331. {arekit-0.24.0 → arekit-0.25.0}/arekit/contrib/utils/data/storages/pandas_based.py +0 -0
  332. {arekit-0.24.0/arekit/contrib/source/ruattitudes/entity → arekit-0.25.0/arekit/contrib/utils/data/writers}/__init__.py +0 -0
  333. {arekit-0.24.0 → arekit-0.25.0}/arekit/contrib/utils/data/writers/base.py +0 -0
  334. {arekit-0.24.0 → arekit-0.25.0}/arekit/contrib/utils/data/writers/csv_native.py +0 -0
  335. {arekit-0.24.0 → arekit-0.25.0}/arekit/contrib/utils/data/writers/csv_pd.py +0 -0
  336. {arekit-0.24.0 → arekit-0.25.0}/arekit/contrib/utils/data/writers/json_opennre.py +0 -0
  337. {arekit-0.24.0/arekit/contrib/source/ruattitudes/opinions → arekit-0.25.0/arekit/contrib/utils/embeddings}/__init__.py +0 -0
  338. {arekit-0.24.0 → arekit-0.25.0}/arekit/contrib/utils/embeddings/rusvectores.py +0 -0
  339. {arekit-0.24.0 → arekit-0.25.0}/arekit/contrib/utils/embeddings/tokens.py +0 -0
  340. {arekit-0.24.0/arekit/contrib/source/rusentiframes → arekit-0.25.0/arekit/contrib/utils/entities}/__init__.py +0 -0
  341. {arekit-0.24.0 → arekit-0.25.0}/arekit/contrib/utils/entities/filter.py +0 -0
  342. {arekit-0.24.0/arekit/contrib/source/rusentrel → arekit-0.25.0/arekit/contrib/utils/entities/formatters}/__init__.py +0 -0
  343. {arekit-0.24.0 → arekit-0.25.0}/arekit/contrib/utils/entities/formatters/str_display.py +0 -0
  344. {arekit-0.24.0 → arekit-0.25.0}/arekit/contrib/utils/entities/formatters/str_simple_sharp_prefixed_fmt.py +0 -0
  345. {arekit-0.24.0/arekit/contrib/source/rusentrel/opinions → arekit-0.25.0/arekit/contrib/utils/io_utils}/__init__.py +0 -0
  346. {arekit-0.24.0 → arekit-0.25.0}/arekit/contrib/utils/io_utils/embedding.py +0 -0
  347. {arekit-0.24.0/arekit/contrib/source/sentinerel → arekit-0.25.0/arekit/contrib/utils/np_utils}/__init__.py +0 -0
  348. {arekit-0.24.0 → arekit-0.25.0}/arekit/contrib/utils/np_utils/embedding.py +0 -0
  349. {arekit-0.24.0 → arekit-0.25.0}/arekit/contrib/utils/np_utils/npz_utils.py +0 -0
  350. {arekit-0.24.0 → arekit-0.25.0}/arekit/contrib/utils/np_utils/vocab.py +0 -0
  351. {arekit-0.24.0/arekit/contrib/source/sentinerel/folding → arekit-0.25.0/arekit/contrib/utils/pipelines}/__init__.py +0 -0
  352. {arekit-0.24.0/arekit/contrib/source/synonyms → arekit-0.25.0/arekit/contrib/utils/pipelines/items}/__init__.py +0 -0
  353. {arekit-0.24.0/arekit/contrib/utils → arekit-0.25.0/arekit/contrib/utils/pipelines/items/sampling}/__init__.py +0 -0
  354. {arekit-0.24.0/arekit/contrib/utils/bert → arekit-0.25.0/arekit/contrib/utils/pipelines/items/text}/__init__.py +0 -0
  355. {arekit-0.24.0 → arekit-0.25.0}/arekit/contrib/utils/pipelines/opinion_collections.py +0 -0
  356. {arekit-0.24.0/arekit/contrib/utils/connotations → arekit-0.25.0/arekit/contrib/utils/pipelines/text_opinion}/__init__.py +0 -0
  357. {arekit-0.24.0/arekit/contrib/utils/data → arekit-0.25.0/arekit/contrib/utils/pipelines/text_opinion/annot}/__init__.py +0 -0
  358. {arekit-0.24.0 → arekit-0.25.0}/arekit/contrib/utils/pipelines/text_opinion/annot/algo_based.py +0 -0
  359. {arekit-0.24.0/arekit/contrib/utils/data/contents → arekit-0.25.0/arekit/contrib/utils/pipelines/text_opinion/filters}/__init__.py +0 -0
  360. {arekit-0.24.0 → arekit-0.25.0}/arekit/contrib/utils/pipelines/text_opinion/filters/base.py +0 -0
  361. {arekit-0.24.0 → arekit-0.25.0}/arekit/contrib/utils/pipelines/text_opinion/filters/distance_based.py +0 -0
  362. {arekit-0.24.0 → arekit-0.25.0}/arekit/contrib/utils/pipelines/text_opinion/filters/entity_based.py +0 -0
  363. {arekit-0.24.0 → arekit-0.25.0}/arekit/contrib/utils/pipelines/text_opinion/filters/limitation.py +0 -0
  364. {arekit-0.24.0/arekit/contrib/utils/data/doc_provider → arekit-0.25.0/arekit/contrib/utils/processing}/__init__.py +0 -0
  365. {arekit-0.24.0/arekit/contrib/utils/data/readers → arekit-0.25.0/arekit/contrib/utils/processing/languages}/__init__.py +0 -0
  366. {arekit-0.24.0 → arekit-0.25.0}/arekit/contrib/utils/processing/languages/mods.py +0 -0
  367. {arekit-0.24.0 → arekit-0.25.0}/arekit/contrib/utils/processing/languages/pos.py +0 -0
  368. {arekit-0.24.0/arekit/contrib/utils/data/service → arekit-0.25.0/arekit/contrib/utils/processing/languages/ru}/__init__.py +0 -0
  369. {arekit-0.24.0 → arekit-0.25.0}/arekit/contrib/utils/processing/languages/ru/cases.py +0 -0
  370. {arekit-0.24.0 → arekit-0.25.0}/arekit/contrib/utils/processing/languages/ru/constants.py +0 -0
  371. {arekit-0.24.0 → arekit-0.25.0}/arekit/contrib/utils/processing/languages/ru/mods.py +0 -0
  372. {arekit-0.24.0 → arekit-0.25.0}/arekit/contrib/utils/processing/languages/ru/number.py +0 -0
  373. {arekit-0.24.0 → arekit-0.25.0}/arekit/contrib/utils/processing/languages/ru/pos_service.py +0 -0
  374. {arekit-0.24.0/arekit/contrib/utils/data/storages → arekit-0.25.0/arekit/contrib/utils/processing/lemmatization}/__init__.py +0 -0
  375. {arekit-0.24.0 → arekit-0.25.0}/arekit/contrib/utils/processing/lemmatization/mystem.py +0 -0
  376. {arekit-0.24.0/arekit/contrib/utils/data/writers → arekit-0.25.0/arekit/contrib/utils/processing/pos}/__init__.py +0 -0
  377. {arekit-0.24.0 → arekit-0.25.0}/arekit/contrib/utils/processing/pos/base.py +0 -0
  378. {arekit-0.24.0 → arekit-0.25.0}/arekit/contrib/utils/processing/pos/mystem_wrap.py +0 -0
  379. {arekit-0.24.0 → arekit-0.25.0}/arekit/contrib/utils/processing/pos/russian.py +0 -0
  380. {arekit-0.24.0/arekit/contrib/utils/embeddings → arekit-0.25.0/arekit/contrib/utils/processing/text}/__init__.py +0 -0
  381. {arekit-0.24.0 → arekit-0.25.0}/arekit/contrib/utils/processing/text/tokens.py +0 -0
  382. {arekit-0.24.0/arekit/contrib/utils/entities → arekit-0.25.0/arekit/contrib/utils/synonyms}/__init__.py +0 -0
  383. {arekit-0.24.0 → arekit-0.25.0}/arekit/contrib/utils/synonyms/simple.py +0 -0
  384. {arekit-0.24.0 → arekit-0.25.0}/arekit/contrib/utils/synonyms/stemmer_based.py +0 -0
  385. {arekit-0.24.0/arekit/contrib/utils/entities/formatters → arekit-0.25.0/arekit/contrib/utils/vectorizers}/__init__.py +0 -0
  386. {arekit-0.24.0 → arekit-0.25.0}/arekit/contrib/utils/vectorizers/bpe.py +0 -0
  387. {arekit-0.24.0 → arekit-0.25.0}/arekit/contrib/utils/vectorizers/random_norm.py +0 -0
  388. {arekit-0.24.0 → arekit-0.25.0}/arekit.egg-info/dependency_links.txt +0 -0
  389. {arekit-0.24.0 → arekit-0.25.0}/arekit.egg-info/requires.txt +0 -0
  390. {arekit-0.24.0 → arekit-0.25.0}/arekit.egg-info/top_level.txt +0 -0
  391. {arekit-0.24.0 → arekit-0.25.0}/setup.cfg +0 -0
arekit-0.25.0/PKG-INFO ADDED
@@ -0,0 +1,82 @@
1
+ Metadata-Version: 2.1
2
+ Name: arekit
3
+ Version: 0.25.0
4
+ Summary: Document level Attitude and Relation Extraction toolkit (AREkit) for sampling and prompting mass-media news into datasets for ML-model training
5
+ Home-page: https://github.com/nicolay-r/AREkit
6
+ Author: Nicolay Rusnachenko
7
+ Author-email: rusnicolay@gmail.com
8
+ License: MIT License
9
+ Keywords: natural language processing,relation extraction,sentiment analysis
10
+ Classifier: Programming Language :: Python
11
+ Classifier: Programming Language :: Python :: 3.6
12
+ Classifier: Topic :: Software Development :: Libraries :: Python Modules
13
+ Classifier: Topic :: Scientific/Engineering :: Information Analysis
14
+ Classifier: Topic :: Text Processing :: Linguistic
15
+ Requires-Python: >=3.6
16
+ Description-Content-Type: text/markdown
17
+ License-File: LICENSE
18
+ Requires-Dist: tqdm
19
+ Requires-Dist: enum34==1.1.10
20
+ Requires-Dist: numpy>=1.14.5
21
+ Requires-Dist: pymystem3==0.2.0
22
+
23
+ # AREkit 0.25.0
24
+
25
+ ![](https://img.shields.io/badge/Python-3.9+-brightgreen.svg)
26
+
27
+ <p align="center">
28
+ <img src="logo.png"/>
29
+ </p>
30
+
31
+ **AREkit** (Attitude and Relation Extraction Toolkit) --
32
+ is a python toolkit, devoted to document level Attitude and Relation Extraction between text objects from mass-media news.
33
+
34
+ ## Description
35
+
36
+
37
+ This toolkit aims at memory-effective data processing in Relation Extraction (RE) related tasks.
38
+
39
+ <p align="center">
40
+ <img src="docs/arekit-pipeline-concept.png"/>
41
+ </p>
42
+
43
+ > Figure: AREkit pipelines design. More on
44
+ > **[ARElight: Context Sampling of Large Texts for Deep Learning Relation Extraction](https://link.springer.com/chapter/10.1007/978-3-031-56069-9_23)** paper
45
+
46
+ In particular, this framework serves the following features:
47
+ * ➿ [pipelines](https://github.com/nicolay-r/AREkit/wiki/Pipelines:-Text-Opinion-Annotation) and iterators for handling large-scale collections serialization without out-of-memory issues.
48
+ * 🔗 EL (entity-linking) API support for objects,
49
+ * ➰ avoidance of cyclic connections,
50
+ * :straight_ruler: distance consideration between relation participants (in `terms` or `sentences`),
51
+ * 📑 relations annotations and filtering rules,
52
+ * *️⃣ entities formatting or masking, and more.
53
+
54
+ The core functionality includes:
55
+ * API for document presentation with EL (Entity Linking, i.e. Object Synonymy) support
56
+ for sentence level relations preparation (dubbed as contexts);
57
+ * API for contexts extraction;
58
+ * Relations transferring from sentence-level onto document-level, and more.
59
+
60
+ ## Installation
61
+
62
+ ```bash
63
+ pip install git+https://github.com/nicolay-r/AREkit.git@0.25.0-rc
64
+ ```
65
+
66
+ ## Usage
67
+
68
+ Please follow the **[tutorial section on project Wiki](https://github.com/nicolay-r/AREkit/wiki/Tutorials)** for mode details.
69
+
70
+ ## How to cite
71
+ A great research is also accompanied by the faithful reference.
72
+ if you use or extend our work, please cite as follows:
73
+
74
+ ```bibtex
75
+ @inproceedings{rusnachenko2024arelight,
76
+ title={ARElight: Context Sampling of Large Texts for Deep Learning Relation Extraction},
77
+ author={Rusnachenko, Nicolay and Liang, Huizhi and Kolomeets, Maxim and Shi, Lei},
78
+ booktitle={European Conference on Information Retrieval},
79
+ year={2024},
80
+ organization={Springer}
81
+ }
82
+ ```
@@ -0,0 +1,60 @@
1
+ # AREkit 0.25.0
2
+
3
+ ![](https://img.shields.io/badge/Python-3.9+-brightgreen.svg)
4
+
5
+ <p align="center">
6
+ <img src="logo.png"/>
7
+ </p>
8
+
9
+ **AREkit** (Attitude and Relation Extraction Toolkit) --
10
+ is a python toolkit, devoted to document level Attitude and Relation Extraction between text objects from mass-media news.
11
+
12
+ ## Description
13
+
14
+
15
+ This toolkit aims at memory-effective data processing in Relation Extraction (RE) related tasks.
16
+
17
+ <p align="center">
18
+ <img src="docs/arekit-pipeline-concept.png"/>
19
+ </p>
20
+
21
+ > Figure: AREkit pipelines design. More on
22
+ > **[ARElight: Context Sampling of Large Texts for Deep Learning Relation Extraction](https://link.springer.com/chapter/10.1007/978-3-031-56069-9_23)** paper
23
+
24
+ In particular, this framework serves the following features:
25
+ * ➿ [pipelines](https://github.com/nicolay-r/AREkit/wiki/Pipelines:-Text-Opinion-Annotation) and iterators for handling large-scale collections serialization without out-of-memory issues.
26
+ * 🔗 EL (entity-linking) API support for objects,
27
+ * ➰ avoidance of cyclic connections,
28
+ * :straight_ruler: distance consideration between relation participants (in `terms` or `sentences`),
29
+ * 📑 relations annotations and filtering rules,
30
+ * *️⃣ entities formatting or masking, and more.
31
+
32
+ The core functionality includes:
33
+ * API for document presentation with EL (Entity Linking, i.e. Object Synonymy) support
34
+ for sentence level relations preparation (dubbed as contexts);
35
+ * API for contexts extraction;
36
+ * Relations transferring from sentence-level onto document-level, and more.
37
+
38
+ ## Installation
39
+
40
+ ```bash
41
+ pip install git+https://github.com/nicolay-r/AREkit.git@0.25.0-rc
42
+ ```
43
+
44
+ ## Usage
45
+
46
+ Please follow the **[tutorial section on project Wiki](https://github.com/nicolay-r/AREkit/wiki/Tutorials)** for mode details.
47
+
48
+ ## How to cite
49
+ A great research is also accompanied by the faithful reference.
50
+ if you use or extend our work, please cite as follows:
51
+
52
+ ```bibtex
53
+ @inproceedings{rusnachenko2024arelight,
54
+ title={ARElight: Context Sampling of Large Texts for Deep Learning Relation Extraction},
55
+ author={Rusnachenko, Nicolay and Liang, Huizhi and Kolomeets, Maxim and Shi, Lei},
56
+ booktitle={European Conference on Information Retrieval},
57
+ year={2024},
58
+ organization={Springer}
59
+ }
60
+ ```
@@ -4,8 +4,9 @@ from arekit.common.pipeline.items.base import BasePipelineItem
4
4
 
5
5
  class EntitiesGroupingPipelineItem(BasePipelineItem):
6
6
 
7
- def __init__(self, value_to_group_id_func):
7
+ def __init__(self, value_to_group_id_func, **kwargs):
8
8
  assert(callable(value_to_group_id_func))
9
+ super(EntitiesGroupingPipelineItem, self).__init__(**kwargs)
9
10
  self.__value_to_group_id_func = value_to_group_id_func
10
11
 
11
12
  def apply_core(self, input_data, pipeline_ctx):
@@ -0,0 +1,66 @@
1
+ from tqdm import tqdm
2
+ from arekit.common.docs.base import Document
3
+ from arekit.common.docs.parsed.base import ParsedDocument
4
+ from arekit.common.pipeline.base import BasePipelineLauncher
5
+ from arekit.common.pipeline.batching import BatchingPipelineLauncher
6
+ from arekit.common.pipeline.context import PipelineContext
7
+ from arekit.common.pipeline.utils import BatchIterator
8
+ from arekit.common.text.parsed import BaseParsedText
9
+
10
+
11
+ class DocumentParsers(object):
12
+
13
+ @staticmethod
14
+ def parse(doc, pipeline_items, parent_ppl_ctx=None, src_key="input", show_progress=False):
15
+ """ This document parser is based on single text parts (sentences)
16
+ that passes sequentially through the pipeline of transformations.
17
+ """
18
+ assert(isinstance(doc, Document))
19
+ assert(isinstance(pipeline_items, list))
20
+ assert(isinstance(parent_ppl_ctx, PipelineContext) or parent_ppl_ctx is None)
21
+
22
+ parsed_sentences = []
23
+
24
+ data_it = range(doc.SentencesCount)
25
+ progress_it = tqdm(data_it, disable=not show_progress)
26
+
27
+ for sent_ind in progress_it:
28
+
29
+ # Composing the context from a single sentence.
30
+ ctx = PipelineContext({src_key: doc.get_sentence(sent_ind)}, parent_ctx=parent_ppl_ctx)
31
+
32
+ # Apply all the operations.
33
+ BasePipelineLauncher.run(pipeline=pipeline_items, pipeline_ctx=ctx, src_key=src_key)
34
+
35
+ # Collecting the result.
36
+ parsed_sentences.append(BaseParsedText(terms=ctx.provide("result")))
37
+
38
+ return ParsedDocument(doc_id=doc.ID, parsed_sentences=parsed_sentences)
39
+
40
+ @staticmethod
41
+ def parse_batch(doc, pipeline_items, batch_size, parent_ppl_ctx=None, src_key="input", show_progress=False):
42
+ """ This document parser is based on batch of sentences.
43
+ """
44
+ assert(isinstance(batch_size, int) and batch_size > 0)
45
+ assert(isinstance(doc, Document))
46
+ assert(isinstance(pipeline_items, list))
47
+ assert(isinstance(parent_ppl_ctx, PipelineContext) or parent_ppl_ctx is None)
48
+
49
+ parsed_sentences = []
50
+
51
+ data_it = BatchIterator(data_iter=iter(range(doc.SentencesCount)), batch_size=batch_size)
52
+ progress_it = tqdm(data_it, total=round(doc.SentencesCount / batch_size), disable=not show_progress)
53
+
54
+ for batch in progress_it:
55
+
56
+ # Composing the context from a single sentence.
57
+ ctx = PipelineContext({src_key: [doc.get_sentence(s_ind) for s_ind in batch]},
58
+ parent_ctx=parent_ppl_ctx)
59
+
60
+ # Apply all the operations.
61
+ BatchingPipelineLauncher.run(pipeline=pipeline_items, pipeline_ctx=ctx, src_key=src_key)
62
+
63
+ # Collecting the result.
64
+ parsed_sentences += [BaseParsedText(terms=result) for result in ctx.provide("result")]
65
+
66
+ return ParsedDocument(doc_id=doc.ID, parsed_sentences=parsed_sentences)
@@ -0,0 +1,21 @@
1
+ from arekit.common.pipeline.context import PipelineContext
2
+ from arekit.common.pipeline.items.base import BasePipelineItem
3
+
4
+
5
+ class BasePipelineLauncher:
6
+
7
+ @staticmethod
8
+ def run(pipeline, pipeline_ctx, src_key=None, has_input=True):
9
+ assert(isinstance(pipeline, list))
10
+ assert(isinstance(pipeline_ctx, PipelineContext))
11
+ assert(isinstance(src_key, str) or src_key is None)
12
+
13
+ for ind, item in enumerate(filter(lambda itm: itm is not None, pipeline)):
14
+ assert(isinstance(item, BasePipelineItem))
15
+ do_force_key = src_key is not None and ind == 0
16
+ input_data = item.get_source(pipeline_ctx, force_key=src_key if do_force_key else None) \
17
+ if has_input or ind > 0 else None
18
+ item_result = item.apply(input_data=input_data, pipeline_ctx=pipeline_ctx)
19
+ pipeline_ctx.update(param=item.ResultKey, value=item_result, is_new_key=False)
20
+
21
+ return pipeline_ctx
@@ -0,0 +1,28 @@
1
+ from arekit.common.pipeline.context import PipelineContext
2
+ from arekit.common.pipeline.items.base import BasePipelineItem
3
+
4
+
5
+ class BatchingPipelineLauncher:
6
+
7
+ @staticmethod
8
+ def run(pipeline, pipeline_ctx, src_key=None):
9
+ assert(isinstance(pipeline, list))
10
+ assert(isinstance(pipeline_ctx, PipelineContext))
11
+ assert(isinstance(src_key, str) or src_key is None)
12
+
13
+ for ind, item in enumerate(filter(lambda itm: itm is not None, pipeline)):
14
+ assert (isinstance(item, BasePipelineItem))
15
+
16
+ # Handle the content of the batch or batch itself.
17
+ content = item.get_source(pipeline_ctx, call_func=False, force_key=src_key if ind == 0 else None)
18
+ handled_batch = [item._src_func(i) if item._src_func is not None else i for i in content]
19
+
20
+ if item.SupportBatching:
21
+ batch_result = list(item.apply(input_data=handled_batch, pipeline_ctx=pipeline_ctx))
22
+ else:
23
+ batch_result = [item.apply(input_data=input_data, pipeline_ctx=pipeline_ctx)
24
+ for input_data in handled_batch]
25
+
26
+ pipeline_ctx.update(param=item.ResultKey, value=batch_result, is_new_key=False)
27
+
28
+ return pipeline_ctx
@@ -13,6 +13,8 @@ class PipelineContext(object):
13
13
  self._d[PARENT_CTX] = parent_ctx
14
14
 
15
15
  def __provide(self, param):
16
+ if param not in self._d:
17
+ raise Exception(f"Key `{param}` is not in dictionary.\n{self._d}")
16
18
  return self._d[param]
17
19
 
18
20
  # region public
@@ -23,7 +25,9 @@ class PipelineContext(object):
23
25
  def provide_or_none(self, param):
24
26
  return self.__provide(param) if param in self._d else None
25
27
 
26
- def update(self, param, value):
28
+ def update(self, param, value, is_new_key=False):
29
+ if is_new_key and param in self._d:
30
+ raise Exception(f"Key `{param}` is already presented in pipeline context dictionary.")
27
31
  self._d[param] = value
28
32
 
29
33
  # endregion
@@ -0,0 +1,49 @@
1
+ from arekit.common.pipeline.context import PipelineContext
2
+
3
+
4
+ class BasePipelineItem(object):
5
+ """ Single pipeline item that might be instatiated and embedded into pipeline.
6
+ """
7
+
8
+ def __init__(self, src_key="result", result_key="result", src_func=None):
9
+ assert(isinstance(src_key, str) or src_key is None)
10
+ assert(callable(src_func) or src_func is None)
11
+ self.__src_key = src_key
12
+ self._src_func = src_func
13
+ self.__result_key = result_key
14
+
15
+ @property
16
+ def ResultKey(self):
17
+ return self.__result_key
18
+
19
+ @property
20
+ def SupportBatching(self):
21
+ """ By default pipeline item is not designed for batching.
22
+ """
23
+ return False
24
+
25
+ def get_source(self, src_ctx, call_func=True, force_key=None):
26
+ """ Extract input element for processing.
27
+ """
28
+ assert(isinstance(src_ctx, PipelineContext))
29
+
30
+ # If there is no information about key, then we consider absence of the source.
31
+ if self.__src_key is None:
32
+ return None
33
+
34
+ # Extracting actual source.
35
+ src_data = src_ctx.provide(self.__src_key if force_key is None else force_key)
36
+ if self._src_func is not None and call_func:
37
+ src_data = self._src_func(src_data)
38
+
39
+ return src_data
40
+
41
+ def apply_core(self, input_data, pipeline_ctx):
42
+ """By default we do nothing."""
43
+ pass
44
+
45
+ def apply(self, input_data, pipeline_ctx=None):
46
+ """ Performs input processing an update it for a further pipeline items.
47
+ """
48
+ output_data = self.apply_core(input_data=input_data, pipeline_ctx=pipeline_ctx)
49
+ return output_data if output_data is not None else input_data
@@ -5,10 +5,14 @@ class FlattenIterPipelineItem(BasePipelineItem):
5
5
  """ Considered to flat iterations of items that represent iterations.
6
6
  """
7
7
 
8
+ def __init__(self, **kwargs):
9
+ super(FlattenIterPipelineItem, self).__init__(**kwargs)
10
+ pass
11
+
8
12
  def __flat_iter(self, iter_data):
9
13
  for iter_item in iter_data:
10
14
  for item in iter_item:
11
15
  yield item
12
16
 
13
17
  def apply_core(self, input_data, pipeline_ctx):
14
- return self.__flat_iter(input_data)
18
+ return self.__flat_iter(input_data)
@@ -3,8 +3,9 @@ from arekit.common.pipeline.items.base import BasePipelineItem
3
3
 
4
4
  class HandleIterPipelineItem(BasePipelineItem):
5
5
 
6
- def __init__(self, handle_func=None):
6
+ def __init__(self, handle_func=None, **kwargs):
7
7
  assert(callable(handle_func))
8
+ super(HandleIterPipelineItem, self).__init__(**kwargs)
8
9
  self.__handle_func = handle_func
9
10
 
10
11
  def __updated_data(self, items_iter):
@@ -3,8 +3,9 @@ from arekit.common.pipeline.items.base import BasePipelineItem
3
3
 
4
4
  class FilterPipelineItem(BasePipelineItem):
5
5
 
6
- def __init__(self, filter_func=None):
6
+ def __init__(self, filter_func=None, **kwargs):
7
7
  assert(callable(filter_func))
8
+ super(FilterPipelineItem, self).__init__(**kwargs)
8
9
  self.__filter_func = filter_func
9
10
 
10
11
  def apply_core(self, input_data, pipeline_ctx):
@@ -3,8 +3,9 @@ from arekit.common.pipeline.items.base import BasePipelineItem
3
3
 
4
4
  class MapPipelineItem(BasePipelineItem):
5
5
 
6
- def __init__(self, map_func=None):
6
+ def __init__(self, map_func=None, **kwargs):
7
7
  assert(callable(map_func))
8
+ super(MapPipelineItem, self).__init__(**kwargs)
8
9
  self._map_func = map_func
9
10
 
10
11
  def apply_core(self, input_data, pipeline_ctx):
@@ -9,5 +9,9 @@ class MapNestedPipelineItem(MapPipelineItem):
9
9
  suppose to be mapped with the passed pipeline context.
10
10
  """
11
11
 
12
+ def __init__(self, **kwargs):
13
+ super(MapNestedPipelineItem, self).__init__(**kwargs)
14
+ pass
15
+
12
16
  def apply_core(self, input_data, pipeline_ctx):
13
17
  return map(lambda item: self._map_func(item, pipeline_ctx), input_data)
@@ -0,0 +1,32 @@
1
+ class BatchIterator:
2
+
3
+ def __init__(self, data_iter, batch_size, end_value=None):
4
+ assert(isinstance(batch_size, int) and batch_size > 0)
5
+ assert(callable(end_value) or end_value is None)
6
+ self.__data_iter = data_iter
7
+ self.__index = 0
8
+ self.__batch_size = batch_size
9
+ self.__end_value = end_value
10
+
11
+ def __iter__(self):
12
+ return self
13
+
14
+ def __next__(self):
15
+ buffer = []
16
+ while True:
17
+ try:
18
+ data = next(self.__data_iter)
19
+ except StopIteration:
20
+ break
21
+ buffer.append(data)
22
+ if len(buffer) == self.__batch_size:
23
+ break
24
+
25
+ if len(buffer) > 0:
26
+ self.__index += 1
27
+ return buffer
28
+
29
+ if self.__end_value is None:
30
+ raise StopIteration
31
+ else:
32
+ return self.__end_value()
@@ -0,0 +1,36 @@
1
+ import sqlite3
2
+
3
+
4
+ class SQLiteProvider(object):
5
+
6
+ @staticmethod
7
+ def write(data_it, target, data2col_func, table_name, columns, sqlite3_column_types,
8
+ id_column_name="id", id_column_type="TEXT"):
9
+ assert(callable(data2col_func))
10
+ assert(isinstance(columns, list))
11
+ assert(isinstance(sqlite3_column_types, list))
12
+ assert(len(columns) == len(sqlite3_column_types))
13
+
14
+ with sqlite3.connect(target) as con:
15
+ cur = con.cursor()
16
+
17
+ # Provide the ID column.
18
+ columns = [id_column_name] + columns
19
+ sqlite3_column_types = [id_column_type] + sqlite3_column_types
20
+
21
+ # Compose the whole columns list.
22
+ content = ", ".join([" ".join(item) for item in zip(columns, sqlite3_column_types)])
23
+ cur.execute(f"CREATE TABLE IF NOT EXISTS {table_name}({content})")
24
+ cur.execute(f"CREATE INDEX IF NOT EXISTS i_id ON {table_name}({id_column_name})")
25
+
26
+ for uid, data in data_it:
27
+ r = cur.execute(f"SELECT EXISTS(SELECT 1 FROM {table_name} WHERE {id_column_name}='{uid}');")
28
+ ans = r.fetchone()[0]
29
+ if ans == 1:
30
+ continue
31
+
32
+ params = ", ".join(tuple(['?'] * (len(columns))))
33
+ cur.execute(f"INSERT INTO {table_name} VALUES ({params})", [str(uid)] + data2col_func(data))
34
+ con.commit()
35
+
36
+ cur.close()
@@ -1,28 +1,34 @@
1
1
  from collections.abc import Iterable
2
2
 
3
3
  from arekit.common.bound import Bound
4
- from arekit.common.text.partitioning.base import BasePartitioning
5
4
 
6
5
 
7
- class StringPartitioning(BasePartitioning):
6
+ class Partitioning(object):
8
7
  """ NOTE: considering that provided parts
9
8
  has no intersections between each other
10
9
  """
11
10
 
11
+ list_reg_types = {
12
+ "str": lambda p, item: p.append(item),
13
+ "list": lambda p, item: p.extend(item)
14
+ }
15
+
16
+ def __init__(self, text_fmt):
17
+ assert(isinstance(text_fmt, str) and text_fmt in self.list_reg_types)
18
+ self.__reg_part = self.list_reg_types[text_fmt]
19
+
12
20
  def provide(self, text, parts_it):
13
- assert(isinstance(text, str))
14
21
  assert(isinstance(parts_it, Iterable))
15
22
 
16
- start = 0
17
23
  parts = []
24
+ start = 0
25
+
18
26
  for value, bound in parts_it:
19
27
  assert(isinstance(bound, Bound))
20
28
  assert(bound.Position >= start)
21
29
 
22
30
  # Release everything till the current value position.
23
- part = text[start:bound.Position]
24
-
25
- parts.append(part)
31
+ self.__reg_part(p=parts, item=text[start:bound.Position])
26
32
 
27
33
  # Release the entity value.
28
34
  parts.extend([value])
@@ -30,7 +36,6 @@ class StringPartitioning(BasePartitioning):
30
36
  start = bound.Position + bound.Length
31
37
 
32
38
  # Release everything after the last entity.
33
- last_part = text[start:len(text)]
34
- parts.extend([last_part])
39
+ self.__reg_part(p=parts, item=text[start:len(text)])
35
40
 
36
41
  return parts
@@ -1,6 +1,5 @@
1
1
  import sys
2
2
  import os
3
- import requests
4
3
  from tqdm import tqdm
5
4
 
6
5
 
@@ -84,46 +83,3 @@ def progress_bar_iter(iterable, desc="", unit='it'):
84
83
  leave=True,
85
84
  ncols=120,
86
85
  unit=unit)
87
-
88
-
89
- def get_default_download_dir():
90
- """ Refered to NLTK toolkit approach
91
- https://github.com/nltk/nltk/blob/8e771679cee1b4a9540633cc3ea17f4421ffd6c0/nltk/downloader.py#L1051
92
- """
93
-
94
- # On Windows, use %APPDATA%
95
- if sys.platform == "win32" and "APPDATA" in os.environ:
96
- homedir = os.environ["APPDATA"]
97
-
98
- # Otherwise, install in the user's home directory.
99
- else:
100
- homedir = os.path.expanduser("~/")
101
- if homedir == "~/":
102
- raise ValueError("Could not find a default download directory")
103
-
104
- return os.path.join(homedir, ".arekit")
105
-
106
-
107
- def download(dest_file_path, source_url):
108
- """ Refered to https://github.com/nicolay-r/ner-bilstm-crf-tensorflow/blob/master/ner/utils.py
109
- Simple http file downloader
110
- """
111
- print(('Downloading from {src} to {dest}'.format(src=source_url, dest=dest_file_path)))
112
-
113
- sys.stdout.flush()
114
- datapath = os.path.dirname(dest_file_path)
115
-
116
- if not os.path.exists(datapath):
117
- os.makedirs(datapath, mode=0o755)
118
-
119
- dest_file_path = os.path.abspath(dest_file_path)
120
-
121
- r = requests.get(source_url, stream=True)
122
- total_length = int(r.headers.get('content-length', 0))
123
-
124
- with open(dest_file_path, 'wb') as f:
125
- pbar = tqdm(total=total_length, unit='B', unit_scale=True)
126
- for chunk in r.iter_content(chunk_size=32 * 1024):
127
- if chunk: # filter out keep-alive new chunks
128
- pbar.update(len(chunk))
129
- f.write(chunk)
@@ -2,7 +2,8 @@ from arekit.common.data.input.providers.const import IDLE_MODE
2
2
  from arekit.common.data.input.providers.contents import ContentsProvider
3
3
  from arekit.common.linkage.base import LinkedDataWrapper
4
4
  from arekit.common.linkage.text_opinions import TextOpinionsLinkage
5
- from arekit.common.pipeline.base import BasePipeline
5
+ from arekit.common.pipeline.base import BasePipelineLauncher
6
+ from arekit.common.pipeline.context import PipelineContext
6
7
  from arekit.common.text_opinions.base import TextOpinion
7
8
 
8
9
 
@@ -13,7 +14,7 @@ class InputTextOpinionProvider(ContentsProvider):
13
14
  results in a TextOpinionLinkage instances.
14
15
  pipeline: id -> ... -> TextOpinionLinkage[]
15
16
  """
16
- assert(isinstance(pipeline, BasePipeline))
17
+ assert(isinstance(pipeline, list))
17
18
  self.__pipeline = pipeline
18
19
  self.__current_id = None
19
20
 
@@ -30,7 +31,16 @@ class InputTextOpinionProvider(ContentsProvider):
30
31
 
31
32
  def from_doc_ids(self, doc_ids, idle_mode=False):
32
33
  self.__current_id = 0
33
- for linkage in self.__pipeline.run(doc_ids, params_dict={IDLE_MODE: idle_mode}):
34
+
35
+ ctx = PipelineContext(d={
36
+ "result": doc_ids,
37
+ IDLE_MODE: idle_mode
38
+ })
39
+
40
+ # Launching pipeline with the passed context
41
+ BasePipelineLauncher.run(pipeline=self.__pipeline, pipeline_ctx=ctx)
42
+
43
+ for linkage in ctx.provide("result"):
34
44
  assert(isinstance(linkage, LinkedDataWrapper))
35
45
  if isinstance(linkage, TextOpinionsLinkage):
36
46
  self.__assign_ids(linkage)
@@ -0,0 +1,14 @@
1
+ from arekit.contrib.utils.data.readers.base import BaseReader
2
+ from arekit.contrib.utils.data.storages.sqlite_based import SQliteBasedRowsStorage
3
+
4
+
5
+ class SQliteReader(BaseReader):
6
+
7
+ def __init__(self, table_name):
8
+ self.__table_name = table_name
9
+
10
+ def extension(self):
11
+ return ".sqlite"
12
+
13
+ def read(self, target):
14
+ return SQliteBasedRowsStorage(path=target, table_name=self.__table_name)
@@ -33,7 +33,12 @@ class RowCacheStorage(BaseRowsStorage):
33
33
 
34
34
  # Expand with columns that are forced to be provided.
35
35
  existed_set = set(self.__column_names)
36
- self.__column_names += [c for c in self.__force_collect_columns if c not in existed_set]
36
+
37
+ # Calculate extension: columns that were not mentioned in column names list.
38
+ extension = [c for c in self.__force_collect_columns if c not in existed_set]
39
+
40
+ self.__column_names += extension
41
+ self.__column_types += [str] * len(extension)
37
42
 
38
43
  def iter_column_names(self):
39
44
  return iter(self.__column_names)
@@ -0,0 +1,17 @@
1
+ import sqlite3
2
+ from arekit.common.data.storages.base import BaseRowsStorage
3
+
4
+
5
+ class SQliteBasedRowsStorage(BaseRowsStorage):
6
+
7
+ def __init__(self, path, table_name):
8
+ self.__path = path
9
+ self.__table_name = table_name
10
+ self.__conn = None
11
+
12
+ def _iter_rows(self):
13
+ with sqlite3.connect(self.__path) as conn:
14
+ cursor = conn.execute(f"select * from {self.__table_name}")
15
+ for row_index, row in enumerate(cursor.fetchall()):
16
+ row_dict = {cursor.description[i][0]: value for i, value in enumerate(row)}
17
+ yield row_index, row_dict