__main__.py 12 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406
  1. import os
  2. import sys
  3. from importlib.util import find_spec
  4. from configparser import ConfigParser
  5. import threading
  6. import json
  7. import time
  8. import schedule
  9. import requests
  10. from flask import Flask, g, redirect, url_for, render_template, jsonify, request, send_from_directory
  11. from flask_cors import CORS
  12. from .web import api
  13. from . import content_system as h_cs
  14. from . import item_collections, view_model as h_vm
  15. from .item_collections import item_collections_bp
  16. from . import schedule_system as h_sched
  17. theme_bootstrap5_enabled = False
  18. if find_spec('theme_bootstrap5'):
  19. from theme_bootstrap5 import hogumathi_theme_bootstrap5_bp
  20. theme_bootstrap5_enabled = True
  21. if find_spec('twitter_v2_facade'):
  22. import twitter_v2_facade
  23. from twitter_v2_facade import twitter_app as twitter_v2
  24. from twitter_v2_facade import oauth2_login
  25. twitter_enabled = True
  26. else:
  27. print('twitter module not found.')
  28. twitter_enabled = False
  29. if find_spec('twitter_v2_live_facade'):
  30. import twitter_v2_live_facade
  31. from twitter_v2_live_facade import twitter_app as twitter_v2_live
  32. import brands
  33. from brands import brands_bp
  34. twitter_live_enabled = True
  35. else:
  36. print('twitter live module not found.')
  37. twitter_live_enabled = False
  38. if find_spec('twitter_archive_facade'):
  39. import twitter_archive_facade
  40. from twitter_archive_facade import twitter_app as twitter_archive
  41. archive_enabled = True
  42. else:
  43. print('twitter archive module not found.')
  44. archive_enabled = False
  45. if find_spec('mastodon_facade'):
  46. from mastodon_facade import twitter_app as mastodon
  47. mastodon_enabled = True
  48. else:
  49. print('mastodon module not found.')
  50. mastodon_enabled = False
  51. if find_spec('feeds_facade'):
  52. import feeds_facade
  53. from feeds_facade import twitter_app as feeds
  54. feeds_enabled = True
  55. else:
  56. print('feeds module not found.')
  57. feeds_enabled = False
  58. if find_spec('youtube_v3_facade'):
  59. import youtube_v3_facade
  60. from youtube_v3_facade import youtube_app as youtube
  61. youtube_enabled = True
  62. else:
  63. print('youtube module not found.')
  64. youtube_enabled = False
  65. if find_spec('messages_facade'):
  66. from messages_facade import messages_facade as messages
  67. messages_enabled = True
  68. else:
  69. print('messages module not found.')
  70. messages_enabled = False
  71. if find_spec('instagram_embed_facade'):
  72. import instagram_embed_facade
  73. instagram_embed_enabled = True
  74. else:
  75. print('instagram_embed module not found.')
  76. instagram_embed_enabled = False
  77. if find_spec('instagram_facade'):
  78. import instagram_facade
  79. instagram_enabled = True
  80. else:
  81. print('instagram module not found.')
  82. instagram_enabled = False
  83. if find_spec('git_facade'):
  84. import git_facade
  85. git_enabled = True
  86. else:
  87. print('git module not found.')
  88. git_enabled = False
  89. if find_spec('bitchute_facade'):
  90. import bitchute_facade
  91. bitchute_enabled = True
  92. else:
  93. print('bitchute module not found.')
  94. bitchute_enabled = False
  95. if find_spec('videojs'):
  96. from videojs import videojs_bp
  97. videojs_enabled = True
  98. else:
  99. print('videojs module not found.')
  100. videojs_enabled = False
  101. if find_spec('visjs'):
  102. from visjs import visjs_bp
  103. visjs_enabled = True
  104. else:
  105. print('messages module not found.')
  106. visjs_enabled = False
  107. if find_spec('socialdata_facade'):
  108. import socialdata_facade
  109. socialdata_enabled = True
  110. else:
  111. print('socialdata module not found.')
  112. socialdata_enabled = False
  113. """
  114. By default we use embed because it's less likely to get banned, because it does not
  115. use scraping. Once we've used scraping for a while and not been banned we can switch to that.
  116. Biz note: scraping could also be an early access feature.
  117. """
  118. print('disabling instagram_facade to avoid scraping... enable at your own risk')
  119. instagram_enabled = False
  120. #messages_enabled = False
  121. #twitter_live_enabled = True
  122. add_account_enabled = True
  123. def record_content_analytics (content_id, content):
  124. if type(content) == h_vm.CollectionPage:
  125. prefix = None
  126. if content_id.startswith('twitter:feed:user:') or content_id.startswith('twitter:bookmarks:') or content_id.startswith('twitter:feed:reverse_chronological:user:'):
  127. prefix = 'twitter:tweet:'
  128. if not prefix:
  129. return;
  130. for item in content.items:
  131. h_cs.invoke_hooks('got_content', f'{prefix}{item.id}', item)
  132. elif type(content) == h_vm.FeedItem:
  133. print(f'record_content_analytics: feed item: {content_id}')
  134. elif type(content) == h_vm.FeedServiceUser:
  135. print(f'record_content_analytics: user: {content_id}')
  136. else:
  137. print(f'record_content_analytics: unknown type: {content_id}')
  138. def browser_native_messaging ():
  139. # stdin = binary stream. ((payload length, payload), ...)
  140. print('hey. awaiting msg.')
  141. # "Note To write or read binary data from/to the standard streams,
  142. # use the underlying binary buffer object. For example, to write bytes to stdout, use sys.stdout.buffer.write(b'abc')."
  143. # https://docs.python.org/3/library/sys.html#sys.stdin
  144. stdin = sys.stdin
  145. msg_len = read_int(stdin.buffer) # example read_int is in Apple Notes exporter code
  146. while msg_len:
  147. print(f'new msg_len: {msg_len}')
  148. msg_bytes = stdin.buffer.read(msg_len)
  149. msg = json.loads(msg_bytes.decode('utf-8'))
  150. print(msg)
  151. print('awaiting next msg...')
  152. msg_len = read_int(stdin.buffer)
  153. return 0
  154. if __name__ == '__main__':
  155. if len(sys.argv) >= 2 and sys.argv[1] == 'browser_native_messaging':
  156. print('browser_native_messaging')
  157. exit(browser_native_messaging())
  158. glitch_enabled = os.environ.get('PROJECT_DOMAIN') and True
  159. t = os.environ.get('BEARER_TOKEN')
  160. PORT = int(os.environ.get('PORT', 5000))
  161. HOST = os.environ.get('HOST', '127.0.0.1')
  162. archive_enabled = os.environ.get('ARCHIVE_TWEETS_PATH') and True
  163. notes_app_url = os.environ.get('NOTES_APP_URL')
  164. bedrss_app_url = os.environ.get('NOTES_APP_URL')
  165. plots_app_url = os.environ.get('PLOTS_APP_URL')
  166. videos_app_url = os.environ.get('VIDEOS_APP_URL')
  167. messages_app_url = os.environ.get('MESSAGES_APP_URL')
  168. if not os.path.exists('.data'):
  169. os.mkdir('.data')
  170. if not os.path.exists('.data/cache'):
  171. os.mkdir('.data/cache')
  172. # HACK - environ from .env isn't set yet when the import happens. We should call an init function somewhere.
  173. oauth2_login.app_access_token = os.environ.get("BEARER_TOKEN")
  174. oauth2_login.app_consumer_key = os.environ.get("TWITTER_CONSUMER_KEY")
  175. oauth2_login.app_secret_key = os.environ.get("TWITTER_CONSUMER_SECRET")
  176. h_cs.register_hook('got_content', record_content_analytics)
  177. @api.before_request
  178. def add_config ():
  179. g.twitter_enabled = twitter_enabled
  180. g.archive_enabled = archive_enabled
  181. g.mastodon_enabled = mastodon_enabled
  182. g.feeds_enabled = feeds_enabled
  183. g.youtube_enabled = youtube_enabled
  184. g.messages_enabled = messages_enabled
  185. g.twitter_live_enabled = twitter_live_enabled
  186. g.add_account_enabled = add_account_enabled
  187. g.glitch_enabled = glitch_enabled
  188. g.videojs_enabled = videojs_enabled
  189. g.visjs_enabled = visjs_enabled
  190. if glitch_enabled:
  191. g.app_url = 'https://{}.glitch.me'.format( os.environ.get('PROJECT_DOMAIN') )
  192. else:
  193. g.app_url = 'http://{}:{}'.format('localhost', PORT)
  194. if notes_app_url:
  195. g.notes_app_url = notes_app_url
  196. if bedrss_app_url:
  197. g.bedrss_app_url = bedrss_app_url
  198. if plots_app_url:
  199. g.plots_app_url = plots_app_url
  200. if videos_app_url:
  201. g.videos_app_url = videos_app_url
  202. if messages_app_url:
  203. g.messages_app_url = messages_app_url
  204. if theme_bootstrap5_enabled:
  205. g.theme_bootstrap5_enabled = theme_bootstrap5_enabled
  206. @api.context_processor
  207. def inject_config ():
  208. config = {}
  209. config['twitter_enabled'] = twitter_enabled
  210. config['archive_enabled'] = archive_enabled
  211. config['mastodon_enabled'] = mastodon_enabled
  212. config['feeds_enabled'] = feeds_enabled
  213. config['youtube_enabled'] = youtube_enabled
  214. config['messages_enabled'] = messages_enabled
  215. config['twitter_live_enabled'] = twitter_live_enabled
  216. config['add_account_enabled'] = add_account_enabled
  217. config['glitch_enabled'] = glitch_enabled
  218. config['videojs_enabled'] = videojs_enabled
  219. config['visjs_enabled'] = visjs_enabled
  220. config['theme_bootstrap5_enabled'] = theme_bootstrap5_enabled
  221. if notes_app_url:
  222. config['notes_app_url'] = notes_app_url
  223. if bedrss_app_url:
  224. config['bedrss_app_url'] = bedrss_app_url
  225. if plots_app_url:
  226. config['plots_app_url'] = plots_app_url
  227. if videos_app_url:
  228. config['videos_app_url'] = videos_app_url
  229. if messages_app_url:
  230. config['messages_app_url'] = messages_app_url
  231. return config
  232. api.secret_key = os.environ.get('FLASK_SECRET')
  233. api.config['TEMPLATES_AUTO_RELOAD'] = True
  234. # the new way
  235. #api.config['notes_app_url'] = notes_app_url
  236. if theme_bootstrap5_enabled:
  237. api.register_blueprint(hogumathi_theme_bootstrap5_bp)
  238. if videojs_enabled:
  239. api.register_blueprint(videojs_bp, url_prefix='/lib/videojs')
  240. if visjs_enabled:
  241. api.register_blueprint(visjs_bp, url_prefix='/lib/visjs')
  242. twitter_v2_facade.register_content_sources()
  243. api.register_blueprint(twitter_v2, url_prefix='/twitter')
  244. if archive_enabled:
  245. twitter_archive_facade.register_content_sources()
  246. api.register_blueprint(twitter_archive, url_prefix='/twitter-archive')
  247. if mastodon_enabled:
  248. api.register_blueprint(mastodon, url_prefix='/mastodon')
  249. if youtube_enabled:
  250. youtube_v3_facade.register_content_sources()
  251. api.register_blueprint(youtube, url_prefix='/youtube')
  252. if messages_enabled:
  253. api.register_blueprint(messages, url_prefix='/messages')
  254. item_collections.register_content_sources()
  255. api.register_blueprint(item_collections_bp, url_prefix='/collections')
  256. if twitter_live_enabled:
  257. twitter_v2_live_facade.register_content_sources()
  258. api.register_blueprint(twitter_v2_live, url_prefix='/twitter-live')
  259. brands.register_content_sources()
  260. api.register_blueprint(brands_bp, url_prefix='/')
  261. if instagram_enabled:
  262. instagram_facade.register_content_sources()
  263. if instagram_embed_enabled:
  264. instagram_embed_facade.register_content_sources()
  265. if feeds_enabled:
  266. if not os.path.exists('.data/cache/feeds'):
  267. os.mkdir('.data/cache/feeds')
  268. feeds_facade.register_content_sources()
  269. api.register_blueprint(feeds, url_prefix='/feeds')
  270. if git_enabled:
  271. git_facade.register_content_sources()
  272. if bitchute_enabled:
  273. bitchute_facade.register_content_sources()
  274. if socialdata_enabled:
  275. socialdata_facade.register_content_sources()
  276. #CORS(api)
  277. sched_app = h_sched.ScheduleApplication()
  278. print(f'created schedule app: {sched_app}')
  279. running = True
  280. def schedule_main():
  281. print(f'running schedule app: {sched_app}')
  282. while running:
  283. sched_app.run_pending()
  284. time.sleep(1)
  285. def say_something ():
  286. print ('-- something')
  287. sched_app.schedule_job(schedule.every(61).seconds, say_something)
  288. schedule_thread = threading.Thread(target=schedule_main)
  289. schedule_thread.start()
  290. api.run(port=PORT, host=HOST)
  291. running = False
  292. schedule_thread.join()