Updated code

This commit is contained in:
Brage 2023-07-18 02:25:31 +02:00
parent d4859b5ab9
commit bf5fcd9093
4 changed files with 22 additions and 11 deletions

View File

@ -5,18 +5,17 @@ class Naming(val fileName: String) {
private set private set
init { init {
cleanedFileName = fileName.apply { cleanedFileName = fileName
removeBracketedText(this) .let { removeBracketedText(it) }
removeParenthesizedText(this) .let { removeParenthesizedText(it) }
removeResolutionAndTags(this) .let { removeResolutionAndTags(it) }
removeInBetweenCharacters(this) .let { removeInBetweenCharacters(it) }
.let { removeExtraWhiteSpace(it) }
removeExtraWhiteSpace(this)
}
} }
fun guessDesiredFileName(): String { fun guessDesiredFileName(): String {
val parts = fileName.split(" - ") val parts = cleanedFileName.split(" - ")
return when { return when {
parts.size == 2 && parts[1].matches(Regex("\\d{4}")) -> { parts.size == 2 && parts[1].matches(Regex("\\d{4}")) -> {
val title = parts[0] val title = parts[0]
@ -36,7 +35,7 @@ class Naming(val fileName: String) {
"$title - $seasonEpisode - $episodeTitle" "$title - $seasonEpisode - $episodeTitle"
} }
else -> fileName else -> cleanedFileName
} }
} }

View File

@ -7,6 +7,15 @@ import org.junit.jupiter.params.ParameterizedTest
import org.junit.jupiter.params.provider.MethodSource import org.junit.jupiter.params.provider.MethodSource
class NamingTest { class NamingTest {
@Test
fun checkThatBracketsGetsRemoved() {
val input = "[AAA] Sir fancy - 13 [1080p HEVC][000000]"
val name = Naming(input)
assertThat(name.guessDesiredTitle()).doesNotContain("[")
}
/* /*
@ParameterizedTest @ParameterizedTest
@MethodSource("serieOnlyTest") @MethodSource("serieOnlyTest")

View File

@ -1,3 +1,3 @@
spring.output.ansi.enabled=always spring.output.ansi.enabled=always
logging.level.org.apache.kafka=WARN logging.level.org.apache.kafka=WARN
logging.level.root=DEBUG #logging.level.root=DEBUG

View File

@ -76,13 +76,16 @@ class KafkaConsumerThread(threading.Thread):
if self.shutdown.is_set(): if self.shutdown.is_set():
break break
logger.info("Received message: key=%s, value=%s", message.key, message.value)
# Sjekk om meldingen har målnøkkelen # Sjekk om meldingen har målnøkkelen
if message.key == "request:metadata:obtain" or message.key == "event:reader:received-file": if message.key == "request:metadata:obtain" or message.key == "event:reader:received-file":
logger.info("Received message: key=%s, value=%s", message.key, message.value)
# Opprett en ny tråd for å håndtere meldingen # Opprett en ny tråd for å håndtere meldingen
handler_thread = MessageHandlerThread(message) handler_thread = MessageHandlerThread(message)
handler_thread.start() handler_thread.start()
else:
logger.info("Ignorert message: key=%s, value=%s", message.key, message.value)
consumer.close() consumer.close()
logger.info("Kafka Consumer stopped") logger.info("Kafka Consumer stopped")