wbnews 1.58.3__py2.py3-none-any.whl

wbnews/.coveragerc

@@ -0,0 +1,23 @@
+[report]
+exclude_lines =
+    print()
+    def api_endpoints_root
+    def get_or_create_model_sql
+    def profile_check
+    if hasattr
+    raise Exception
+    raise Http404
+    except:
+
+
+[run]
+omit = */migrations/*
+    manage.py
+    */tests/*
+    */wbnews_config/*
+    */apps.py
+    */docs/*
+    */dynamic_preferences_registry.py
+    */permissions.py
+    */preferences/*
+    */.venv/*

wbnews/__init__.py

@@ -0,0 +1 @@
+__version__ = "1.0.0"

wbnews/admin.py

@@ -0,0 +1,30 @@
+from django.contrib import admin
+
+from .models import News, NewsRelationship, NewsSource
+
+
+@admin.register(NewsRelationship)
+class NewsRelationshipAdmin(admin.ModelAdmin):
+    list_display = ["news", "content_object"]
+    autocomplete_fields = ["news"]
+
+
+@admin.register(News)
+class NewsAdmin(admin.ModelAdmin):
+    search_fields = ("title", "description")
+    raw_id_fields = ["import_source"]
+    autocomplete_fields = [
+        "source",
+    ]
+    list_display = ["title", "language", "tags", "source", "datetime"]
+
+    list_filter = ("source",)
+
+    def get_queryset(self, request):
+        return News.all_objects.select_related("source")
+
+
+@admin.register(NewsSource)
+class NewsSourceAdmin(admin.ModelAdmin):
+    search_fields = ("type", "title", "identifier", "description", "author", "endpoint")
+    list_filter = ("type",)

wbnews/apps.py

@@ -0,0 +1,9 @@
+from django.apps import AppConfig
+from django.utils.module_loading import autodiscover_modules
+
+
+class NewConfig(AppConfig):
+    name = "wbnews"
+
+    def ready(self) -> None:
+        autodiscover_modules("news")

wbnews/factories.py

@@ -0,0 +1,36 @@
+import factory
+from django.conf.global_settings import LANGUAGES
+from django.utils import timezone
+from faker import Factory
+
+from wbnews.models import News, NewsSource
+
+langs = [n for (n, v) in LANGUAGES]
+faker = Factory.create()
+
+
+class NewsSourceFactory(factory.django.DjangoModelFactory):
+    title = factory.Sequence(lambda n: f"source_{n}")
+    identifier = factory.Sequence(lambda n: f"http://myurl_{n}.com")
+    image = factory.Faker("url")
+    description = factory.Faker("sentence", nb_words=32)
+    author = factory.Faker("name")
+    endpoint = factory.Faker("url")
+
+    class Meta:
+        model = NewsSource
+
+
+class NewsFactory(factory.django.DjangoModelFactory):
+    datetime = factory.LazyFunction(timezone.now)
+    title = factory.Faker("sentence", nb_words=32)
+    description = factory.Faker("sentence", nb_words=32)
+    summary = factory.Faker("sentence", nb_words=32)
+    language = factory.Iterator(langs)
+    link = factory.Faker("url")
+    guid = factory.LazyAttribute(lambda o: News.get_default_guid(o.title, o.link))
+    source = factory.SubFactory(NewsSourceFactory)
+
+    class Meta:
+        model = News
+        django_get_or_create = ("guid",)

wbnews/filters/__init__.py

@@ -0,0 +1 @@
+from .news import NewsFilterSet, NewsRelationshipFilterSet

wbnews/filters/news.py

@@ -0,0 +1,46 @@
+from wbcore import filters as wb_filters
+
+from wbnews.models import News, NewsRelationship, NewsSource
+
+
+class NewsFilterSet(wb_filters.FilterSet):
+    datetime = wb_filters.DateTimeRangeFilter()
+
+    class Meta:
+        model = News
+        fields = {"title": ["icontains"], "source": ["exact"], "language": ["exact"]}
+
+
+class NewsRelationshipFilterSet(wb_filters.FilterSet):
+    datetime = wb_filters.DateTimeRangeFilter(method="filter_datetime")
+    title = wb_filters.CharFilter(lookup_expr="icontains")
+    description = wb_filters.CharFilter(lookup_expr="icontains")
+    summary = wb_filters.CharFilter(lookup_expr="icontains")
+    source = wb_filters.ModelMultipleChoiceFilter(
+        label="Source",
+        queryset=NewsSource.objects.all(),
+        endpoint=NewsSource.get_representation_endpoint(),
+        value_key=NewsSource.get_representation_value_key(),
+        label_key=NewsSource.get_representation_label_key(),
+        method="filter_source",
+    )
+    content_type = wb_filters.CharFilter(method="fake_filter", hidden=True)
+    object_id = wb_filters.CharFilter(method="fake_filter", hidden=True)
+
+    def filter_datetime(self, queryset, name, value):
+        if value:
+            return queryset.filter(news__datetime__gte=value.lower, news__datetime__lte=value.upper)
+        return queryset
+
+    def filter_source(self, queryset, name, value):
+        if value:
+            return queryset.filter(news__source__in=value)
+        return queryset
+
+    class Meta:
+        model = NewsRelationship
+        fields = {
+            "analysis": ["icontains"],
+            "sentiment": ["exact"],
+            "important": ["exact"],
+        }