init

2026-09-02 23:47:00 +03:00
parent b4d0b0c8e9
commit a8830bcd91
77 changed files with 2998 additions and 0 deletions
@@ -0,0 +1,80 @@
+package agzam.vyatsu.data.remote
+
+import agzam.vyatsu.domain.model.DateRange
+import agzam.vyatsu.domain.model.Group
+import agzam.vyatsu.domain.model.Season
+import android.util.Log
+import org.jsoup.Jsoup
+
+/**
+ * Основано на https://github.com/alirzaev/vyatsu-schedule-backend/blob/master/utils/extractors.py
+ * @author Agzam
+ */
+object GroupsParser {
+
+    private val groupPattern = Regex("""/reports/schedule/Group/(\d{4,})_([12])_(\d{8})_(\d{8})\.pdf""")
+
+    private val shorthands = mapOf(
+        "Институт биологии и биотехнологии (факультет)(ОРУ)" to "ИББТ",
+        "Институт химии и экологии (факультет) (ОРУ)" to "ИнХимЭк",
+        "Факультет автоматики и вычислительной техники (ОРУ)" to "ФАВТ",
+        "Факультет истории, политических наук и культурологии (ОРУ)" to "ФИПНК",
+        "Факультет компьютерных и физико-математических наук (ОРУ)" to "ФКФМН",
+        "Факультет лингвистики (ОРУ)" to "ФЛ",
+        "Факультет менеджмента и сервиса (ОРУ)" to "ФМиС",
+        "Факультет педагогики и психологии (ОРУ)" to "ФПП",
+        "Факультет строительства и архитектуры (ОРУ)" to "ФСА",
+        "Факультет технологий, инжиниринга и дизайна (ОРУ)" to "ФТИД",
+        "Факультет физической культуры и спорта (ОРУ)" to "ФФКС",
+        "Факультет филологии и медиакоммуникаций (ОРУ)" to "ФФМ",
+        "Факультет экономики и финансов (ОРУ)" to "ФЭиФ",
+        "Электротехнический факультет (ОРУ)" to "ЭТФ",
+        "Юридический институт (факультет) (ОРУ)" to "ЮИ"
+    )
+
+    /**
+     * Извлекает список групп с HTML страницы
+     */
+    fun parseGroups(html: String): List<Group> {
+        val doc = Jsoup.parse(html)
+        val container = doc.selectFirst("div.column-center_rasp") ?: return emptyList()
+        val tables = container.select("table[style~=border: none !important]")
+
+        return tables.flatMap { table ->
+            table.select("> tbody > tr, > tr").flatMap { row ->
+                Log.v("ROW", row.html());
+                val faculty = shortname(row.selectFirst("div.fak_name")?.text().orEmpty()) //
+
+                row.select("div.grpPeriod").map { grp ->
+                    Group(
+                        id = grp.attr("data-grp_period_id").dropLast(1),
+                        name = removeParenthesized(grp.text()),
+                        faculty = faculty
+                    )
+                }
+            }
+        }
+    }
+
+    fun parseDateRanges(html: String): Map<String, Map<Season, List<DateRange>>> {
+        val result = HashMap<String, HashMap<Season, MutableList<DateRange>>>()
+        groupPattern.findAll(html).forEach { m ->
+            val groupId = m.groupValues[1]
+            val season = if (m.groupValues[2] == "1") Season.AUTUMN else Season.SPRING
+            val range = DateRange(m.groupValues[3], m.groupValues[4])
+            result
+                .getOrPut(groupId) { hashMapOf(Season.AUTUMN to mutableListOf(), Season.SPRING to mutableListOf()) }
+                .getOrPut(season) { mutableListOf() }
+                .add(range)
+        }
+        return result
+    }
+
+    private fun shortname(name: String): String {
+        val stripped = removeParenthesized(name)
+        val sh = shorthands[name]
+        return if (sh != null) "$sh ($stripped)" else stripped
+    }
+
+    private fun removeParenthesized(name: String): String = name.replace(Regex("""\(.*?\)\s?"""), "").trim()
+}